Update app.py

app.py

@@ -3,10 +3,25 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, AzureOpenAIModel
+from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, AzureOpenAIModel, tool
 import os
 from dotenv import load_dotenv
 
+@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."
 load_dotenv()
 # (Keep Constants as is)
 # --- Constants ---
@@ -24,13 +39,20 @@ class BasicAgent:
                         api_version=os.environ.get("OPENAI_API_VERSION")
                     )
         self.agent = CodeAgent(
-            tools=[DuckDuckGoSearchTool()], 
+            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)