Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, login
 
 from smolagents import CodeAgent
 from smolagents import OpenAIServerModel
@@ -19,37 +19,41 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
+# Global variables
+HF_DATASET_TOKEN = os.getenv("HF_DATASET_TOKEN")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
 # 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):
+    
+    global HF_DATASET_TOKEN
+    login(token = HF_DATASET_TOKEN)
+    
     response = None
     try:
         response = get_GAIA_dataset_validation_file(file_name)
@@ -62,10 +66,11 @@ class BasicAgent:
     def __init__(self):
         print("Starting the initialization of model.")
         
-        model_api_key = os.getenv("OPENAI_API_KEY")
+        global OPENAI_API_KEY
+        
         model = OpenAIServerModel(
             model_id="gpt-4o-mini-2024-07-18",
-            api_key = model_api_key 
+            api_key = OPENAI_API_KEY 
         )
         print("Core model has been initialized.")