Update app.py

app.py

@@ -3,7 +3,8 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, LiteLLMModel, GoogleSearchTool
+import asyncio
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, LiteLLMModel, GoogleSearchTool, WikipediaSearchTool 
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -20,16 +21,26 @@ class BasicAgent:
         )
         self.agent = CodeAgent(
             model = model,
-            tools=[GoogleSearchTool()],
+            tools=[GoogleSearchTool(), WikipediaSearchTool() ],
             add_base_tools=True,  # Add any additional base tools
             max_steps=15,
             planning_interval=3   # Enable planning every 3 steps
         )
-    def __call__(self, question: str) -> str:
+    async def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = self.agent.run(question)
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        # fixed_answer = self.agent.run(question)
+        response = await asyncio.to_thread(
+            self.agent.run,
+            question=question
+        )
+
+        # Ensure response is a string, fixing the integer error
+        response = str(response) 
+        if response is None: 
+            print(f"No answer found.")
+       
+        print(f"Agent returning fixed answer: {response}")
+        return response
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """