Update app.py

app.py

@@ -143,15 +143,20 @@ class BasicAgent:
         llm = ChatOpenAI(model="gpt-4o", temperature=0)
         self.agent= CodeAgent(
             tools = [add, subtract, multiply, divide, web_search, image_generation_tool],
-            model=llm,
+            llm=llm,
         )
-    def __call__(self, question: str,context: str="") -> str:
+    def __call__(self, question: str, context: str = "") -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         # Inject system prompt + question
         question_with_prompt = f"{self.system_prompt}\n\nContext: {context}\n\nQuestion: {question.strip()}"
         answer = self.agent.run(question_with_prompt)
-        print(f"Agent returning answer: {answer.strip()}")
-        return answer.strip()
+        # Fix: handle dict or string
+        if isinstance(answer, dict) and "content" in answer:
+            result = answer["content"]
+        else:
+            result = str(answer)
+        print(f"Agent returning answer: {result.strip()}")
+        return result.strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """