Enhance file handling in app.py by adding functionality to download associated files for questions, create a download directory, and log download status. Update agent invocation to include file paths for processing.

app.py

@@ -8,11 +8,15 @@ from agent import BasicAgent
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
+DOWNLOAD_DIR = "downloaded_task_files" # Directory to save downloaded files
+
+# --- Basic Agent Definition is now imported from agent.py ---
+# (Remove the inline BasicAgent class definition from here if it exists)
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
-    Fetches all questions, runs the BasicAgent on them, submits all answers,
-    and displays the results.
+    Fetches all questions, downloads associated files, runs the BasicAgent on them,
+    submits all answers, and displays the results.
     """
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code
@@ -28,6 +32,14 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
 
+    # Create download directory if it doesn't exist
+    try:
+        os.makedirs(DOWNLOAD_DIR, exist_ok=True)
+        print(f"Ensured download directory exists: {DOWNLOAD_DIR}")
+    except OSError as e:
+        print(f"Error creating download directory {DOWNLOAD_DIR}: {e}")
+        return f"Error creating download directory: {e}", None
+
     # 1. Instantiate Agent ( modify this part to create your agent)
     try:
         agent = BasicAgent()
@@ -66,16 +78,60 @@ 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")
+        file_name = item.get("file_name")
+        local_file_path = None
+        download_status = "No file specified"
+
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue
+
+        if file_name:
+            # Construct download URL and local file path
+            # Assuming files are served from a /files/ endpoint on the API server
+            download_url = f"{api_url}/files/{file_name}"
+            local_file_path = os.path.join(DOWNLOAD_DIR, file_name)
+            print(f"Attempting to download file for task {task_id}: {file_name} from {download_url}")
+            try:
+                file_response = requests.get(download_url, stream=True, timeout=30)
+                file_response.raise_for_status()
+                with open(local_file_path, 'wb') as f:
+                    for chunk in file_response.iter_content(chunk_size=8192):
+                        f.write(chunk)
+                download_status = f"Successfully downloaded to {local_file_path}"
+                print(download_status)
+            except requests.exceptions.RequestException as e:
+                download_status = f"Failed to download {file_name}: {e}"
+                print(download_status)
+                local_file_path = None # Ensure agent doesn't get a path to a non-existent/failed file
+            except Exception as e:
+                download_status = f"An unexpected error occurred downloading {file_name}: {e}"
+                print(download_status)
+                local_file_path = None
+
+
         try:
-            submitted_answer = agent(question_text)
+            submitted_answer = agent(question_text, file_path=local_file_path)
             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})
+            results_log.append({
+                "Task ID": task_id,
+                "Question": question_text,
+                "File Name": file_name if file_name else "N/A",
+                "Download Status": download_status,
+                "Local File Path": local_file_path if local_file_path else "N/A",
+                "Submitted Answer": submitted_answer
+            })
         except Exception as e:
+             error_message = f"AGENT ERROR: {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}"})
+             results_log.append({
+                "Task ID": task_id,
+                "Question": question_text,
+                "File Name": file_name if file_name else "N/A",
+                "Download Status": download_status,
+                "Local File Path": local_file_path if local_file_path else "N/A",
+                "Submitted Answer": error_message
+            })
 
     if not answers_payload:
         print("Agent did not produce any answers to submit.")