Update guardrail.py

guardrail.py

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