add oauth

app.py

@@ -3,13 +3,13 @@ from datasets import load_dataset, Dataset, concatenate_datasets
 from datetime import datetime
 import os
 from huggingface_hub import hf_hub_download, whoami
-from huggingface_hub import HfApi, ModelCard, whoami
 
 # Load your private Hugging Face dataset
 DATASET_NAME = "andito/technical_interview_internship_2025"
 TOKEN = os.environ.get("HF_TOKEN")
 EXERCISE_URL = os.environ.get("EXERCISE")
 dataset = load_dataset(DATASET_NAME, split="train")
+whitelist = os.environ.get("WHITELIST").split(",")
 
 # Function to fetch the exercise file if not already downloaded
 def fetch_exercise_file():
@@ -21,6 +21,8 @@ def log_to_hf_dataset(oauth_token: gr.OAuthToken | None):
         return "You have to be logged in.", "README.md"
 
     username = whoami(token=oauth_token.token)["name"]
+    if username not in whitelist:
+        return "You are not authorized to download the exercise.", "README.md"
 
     # Get current timestamp
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")