Update app.py

app.py

@@ -27,20 +27,21 @@ 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:
-        # Authenticate
+        # 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."
 
-        login(token)  # log in using the token
+        login(token)
 
-        # Load dataset
-        dataset = load_dataset("gaia-benchmark/GAIA", split="dev", trust_remote_code=True)
+        # Load only Level 1 questions from the dev set
+        dataset = load_dataset("gaia-benchmark/GAIA", "2023_level1", split="dev", trust_remote_code=True)
 
         # Save to CSV
-        output_path = "gaia_dev_data.csv"
+        output_path = "gaia_dev_level1.csv"
         keys = dataset[0].keys()
 
         with open(output_path, "w", newline="", encoding="utf-8") as f:
@@ -49,7 +50,7 @@ def download_gaia_to_csv():
             for row in dataset:
                 writer.writerow(row)
 
-        return f"✅ GAIA dev dataset saved as CSV at {output_path}"
+        return f"✅ GAIA Level 1 dev dataset saved to {output_path}"
     except Exception as e:
         return f"❌ Error: {e}"