Update app.py

app.py

@@ -28,19 +28,15 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 #         return fixed_answer
 
 
-def download_gaia_to_csv():
+def download_gaia_to_csv_and_return_path():
     try:
-        # Authenticate with HF token
         token = os.getenv("HF_TOKEN")
         if not token:
-            return "❌ HF_TOKEN not found. Please set it in your Space's secrets."
+            return None, "❌ HF_TOKEN not found."
 
         login(token)
-
-        # Load GAIA Level 1 validation set
         dataset = load_dataset("gaia-benchmark/GAIA", "2023_level1", split="validation", trust_remote_code=True)
 
-        # Save to CSV
         output_path = "gaia_level1_validation.csv"
         keys = dataset[0].keys()
 
@@ -50,9 +46,9 @@ def download_gaia_to_csv():
             for row in dataset:
                 writer.writerow(row)
 
-        return f"✅ GAIA Level 1 validation set saved to {output_path}"
+        return output_path, f"✅ File saved: {output_path} — ready for download."
     except Exception as e:
-        return f"❌ Error: {e}"
+        return None, f"❌ Error: {e}"
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
@@ -198,6 +194,8 @@ with gr.Blocks() as demo:
     test_eval_button = gr.Button("🔍 Evaluate on Local GAIA Test Set")
 
     download_db_button = gr.Button("=== Download DB ====")
+    file_output = gr.File(label="Download CSV")
+    status_output = gr.Textbox(label="Status")
 
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
@@ -215,9 +213,9 @@ with gr.Blocks() as demo:
         outputs=[status_output, results_table]
     )
 
-    download_db_button.click(
-        fn=download_gaia_to_csv,
-        outputs=[status_output]
+    download_button.click(
+        fn=download_gaia_to_csv_and_return_path,
+        outputs=[file_output, status_output]
     )