Update app.py

app.py

@@ -178,32 +178,24 @@ class BasicAgent:
 
         self.search_tool = DuckDuckGoSearchTool()
 
-        # Dummy model placeholder, not used
         self.agent = CodeAgent(
             tools=[self.search_tool],
-            model=self,  # Pass self as the model
+            model=self,  # Pass self as model
             planning_interval=3
         )
 
-    def __call__(self, prompt: str, **kwargs) -> object:
-        """Directly call Gemini API for each prompt."""
+    def __call__(self, prompt: str, **kwargs) -> str:
+        """Directly call Gemini API for each prompt, returning a string."""
         try:
             response = client.generate_content(
                 model=model_id,
                 contents=[{"role": "user", "parts": [{"text": prompt}]}],
                 generation_config=generation_config
             )
-            answer = response.candidates[0].content.parts[0].text.strip()
-            # Wrap in a simple object with .content attribute
-            class Output:
-                def __init__(self, content):
-                    self.content = content
-            return Output(content=answer)
+            answer = response.text
+            return answer 
         except Exception as e:
-            class Output:
-                def __init__(self, content):
-                    self.content = content
-            return Output(content=f"Error: {str(e)}")
+            return f"Error during Gemini call: {str(e)}"
 
     def run_agent(self, question: str) -> str:
         """Run the agent on a question."""
@@ -214,7 +206,7 @@ class BasicAgent:
             return result
         except Exception as e:
             return f"Error running agent: {str(e)}"
-
+            
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,