fix: add function to recover task files from endpoint

agent.py

@@ -2,6 +2,7 @@ import os
 from pathlib import Path
 from typing import TypedDict, Annotated
 from uuid import uuid4
+import requests
 
 from langgraph.graph.message import add_messages
 from langchain_core.messages import AnyMessage, HumanMessage, AIMessage, SystemMessage
@@ -9,6 +10,7 @@ from langgraph.prebuilt import ToolNode
 from langgraph.graph import START, StateGraph
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.prebuilt import tools_condition
+
 # from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_core.rate_limiters import InMemoryRateLimiter
@@ -27,6 +29,11 @@ from tools import basic_tools
 # Google's chat interface
 RPM = os.environ.get("AGENT_MODEL_RPM", 9)
 TPM = os.environ.get("AGENT_MODEL_TPM", 250000)
+FILES_ENDPOINT = os.environ.get(
+    "FILES_ENDPOINT", "https://agents-course-unit4-scoring.hf.space"
+)
+TARGET_FILES_DIR = os.environ.get("TARGET_FILES_DIR", "/tmp/task_file")
+
 limiter = InMemoryRateLimiter(
     requests_per_second=(RPM / 60),
     check_every_n_seconds=(RPM / 70),
@@ -34,7 +41,7 @@ limiter = InMemoryRateLimiter(
 )
 chat = ChatGoogleGenerativeAI(
     # model="gemini-2.0-flash-lite",
-    model=os.environ.get("AGENT_MODEL","gemini-2.5-flash-preview-04-17"),
+    model=os.environ.get("AGENT_MODEL", "gemini-2.5-flash-preview-04-17"),
     temperature=os.environ.get("AGENT_MODEL_TEMP", 0.25),
     max_retries=os.environ.get("AGENT_MODEL_RETRIES", 2),
     verbose=True,
@@ -44,15 +51,18 @@ chat = ChatGoogleGenerativeAI(
 chat_with_tools = chat.bind_tools(basic_tools)
 memory = MemorySaver()
 
+
 # Generate the AgentState and Agent graph
 class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]
 
+
 def assistant(state: AgentState):
     return {
         "messages": [chat_with_tools.invoke(state["messages"])],
     }
 
+
 ## The graph
 builder = StateGraph(AgentState)
 
@@ -70,12 +80,10 @@ builder.add_conditional_edges(
 )
 builder.add_edge("tools", "assistant")
 
+
 def create_config():
-    return {
-        "configurable": {
-            "thread_id": str(uuid4())
-        }
-    }
+    return {"configurable": {"thread_id": str(uuid4())}}
+
 
 def get_system_prompt(prompt_file: Path = None):
     if prompt_file is None:
@@ -83,9 +91,33 @@ def get_system_prompt(prompt_file: Path = None):
     # load the system prompt from the file
     with prompt_file.open("r", encoding="utf-8") as f:
         system_prompt = f.read()
-    
+
     # System message
     return SystemMessage(content=system_prompt)
 
+
 def insert_file_into_query(query: str, file_name: str = ""):
     return f"""{query} - Adjacent files path > {file_name}"""
+
+
+def download_requested_file(
+    task_id: str,
+    question_file: str,
+    endpoint: str = FILES_ENDPOINT,
+    target_dir: str = TARGET_FILES_DIR,
+):
+    if question_file == "":
+        return
+
+    target_path = Path(target_dir)
+    if not target_path.exists():
+        target_path.mkdir(parents=True)
+    # Create path
+    file_path = target_path / question_file
+    # Download file
+    request = requests.get(
+        f"{endpoint}/files/{task_id}", timeout=30, allow_redirects=True
+    )
+    with file_path.open("wb") as file_:
+        file_.write(request.content)
+    return file_path

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 from pprint import pprint
-from agent import builder, HumanMessage, memory, create_config, get_system_prompt, insert_file_into_query
+from agent import builder, HumanMessage, memory, create_config, get_system_prompt, insert_file_into_query, download_requested_file
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -98,10 +98,19 @@ 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")
-        question_file = item.get("file_name", "")
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
+        
+        # Get task file if any
+        question_file = item.get("file_name", "")
+        try:
+            file_path = download_requested_file(task_id, question_file)
+        except Exception as err:
+            print(f"Could not download file ({question_file}) -> {err}")
+            file_path = None
+
+        question_file = str(file_path) if file_path is not None else ""
         try:
             submitted_answer = run_agent(question_text, question_file)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
@@ -109,6 +118,10 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         except Exception as e:
              print(f"Error running agent on task {task_id}: {e}")
              results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
+        
+        # Clean files
+        if file_path is not None:
+            os.remove(file_path)
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")