Switching to CodeAgent.

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents.agent import Agent
+from smolagents import CodeAgent
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -14,14 +14,13 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized with smolagents.")
-        self.agent = Agent()  # Initialize smolagents' Agent
+        self.agent = CodeAgent()  # Initialize CodeAgent
     
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        print(f"Agent received question: {question[:50]}...")
         
-        # Perform a web search using smolagents
         try:
-            search_result = self.agent.search(question)  # Use smolagents' search capability [[1]]
+            search_result = self.agent.run(question)  # CodeAgent typically uses .run()
             answer = f"Search result for '{question[:50]}...': {search_result}"
         except Exception as e:
             answer = f"Error during search: {e}"