Upload app.py

app.py

@@ -11,6 +11,8 @@ from src.agent import Agent
 from src.tools import (
     multiply, add, subtract, divide, modulus, wiki_search
 )
+from src.utils import download_file
+
 
 tools = [
     multiply, add, subtract, divide, modulus, wiki_search
@@ -33,6 +35,7 @@ langfuse = get_client()
 # Initialize the Langfuse handler
 langfuse_handler = CallbackHandler()
 
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -57,6 +60,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
+    download_url = f"{api_url}/files/"
 
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
@@ -103,11 +107,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", None)
+        # If question_file is not None, download it in /data folder
+        if question_file is not None:
+            download_file(
+                url=download_url+task_id,
+                local_filepath=os.path.join("data", question_file)
+            )
+
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            reasoning, submitted_answer = agent(question_text)
+            reasoning, submitted_answer = agent(question_text, question_file)
             answers_payload.append(
                 {"task_id": task_id, "submitted_answer": submitted_answer}
             )