Update app.py

app.py

@@ -28,20 +28,20 @@ class GeminiFlashModel:
         self.model = genai.GenerativeModel(model_name)
 
     def generate(self, messages, stop_sequences=None, **kwargs):
-        # Insert system prompt if missing
         if isinstance(messages, list):
             if not any(m["role"] == "system" for m in messages):
                 messages = [{"role": "system", "content": SYSTEM_PROMPT}] + messages
-        else:
-            raise TypeError("Expected 'messages' to be a list of dicts.")
+            else:
+                raise TypeError("Expected 'messages' to be a list of dicts.")
+                
+                prompt = "\n".join(f"{m['role'].capitalize()}: {m['content']}" for m in messages)
+                
+                try:
+                    response = self.model.generate_content(prompt)
+                    return {"content": response.text.strip()}  # ✅ Return a dict with 'content'
+                except Exception as e:
+                    return {"content": f"GENERATION ERROR: {e}"}  # ✅ Error message also in 'content'
 
-        prompt = "\n".join(f"{m['role'].capitalize()}: {m['content']}" for m in messages)
-
-        try:
-            response = self.model.generate_content(prompt)
-            return response.text.strip()
-        except Exception as e:
-            return f"GENERATION ERROR: {e}"
 
 # Agent using Gemini
 class MyAgent: