Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool
+from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, tool
 from huggingface_hub import InferenceClient
 # (Keep Constants as is)
 # --- Constants ---
@@ -11,18 +11,40 @@ 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 = InferenceClient()
+        self.model = InferenceClientModel()
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()], 
-            model="deepseek-ai/DeepSeek-R1:novita",
+            tools=[DuckDuckGoSearchTool(), download_file], 
+            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"Answer the following question. Provide ONLY the final answer: {question}"
+        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:'."
+        )
         result = self.agent.run(prompt)
             
         # Clean up the output to ensure it's just the answer (GAIA requires exact match)