Update app.py

app.py

@@ -1,16 +1,16 @@
 import os
 import gradio as gr
 import requests
-from smolagents import HfApiModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool
-import pandas as pd
-import tempfile
+from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, CodeAgent, WikipediaSearchTool
 from pathlib import Path
+import tempfile
+import pandas as pd
 import re
 
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
-# --- File Handling ---
+# --- File Download Helper ---
 def download_file_if_any(base_api_url: str, task_id: str) -> str | None:
     url = f"{base_api_url}/files/{task_id}"
     try:
@@ -20,12 +20,14 @@ def download_file_if_any(base_api_url: str, task_id: str) -> str | 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:
         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
@@ -33,31 +35,26 @@ def download_file_if_any(base_api_url: str, task_id: str) -> str | None:
         f.write(resp.content)
     return str(file_path)
 
-# --- Basic Agent Definition ---
+# --- Basic Agent ---
 class BasicAgent:
     def __init__(self):
-        model=OpenAIServerModel(model_id="gpt-4o"),
         self.agent = CodeAgent(
-            model=model,
+            model=OpenAIServerModel(model_id="gpt-4o"),
             tools=[DuckDuckGoSearchTool(), WikipediaSearchTool()],
             add_base_tools=True,
-            additional_authorized_imports=['pandas']
+            additional_authorized_imports=[]
         )
         print("BasicAgent initialized.")
 
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        try:
-            fixed_answer = self.agent.run(question)
-            print(f"Agent returning answer: {fixed_answer}")
-            return fixed_answer
-        except Exception as e:
-            print(f"Error during inference: {e}")
-            return f"AGENT ERROR: {e}"
+        fixed_answer = self.agent.run(question)
+        print(f"Agent returning answer: {fixed_answer}")
+        return fixed_answer
 
-# --- Run and Submit ---
+# --- Evaluation Logic ---
 def run_and_submit_all(profile: gr.OAuthProfile | None):
-    space_id = "YajieXu/Final_Assignment_Template"
+    space_id = "l3xv/Final_Assignment_Template"
     if profile:
         username = f"{profile.username}"
     else:
@@ -86,16 +83,20 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     for item in questions_data:
         task_id = item.get("task_id")
         question_text = item.get("question")
+
         try:
             file_path = download_file_if_any(api_url, task_id)
         except Exception as e:
             file_path = None
+            print(f"[file fetch error] {task_id}: {e}")
 
-        q_for_agent = f"{question_text}\n\n---\nFile: {file_path}\n---\n" if file_path else question_text
+        q_for_agent = (
+            f"{question_text}\n\n---\nA file was downloaded for this task and saved locally at:\n{file_path}\n---\n\n"
+            if file_path else question_text
+        )
 
         if not task_id or question_text is None:
             continue
-
         try:
             submitted_answer = agent(q_for_agent)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
@@ -123,7 +124,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         return f"Submission Failed: {e}", pd.DataFrame(results_log)
 
-# --- UI ---
+# --- Gradio UI ---
 with gr.Blocks() as demo:
     gr.Markdown("# Basic Agent Evaluation Runner")
     gr.Markdown("""