Update app.py

app.py

@@ -1,92 +1,82 @@
-import os
 import gradio as gr
-import requests
 import inspect
-import pandas as pd
-from smolagents.agents import ToolUseAgent
-from smolagents.tools import Tool
-import ast
-import math
+from smolagents import CodeAgent, load_tool, InferenceClientModel
+from smolagents.tool import tool
+import math, ast, os, requests, pandas as pd
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# ---- Tool 1: Calculator ----
-def safe_eval_math(expr: str) -> str:
+# --- Tool: Safe Math Evaluator ---
+@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.
+    """
     try:
-        # Only allow safe functions and math operators
-        allowed_names = {"sqrt": math.sqrt, "log": math.log, "pow": pow, "abs": abs, "round": round}
-        code = compile(expr, "<string>", "eval")
-        for node in ast.walk(code):
-            if isinstance(node, ast.Name) and node.id not in allowed_names:
-                raise ValueError(f"Use of '{node.id}' not allowed")
-        return str(eval(code, {"__builtins__": {}}, allowed_names))
+        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"
+        return str(eval(compile(tree, "<", "eval"), {"__builtins__": {}}, allowed))
     except Exception as e:
         return f"Error: {e}"
 
-calculator_tool = Tool(
-    name="calculator",
-    description="Evaluates basic math expressions like '2 + 2 * sqrt(9)'",
-    func=safe_eval_math,
-)
-
-# ---- Tool 2: Python Code Evaluator ----
-def run_python_code(code: str) -> str:
+# --- Tool: Python Code 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.
+    """
     try:
-        # Eval in restricted namespace
-        local_vars = {}
-        exec(code, {"__builtins__": {}}, local_vars)
-        return str(local_vars)
+        gl = {}
+        loc = {}
+        exec(code, gl, loc)
+        if loc:
+            return str(loc)
+        return ""
     except Exception as e:
-        return f"Code Error: {e}"
-
-python_tool = Tool(
-    name="python_eval",
-    description="Runs simple Python code blocks and returns the result. Use this for code-based reasoning.",
-    func=run_python_code,
-)
+        return f"Error: {e}"
 
-# ---- Tool 3: File Downloader (Optional, if GAIA task uses /files/{task_id}) ----
-def download_file(task_id: str) -> str:
-    base_url = os.getenv("DEFAULT_API_URL", "https://agents-course-unit4-scoring.hf.space")
-    url = f"{base_url}/files/{task_id}"
+# --- Tool: File Downloader for GAIA ---
+@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.
+    """
+    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()
-        file_path = f"/tmp/{task_id}"
-        with open(file_path, "wb") as f:
-            f.write(r.content)
-        return f"File downloaded to {file_path}"
+        r = requests.get(url); r.raise_for_status()
+        path = f"/tmp/{task_id}"
+        with open(path, "wb") as f: f.write(r.content)
+        return path
     except Exception as e:
-        return f"Failed to download file: {e}"
+        return f"Error: {e}"
 
-file_tool = Tool(
-    name="download_file",
-    description="Downloads file for a given GAIA task_id.",
-    func=download_file,
-)
+model = InferenceClientModel()
+agent = CodeAgent(tools=[calc, run_py, download], model=model)
 
-# ---- The Smart Tool Agent ----
 class ToolAgent:
     def __init__(self):
-        print("Initializing ToolAgent with smolagents...")
-        self.agent = ToolUseAgent(
-            tools=[calculator_tool, python_tool, file_tool],
-            system_prompt=(
-                "You are a helpful agent for solving GAIA benchmark questions. "
-                "Use tools when appropriate. Return only the exact answer."
-            )
-        )
+        self.agent = agent  # from above
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received: {question[:80]}")
-        try:
-            result = self.agent.run(question)
-            print(f"Raw Result: {result}")
-            return result.strip()
-        except Exception as e:
-            print(f"Error while answering: {e}")
-            return "Agent error"
+        result = self.agent.run(question)
+        return result.strip()
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """