Update app.py

app.py

@@ -1,80 +1,29 @@
+## app.py
 import os
 import gradio as gr
 import requests
 import pandas as pd
-from pathlib import Path
-import tempfile
-from smolagents import CodeAgent, OpenAIServerModel
-from smolagents import DuckDuckGoSearchTool, WikipediaSearchTool
-from tools import AnswerTool, SpeechToTextTool, ExcelToTextTool
+
+from tools import AnswerTool  # Solo AnswerTool
 
 # --- 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):
-        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,
-        )
+
+        self.tool = AnswerTool()
 
     def __call__(self, question: str, task_id: str = None) -> str:
-        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)
+
+        return self.tool.forward(question)
 
 
 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:
@@ -84,6 +33,7 @@ def run_and_submit_all(username):
     except Exception as e:
         return f"Error fetching questions: {e}", None
 
+
     agent = BasicAgent()
     results = []
     payload = []
@@ -93,7 +43,7 @@ def run_and_submit_all(username):
         if not (tid and text):
             continue
         try:
-            ans = agent(text, task_id=tid)
+            ans = agent(text)
         except Exception as e:
             ans = f"ERROR: {e}"
         results.append({"Task ID": tid, "Question": text, "Answer": ans})
@@ -102,6 +52,7 @@ 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",
@@ -129,11 +80,12 @@ 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, task_id=q.get("task_id"))
+        ans = BasicAgent()(question)
         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(
@@ -154,7 +106,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__":