Update agent.py

agent.py

@@ -4,6 +4,9 @@ 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
 
+if not GROQ_API_KEY:
+    raise ValueError("GROQ_API_KEY is missing! Set it in 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:
@@ -20,7 +23,14 @@ def call_groq(system_prompt, user_input):
         ]
     }
     response = requests.post(GROQ_API_URL, json=payload, headers=headers)
-    return response.json()['choices'][0]['message']['content']
+    if response.status_code != 200:
+        raise Exception(f"Groq API error: {response.status_code} - {response.text}")
+    
+    data = response.json()
+    if 'choices' not in data or not data['choices']:
+        raise Exception("Groq API returned no choices.")
+
+    return data['choices'][0]['message']['content']
 
 def ad_copy_agent(product, description, audience, tone):
     system_prompt = load_prompt("ad_copy_prompt.txt")