Update app.py

app.py

@@ -3,7 +3,7 @@ import time
 import gradio as gr
 import requests
 import pandas as pd
-from smolagents import ToolCallingAgent, OpenAIServerModel, PythonInterpreterTool, Tool 
+from smolagents import CodeAgent, OpenAIServerModel, PythonInterpreterTool, Tool 
 from smolagents import FinalAnswerTool
 
 # --- Constants ---
@@ -136,7 +136,7 @@ class BasicAgent:
             api_base="https://api.groq.com/openai/v1",
             api_key=os.getenv("GROQ_API_KEY")
         )
-        self.agent = ToolCallingAgent(
+        self.agent = CodeAgent(  # <-- back to CodeAgent
             model=model,
             tools=[
                 WebSearchTool(),
@@ -152,26 +152,24 @@ class BasicAgent:
     def __call__(self, question: str, task_id: str = "") -> str:
         try:
             prompt = f"""Answer the following question accurately.
-    Return ONLY the final answer with no explanation, no punctuation, no extra words.
-    - If the answer is a number, return just the number.
-    - If the answer is a name, return just the name.
-    - If the answer is a list, return comma separated values in alphabetical order.
-    - If the question asks about a YouTube video, use the youtube_transcript tool.
-    - If the question mentions Wikipedia, use the wikipedia_search tool.
-    - If the question references an attached file, use download_file with the task_id below.
-    Task ID: {task_id}
-    
-    Question: {question}"""
+Return ONLY the final answer with no explanation, no punctuation, no extra words.
+- If the answer is a number, return just the number.
+- If the answer is a name, return just the name.
+- If the answer is a list, return comma separated values in alphabetical order.
+- If the question asks about a YouTube video, use the youtube_transcript tool.
+- If the question mentions Wikipedia, use the wikipedia_search tool.
+- If the question references an attached file, use download_file with the task_id below.
+Task ID: {task_id}
+
+Question: {question}"""
             result = self.agent.run(prompt)
-            # handle case where result is a list of content blocks
             if isinstance(result, list):
                 for block in result:
                     if isinstance(block, dict) and block.get('type') == 'text':
                         return block['text'].strip()
             return str(result).strip()
         except Exception as e:
-            err = str(e)
-            print(f"Agent error: {err}")
+            print(f"Agent error: {e}")
             return "I don't know"