Update app.py

app.py

@@ -10,12 +10,56 @@ from smolagents import Tool
 from smolagents import PythonInterpreterTool
 from smolagents import DuckDuckGoSearchTool
 
+from huggingface_hub import hf_hub_download
+
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+
+# GAIA Dataset file dowloading
+"""
+    Basic function to download a GAIA dataset validation file
+"""
+def get_GAIA_dataset_validation_file(file_name: str):
+    
+    response = hf_hub_download(
+        repo_id="gaia-benchmark/GAIA", 
+        filename= f"2023/validation/{file_name}", 
+        repo_type="dataset"
+    )
+    
+    return response
+
+"""
+    Basic function to download a GAIA dataset test file
+"""
+def get_GAIA_dataset_test_file(file_name: str):
+
+    response = hf_hub_download(
+        repo_id="gaia-benchmark/GAIA", 
+        filename= f"2023/test/{file_name}", 
+        repo_type="dataset"
+    )
+    
+    return response
+
+"""
+    Basic function to download a GAIA dataset file (validation attempted first)
+"""
+def get_GAIA_dataset_file(file_name: str):
+    response = None
+    try:
+        response = get_GAIA_dataset_validation_file(file_name)
+    except:
+        response = get_GAIA_dataset_test_file(file_name)
+            
+    return response
+
+
+
 class BasicAgent:
     def __init__(self):
         print("Starting the initialization of model.")
@@ -62,6 +106,12 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
     """
+    print("!!!!!!!!!!!!! HANDLING DATASET FILE")
+    response = get_GAIA_dataset_file("076c8171-9b3b-49b9-a477-244d2a532826.xlsx")
+    print(response)
+
+    return
+    
     # --- Determine HF Space Runtime URL and Repo URL ---
     space_id = os.getenv("SPACE_ID") # Get the SPACE_ID for sending link to the code