Update agent.py

agent.py

@@ -0,0 +1,36 @@
+import os
+import requests
+
+GROQ_API_URL = "https://api.groqcloud.com/v1/chat/completions"
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Secure key from Hugging Face Secrets
+
+# Helper function to load system prompt from txt files
+def load_prompt(filename):
+    with open(f'config/system_prompts/{filename}', 'r') as file:
+        return file.read()
+
+# Function to call Groq API with system prompt and user input
+def call_groq(system_prompt, user_input):
+    headers = {"Authorization": f"Bearer {GROQ_API_KEY}"}
+    payload = {
+        "model": "mixtral-8x7b-32768",
+        "messages": [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_input}
+        ]
+    }
+    response = requests.post(GROQ_API_URL, json=payload, headers=headers)
+    return response.json()['choices'][0]['message']['content']
+
+def ad_copy_agent(product, description, audience, tone):
+    system_prompt = load_prompt("ad_copy_prompt.txt")
+    user_input = f"Product: {product}\nDescription: {description}\nAudience: {audience}\nTone: {tone}"
+    return call_groq(system_prompt, user_input)
+
+def sentiment_agent(social_data):
+    system_prompt = load_prompt("sentiment_prompt.txt")
+    return call_groq(system_prompt, social_data)
+
+def timing_agent(platform):
+    system_prompt = load_prompt("timing_prompt.txt")
+    return call_groq(system_prompt, f"Platform: {platform}")