Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import inspect
 from smolagents import CodeAgent, load_tool, InferenceClientModel
-from smolagents.tool import tool
+from smolagents.tools import tool
 import math, ast, os, requests, pandas as pd
 # (Keep Constants as is)
 # --- Constants ---
@@ -11,68 +11,59 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 @tool
 def calc(expr: str) -> str:
     """
-    Evaluate an arithmetic expression safely.
-    Args:
-      expr (str): math expression like '2+3*sqrt(16)'.
-    Returns:
-      str: the numeric result or an error message.
+    Safely evaluate a math expression like '2 + sqrt(9)'.
     """
     try:
         allowed = {"sqrt": math.sqrt, "log": math.log, "pow": pow, "abs": abs, "round": round}
         tree = ast.parse(expr, mode="eval")
-        for node in ast.walk(tree):
-            if isinstance(node, ast.Name) and node.id not in allowed:
-                return f"Error: '{node.id}' not allowed"
+        for n in ast.walk(tree):
+            if isinstance(n, ast.Name) and n.id not in allowed:
+                return f"Error: '{n.id}' not allowed"
         return str(eval(compile(tree, "<", "eval"), {"__builtins__": {}}, allowed))
     except Exception as e:
         return f"Error: {e}"
 
-# --- Tool: Python Code Runner ---
+# --- Tool: Python Runner ---
 @tool
 def run_py(code: str) -> str:
     """
-    Execute snippet and return printed output or resulting variables.
-    Args:
-      code (str): Python code block.
-    Returns:
-      str: result or error text.
+    Execute Python code snippet; returns local variables dictionary or errors.
     """
     try:
-        gl = {}
-        loc = {}
+        gl, loc = {}, {}
         exec(code, gl, loc)
-        if loc:
-            return str(loc)
-        return ""
+        return str(loc)
     except Exception as e:
         return f"Error: {e}"
 
-# --- Tool: File Downloader for GAIA ---
+# --- Tool: File Downloader ---
 @tool
 def download(task_id: str) -> str:
     """
-    Download GAIA task file.
-    Args:
-      task_id (str): ID from GET /random-question.
-    Returns:
-      str: local file path or error.
+    Download a file for GAIA task_id; returns local file path.
     """
     base = os.getenv("DEFAULT_API_URL", "https://agents-course-unit4-scoring.hf.space")
     url = f"{base}/files/{task_id}"
     try:
-        r = requests.get(url); r.raise_for_status()
+        resp = requests.get(url); resp.raise_for_status()
         path = f"/tmp/{task_id}"
-        with open(path, "wb") as f: f.write(r.content)
+        with open(path, "wb") as f: f.write(resp.content)
         return path
     except Exception as e:
         return f"Error: {e}"
 
+# --- Initialize the agent ---
 model = InferenceClientModel()
-agent = CodeAgent(tools=[calc, run_py, download], model=model)
+agent = CodeAgent(
+    tools=[calc, run_py, download],
+    model=model,
+    add_base_tools=True,  # include default Python REPL and web search
+    additional_authorized_imports=["math", "requests", "pandas"]
+)
 
 class ToolAgent:
     def __init__(self):
-        self.agent = agent  # from above
+        self.agent = agent
 
     def __call__(self, question: str) -> str:
         result = self.agent.run(question)