Create agents.py

agents.py

@@ -0,0 +1,62 @@
+import requests
+import os
+
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+LLAMA_MODEL = "llama3-8b-8192"
+SLM_MODEL = "slm-1b"
+
+def call_groq(prompt, model=LLAMA_MODEL):
+    response = requests.post(
+        "https://api.groq.com/openai/v1/chat/completions",
+        headers={"Authorization": f"Bearer {GROQ_API_KEY}"},
+        json={
+            "model": model,
+            "messages": [{"role": "user", "content": prompt}],
+            "temperature": 0.7
+        }
+    )
+    return response.json()["choices"][0]["message"]["content"]
+
+def get_prompt(entry, mood, mode):
+    if "Workplace" in mode:
+        return f"""
+You are a workplace CBT/NLP counselor. Analyze this journal:
+
+"{entry}"
+
+Mood: {mood}
+
+Return:
+- Summary
+- Workplace Stressors
+- Cognitive Distortions
+- Reframe
+- Affirmation
+"""
+    else:
+        return f"""
+You are a CBT/NLP journaling agent. Analyze the following:
+
+"{entry}"
+
+Mood: {mood}
+
+Return:
+- Summary
+- Emotional Triggers
+- Distortions
+- CBT Reframe
+- Daily Affirmation
+"""
+
+def run_journaling_pipeline(mood, entry, mode):
+    if not entry.strip():
+        return {"error": "Empty entry."}
+    prompt = get_prompt(entry, mood, mode)
+    output = call_groq(prompt)
+    return {
+        "mood": mood,
+        "mode": mode,
+        "entry": entry,
+        "response": output
+    }