fixes

app.py

@@ -1,29 +1,81 @@
-## app.py
+app.py
 import os
 import gradio as gr
 import requests
 import pandas as pd
-
-from tools import AnswerTool  # Solo AnswerTool
+from pathlib import Path
+import tempfile
+from smolagents import CodeAgent, OpenAIServerModel
+from smolagents import DuckDuckGoSearchTool, WikipediaSearchTool
+from tools import AnswerTool, SpeechToTextTool, ExcelToTextTool
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+
+def download_file_if_any(base_api_url: str, task_id: str) -> str | None:
+    """
+    Try GET /files/{task_id}.
+    • On HTTP 200 → save to a temp dir and return local path.
+    • On 404 → return None.
+    """
+    url = f"{base_api_url}/files/{task_id}"
+    try:
+        resp = requests.get(url, timeout=30)
+        if resp.status_code == 404:
+            return None
+        resp.raise_for_status()
+    except requests.exceptions.HTTPError as e:
+        raise e
+
+    cdisp = resp.headers.get("content-disposition", "")
+    filename = task_id
+    if "filename=" in cdisp:
+        import re
+        m = re.search(r'filename="([^\"]+)"', cdisp)
+        if m:
+            filename = m.group(1)
+
+    tmp_dir = Path(tempfile.gettempdir()) / "gaia_files"
+    tmp_dir.mkdir(exist_ok=True)
+    file_path = tmp_dir / filename
+    with open(file_path, "wb") as f:
+        f.write(resp.content)
+    return str(file_path)
+
 class BasicAgent:
     def __init__(self):
-
-        self.tool = AnswerTool()
+        model = OpenAIServerModel(model_id="gpt-4o")
+        tools = [
+            SpeechToTextTool(),
+            ExcelToTextTool(),
+            DuckDuckGoSearchTool(),
+            WikipediaSearchTool(),
+            AnswerTool(),
+        ]
+        self.agent = CodeAgent(
+            model=model,
+            tools=tools,
+            add_base_tools=False,
+            additional_authorized_imports=["pandas", "openpyxl"],
+            max_steps=4,
+            verbosity_level=0,
+            planning_interval=1,
+        )
 
     def __call__(self, question: str, task_id: str = None) -> str:
-
-        return self.tool.forward(question)
+        prompt = question
+        if task_id:
+            file_path = download_file_if_any(DEFAULT_API_URL, task_id)
+            if file_path:
+                prompt += f"\n\n---\nA file was downloaded for this task and saved locally at:\n{file_path}\n---\n"
+        return self.agent.run(prompt)
 
 
 def run_and_submit_all(username):
     if not username:
         return "Please enter your Hugging Face username.", None
 
-
     try:
         resp = requests.get(f"{DEFAULT_API_URL}/questions", timeout=15)
         if resp.status_code == 429:
@@ -33,7 +85,6 @@ def run_and_submit_all(username):
     except Exception as e:
         return f"Error fetching questions: {e}", None
 
-
     agent = BasicAgent()
     results = []
     payload = []
@@ -43,7 +94,7 @@ def run_and_submit_all(username):
         if not (tid and text):
             continue
         try:
-            ans = agent(text)
+            ans = agent(text, task_id=tid)
         except Exception as e:
             ans = f"ERROR: {e}"
         results.append({"Task ID": tid, "Question": text, "Answer": ans})
@@ -52,7 +103,6 @@ def run_and_submit_all(username):
     if not payload:
         return "Agent returned no answers.", pd.DataFrame(results)
 
-
     submission = {
         "username": username,
         "agent_code": f"https://huggingface.co/spaces/{os.getenv('SPACE_ID')}/tree/main",
@@ -80,12 +130,11 @@ def test_random_question(username):
     try:
         q = requests.get(f"{DEFAULT_API_URL}/random-question", timeout=15).json()
         question = q.get("question", "")
-        ans = BasicAgent()(question)
+        ans = BasicAgent()(question, task_id=q.get("task_id"))
         return question, ans
     except Exception as e:
         return f"Error during test: {e}", ""
 
-# --- Gradio UI ---
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown(
@@ -106,7 +155,7 @@ with gr.Blocks() as demo:
     question_out = gr.Textbox(label="Random Question", lines=3, interactive=False)
     answer_out   = gr.Textbox(label="Agent Answer", lines=3, interactive=False)
 
-    run_btn.click(fn=run_and_submit_all, inputs=[username_input], outputs=[status_out, table_out])
+    run_btn.click(fn=run_and_submit_all,  inputs=[username_input], outputs=[status_out, table_out])
     test_btn.click(fn=test_random_question, inputs=[username_input], outputs=[question_out, answer_out])
 
 if __name__ == "__main__":