Update guardrail.py

guardrail.py

@@ -1,8 +1,12 @@
 from together import Together
-from helper import get_together_api_key
+from dotenv import load_dotenv
+import os
 
+# Load environment variables
+load_dotenv()
+api_key = os.getenv("API_KEY")
 # Initialize Together client
-client = Together(api_key=get_together_api_key())
+client = Together(api_key=api_key)
 
 # Set the safety content policy
 safe_content_policy = {