Update app.py

app.py

@@ -6,6 +6,11 @@ from datasets import load_dataset
 import io
 import ast 
 import re
+import json
+from dotenv import load_dotenv # Only needed if running locally
+
+# Load environment variables from a .env file if it exists locally
+load_dotenv()
 
 # --- 1. CONFIGURATION ---
 DEBUG_TESTING = False 
@@ -17,18 +22,33 @@ token = 'GbeqFrdNnENcHiJtUnTKcAbVkneXrlOkHb'
 HF_DATASET_REPO = "akaburia/policy-evaluations" 
 HF_TOKEN = HF + '_' + token
 
-APPROVED_EMAILS = {
-    "kaburiaaustin1@tahmo.org": "user1",
-    "E.Ramos@tudelft.nl" : "user2",
-    "eunice.pramos@gmail.com" : "user3",
-    "E.Abraham@tudelft.nl" : "user4",
-    "dene.abv@gmail.com" : "user5",
-    "rafatoufofana.abv@gmail.com" : "user6",
-    "annorfrank@tahmo.org" : "user7",
-    "n.marley@tahmo.org" : "user8",
-    "H.F.Hagenaars@tudelft.nl" : "user9",
-    "kaburiaaustin1@gmail.com" : "user10"
-}
+
+emails_env_string = os.environ.get("APPROVED_EMAILS", "{}")
+
+# 2. Parse the string into a Python dictionary
+try:
+    APPROVED_EMAILS = json.loads(emails_env_string)
+    
+    # Force all keys to lowercase right at startup
+    APPROVED_EMAILS = {k.lower(): v for k, v in APPROVED_EMAILS.items()}
+    
+except (json.JSONDecodeError, TypeError) as e:
+    print(f"⚠️ Error parsing APPROVED_EMAILS from environment variables: {e}")
+    # FALLBACK DICTIONARY: All emails here MUST be lowercase!
+    APPROVED_EMAILS = {
+        "kaburiaaustin1@tahmo.org": "user1",
+        "e.ramos@tudelft.nl" : "user2",
+        "eunice.pramos@gmail.com" : "user3",
+        "e.abraham@tudelft.nl" : "user4",
+        "dene.abv@gmail.com" : "user5",
+        "rafatoufofana.abv@gmail.com" : "user6",
+        "annorfrank@tahmo.org" : "user7",
+        "n.marley@tahmo.org" : "user8",
+        "h.f.hagenaars@tudelft.nl" : "user9",
+        "kaburiaaustin1@gmail.com" : "user10",
+        "faridakone@gmail.com": "user11"
+    }
+    
 
 DRILL_DOWN_MAP = {
     "coherent": ["+3 Indivisible", "+2 Reinforcing", "+1 Enabling"],
@@ -449,20 +469,23 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         }
     
     def login_and_load(email):
-        if email not in APPROVED_EMAILS:
-            return {progress_bar: gr.Markdown(f"<font color='red'>Error: Email '{email}' is not authorized.</font>"), login_box: gr.Group(visible=True)}
+        # Sanitise the input by stripping whitespace and converting to lowercase
+        clean_email = email.strip().lower()
         
-        user_tag = APPROVED_EMAILS[email] 
+        if clean_email not in APPROVED_EMAILS:
+            return {progress_bar: gr.Markdown(f"<font color='red'>Error: Email '{email}' is not authorized.</font>"), login_box: gr.Group(visible=True)}
+            
+        user_tag = APPROVED_EMAILS[clean_email] 
         
         if DEBUG_TESTING:
             full_df, pending_df, status = load_data_from_local()
             token_to_store = "debug_mode"
         else:
-            if not HF_TOKEN:
-                 return {progress_bar: gr.Markdown(f"<font color='red'>Error: HF_TOKEN is missing.</font>"), login_box: gr.Group(visible=True)}
+            if not HF_TOKEN: 
+                return {progress_bar: gr.Markdown(f"<font color='red'>Error: HF_TOKEN is missing.</font>"), login_box: gr.Group(visible=True)}
             full_df, pending_df, status = load_data_from_hub(HF_TOKEN)
             token_to_store = HF_TOKEN
-
+            
         if full_df is None: return {progress_bar: gr.Markdown(f"<font color='red'>{status}</font>"), login_box: gr.Group(visible=True)}
         
         first_item_updates = load_next_item(pending_df, 0)
@@ -473,6 +496,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         first_item_updates[user_tag_state] = user_tag 
         first_item_updates[login_box] = gr.Group(visible=False) 
         first_item_updates[annotation_box] = gr.Group(visible=True) 
+        
         return first_item_updates
 
     login_btn.click(