Update app.py

app.py

@@ -136,6 +136,42 @@ class BasicAgent:
             return error_msg
 
 
+def download_file(task_id, base_url, filename="MY_NAME", headers=None):
+    """
+    Download a file from the API endpoint and save it locally.
+
+    Args:
+        task_id (str): The task ID for the file to download
+        base_url (str): Base URL of the API (e.g., 'https://api.example.com')
+        filename (str): Local filename to save as (default: 'MY_NAME')
+        headers (dict): Optional headers for authentication/authorization
+
+    Returns:
+        bool: True if download successful, False otherwise
+    """
+    try:
+        # Construct the full URL
+        url = f"{base_url}/files/{task_id}"
+
+        # Make the GET request
+        response = requests.get(url, headers=headers, stream=True)
+
+        # Check if request was successful
+        response.raise_for_status()
+
+        # Save the file locally
+        with open(filename, 'wb') as file:
+            for chunk in response.iter_content(chunk_size=8192):
+                if chunk:
+                    file.write(chunk)
+
+        print(f"File downloaded successfully as '{filename}'")
+        return True
+    except Exception as e:
+        print(f"An exception occurred during file download: {e}")
+        return False
+
+
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -196,6 +232,9 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         task_id = item.get("task_id")
         question_text = item.get("question")
         file_name = item.get("file_name")
+        if file_name:
+            # download and save the file
+            download_file(task_id=task_id, base_url=api_url, filename=file_name)
         if not task_id or question_text is None:
             print(f"Skipping item with missing task_id or question: {item}")
             continue