Update app.py

app.py

@@ -7,6 +7,9 @@ import pandas as pd
 from evaluate_agent import evaluate_agent, test_questions
 from LG_agent import BasicAgent
 # from LG_agent import test_questions, evaluate_agent
+import json
+import csv
+from datasets import load_dataset
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -23,6 +26,29 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 #         print(f"Agent returning fixed answer: {fixed_answer}")
 #         return fixed_answer
 
+def download_gaia_to_csv():
+    try:
+        # Load dataset
+        dataset = load_dataset("gaia-benchmark/GAIA", split="dev", trust_remote_code=True)
+
+        # Define output file
+        output_path = "gaia_dev_data.csv"
+
+        # Extract all keys from the first entry for CSV headers
+        keys = dataset[0].keys()
+
+        # Write to CSV
+        with open(output_path, "w", newline="", encoding="utf-8") as f:
+            writer = csv.DictWriter(f, fieldnames=keys)
+            writer.writeheader()
+            for row in dataset:
+                writer.writerow(row)
+
+        return f"✅ GAIA dev dataset saved as CSV at {output_path}"
+    except Exception as e:
+        return f"❌ Error: {e}"
+
+
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
@@ -166,6 +192,8 @@ with gr.Blocks() as demo:
 
     test_eval_button = gr.Button("🔍 Evaluate on Local GAIA Test Set")
 
+    download_db_button = gr.Button("=== Download DB ====")
+
     run_button = gr.Button("Run Evaluation & Submit All Answers")
 
     status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
@@ -182,6 +210,11 @@ with gr.Blocks() as demo:
         outputs=[status_output, results_table]
     )
 
+    download_db_button.click(
+        fn=download_gaia_to_csv,
+        outputs=[status_output]
+    )
+
 
 if __name__ == "__main__":
     print("\n" + "-"*30 + " App Starting " + "-"*30)