Update app.py

app.py

@@ -51,9 +51,10 @@ class GeminiModel:
                 contents=[{"role": "user", "parts": [{"text": prompt}]}],
                 generation_config=self.generation_config
             )
-            return response.candidates[0].content.parts[0].text.strip()
+            # Return a dictionary that CodeAgent expects
+            return {"content": response.candidates[0].content.parts[0].text.strip()}
         except Exception as e:
-            return f"Error during Gemini call: {str(e)}"
+            return {"content": f"Error during Gemini call: {str(e)}"}"
 
 # Define BasicAgent properly
 class BasicAgent: