Update app.py

app.py

@@ -8,20 +8,86 @@ import pandas as pd
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- Basic Agent Definition ---
-# ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-class BasicAgent:
+# ---- Tool 1: Calculator ----
+def safe_eval_math(expr: str) -> str:
+    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))
+    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:
+    try:
+        # Eval in restricted namespace
+        local_vars = {}
+        exec(code, {"__builtins__": {}}, local_vars)
+        return str(local_vars)
+    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,
+)
+
+# ---- 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}"
+    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}"
+    except Exception as e:
+        return f"Failed to download file: {e}"
+
+file_tool = Tool(
+    name="download_file",
+    description="Downloads file for a given GAIA task_id.",
+    func=download_file,
+)
+
+# ---- The Smart Tool Agent ----
+class ToolAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
+        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."
+            )
+        )
+
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
-        fixed_answer = "This is a default answer."
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        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"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
+    Fetches all questions, runs the ToolAgent on them, submits all answers,
     and displays the results.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
@@ -40,7 +106,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
-        agent = BasicAgent()
+        agent = ToolAgent()
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None