Update app.py

app.py

@@ -3,48 +3,25 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, tool
-
+from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
-@tool
-def download_file(task_id: str) -> str:
-    """
-    Downloads a file associated with a specific GAIA task.
-    Args:
-        task_id: The ID of the task (e.g., '3f2b1...')
-    """
-    url = f"https://agents-course-unit4-scoring.hf.space/files/{task_id}"
-    response = requests.get(url)
-    if response.status_code == 200:
-        file_path = f"{task_id}_data"
-        with open(file_path, "wb") as f:
-            f.write(response.content)
-        return f"File downloaded to {file_path}. You should now use a tool to read it."
-    return "File not found."
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
         self.model = InferenceClientModel()
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool(), download_file], 
+            tools=[DuckDuckGoSearchTool()], 
             model=self.model,
             add_base_tools=True # Adds useful tools like image generation/visit web page
         )
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        prompt = (
-            f"Question: {question}\n\n"
-            "Instructions:\n"
-            "1. Use tools to find the answer if needed.\n"
-            "2. If a file is mentioned, use 'download_file' first.\n"
-            "3. Your final output must be ONLY the answer value (e.g., '1969', 'Paris', '4.2').\n"
-            "4. Do NOT include phrases like 'The answer is' or 'Final Answer:'."
-        )
+        prompt = f"Answer the following question. Provide ONLY the final answer: {question}"
         result = self.agent.run(prompt)
             
         # Clean up the output to ensure it's just the answer (GAIA requires exact match)