get file

.gitignore

@@ -0,0 +1,6 @@
+/.idea/.gitignore
+/.idea/Final_Assignment_Template.iml
+/.idea/misc.xml
+/.idea/modules.xml
+/.idea/vcs.xml
+/app2.py

app.py

@@ -4,21 +4,61 @@ import requests
 import inspect
 import pandas as pd
 from smolagents import OpenAIServerModel, DuckDuckGoSearchTool, PythonInterpreterTool, CodeAgent
+from pathlib import Path
+import tempfile
 
 # (Keep Constants as is)
 # --- 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.
+    • On other errors → raise so caller can log / handle.
+    """
+    url = f"{base_api_url}/files/{task_id}"
+    try:
+        resp = requests.get(url, timeout=30)
+        if resp.status_code == 404:
+            return None          # no file
+        resp.raise_for_status()   # raise on 4xx/5xx ≠ 404
+    except requests.exceptions.HTTPError as e:
+        # propagate non-404 errors (403, 500, …)
+        raise e
+
+    # ▸ Save bytes to a named file inside the system temp dir
+    #    Try to keep original extension from Content-Disposition if present.
+    cdisp = resp.headers.get("content-disposition", "")
+    filename = task_id                                 # default base name
+    if "filename=" in cdisp:
+        import re
+        m = re.search(r'filename="([^"]+)"', cdisp)
+        if m:
+            filename = m.group(1)                      # keep provided name
+
+    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)
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 class BasicAgent:
     def __init__(self):
-        print("BasicAgent initialized.")
-    def __call__(self, question: str) -> str:
         self.agent = CodeAgent(
-            model = OpenAIServerModel(model_id="gpt-4o"),
-            tools=[DuckDuckGoSearchTool(), PythonInterpreterTool()]
+            model=OpenAIServerModel(model_id="gpt-4o"),
+            tools=[DuckDuckGoSearchTool()],
+            add_base_tools=True,
         )
+
+        print("BasicAgent initialized.")
+
+    def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         fixed_answer = self.agent.run(question)
         print(f"Agent returning answer: {fixed_answer}")
@@ -81,11 +121,31 @@ 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")
+
+        # ----------fetch any attached file ----------
+        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}")
+
+        # ---------- Build the prompt sent to the agent ----------
+        if file_path:
+            q_for_agent = (
+                f"{question_text}\n\n"
+                f"---\n"
+                f"A file was downloaded for this task and saved locally at:\n"
+                f"{file_path}\n"
+                f"---\n\n"
+            )
+        else:
+            q_for_agent = question_text
+
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(q_for_agent)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: