test

app.py

@@ -1,4 +1,3 @@
-# --- Application Logic ---
 import gradio as gr
 import os
 from huggingface_hub import HfApi, hf_hub_url
@@ -7,30 +6,92 @@ import json
 # IMPORTANT: This secret should be set in the Hugging Face Space settings, NOT hardcoded here.
 # We retrieve it using os.environ.get() which accesses the Space's environment variables/secrets.
 # Provide a default empty string for local testing where the secret won't exist.
-authorized_users_str = os.environ.get("AUTHORIZED_USER_IDS")
-AUTHORIZED_USER_IDS = set(authorized_users_str.split(','))
+authorized_users_str = os.environ.get("AUTHORIZED_USER_IDS", "")
+AUTHORIZED_USER_IDS = set(authorized_users_str.split(',') if authorized_users_str else [])
+
+# Add a fallback list for development or if environment variable isn't set
+if not AUTHORIZED_USER_IDS:
+    print("Warning: No authorized users found in environment variables.")
+    print("Using fallback authorized users list for development.")
+    # Fallback list - only used if AUTHORIZED_USER_IDS environment variable isn't set
+    AUTHORIZED_USER_IDS = set([
+        "authorized_user1",
+        "authorized_user2",
+        "your_username_here"
+    ])
+
+print(f"Current authorized users: {AUTHORIZED_USER_IDS}")
 
 # Function to determine if the current user has full access
 def check_user_access():
     """Check if the current user has full access based on their Hugging Face identity."""
     try:
-        # In Hugging Face Spaces, user information is available in environment variables
-        # SPACE_ID contains the space name (e.g., username/space-name)
-        # USER contains the logged-in username when using the space
-        username = os.environ.get("USER")
+        # Get username from Gradio's authentication context
+        # In Hugging Face Spaces, this will provide the logged-in username
+        username = None
+        
+        # First, check if we're running in a Hugging Face Space
+        if hasattr(gr, "me") and callable(gr.me):
+            try:
+                # gr.me() returns user information when authenticated in Spaces
+                user_info = gr.me()
+                if user_info:
+                    username = user_info.get("username")
+                    print(f"Detected username via gr.me(): {username}")
+            except Exception as e:
+                print(f"Error using gr.me(): {e}")
+        
+        # Fallback methods to get username
+        if not username:
+            # Try USER environment variable (as mentioned in your code)
+            username = os.environ.get("USER")
+            if username:
+                print(f"Using USER env var: {username}")
+            
+            # Try SPACE_AUTHOR (might contain space owner's username)
+            if not username:
+                space_author = os.environ.get("SPACE_AUTHOR")
+                if space_author:
+                    username = space_author
+                    print(f"Using SPACE_AUTHOR: {username}")
         
+        # For local development/testing, you can set a mock username
+        if not username and os.environ.get("DEVELOPMENT_MODE"):
+            username = os.environ.get("MOCK_USERNAME", "test_user")
+            print(f"Using mock username: {username}")
+        
+        # Debug output
+        print(f"Detected username: {username}")
+        print(f"Authorized users: {AUTHORIZED_USER_IDS}")
+            
         # If still no username, user is not logged in
         if not username:
+            print("No username detected, assuming not logged in")
             return False, None
             
         # Check if user is in the authorized list
-        has_full_access = username in AUTHORIZED_USER_ID
+        has_full_access = username in AUTHORIZED_USER_IDS
+        print(f"Full access granted: {has_full_access}")
         
         return has_full_access, username
     except Exception as e:
         print(f"Error checking user access: {e}")
         return False, None
 
+# Function to add the current user to the authorized list (admin function)
+def add_current_user_to_authorized():
+    _, username = check_user_access()
+    if not username:
+        return "Not logged in - can't add user", False
+    
+    global AUTHORIZED_USER_IDS
+    if username not in AUTHORIZED_USER_IDS:
+        AUTHORIZED_USER_IDS.add(username)
+        print(f"Added {username} to authorized users. New list: {AUTHORIZED_USER_IDS}")
+        return f"Added {username} to authorized users!", True
+    else:
+        return f"{username} is already an authorized user", True
+
 # Example data processing function (limited functionality)
 def process_data_limited(input_text):
     """Process data with limited functionality."""
@@ -63,6 +124,14 @@ def handle_submission(input_text):
         limited_user_display = f"Limited access for {username}" if username else "Limited access (not logged in)"
         return process_data_limited(input_text), limited_user_display, False
 
+# Function to get the current login status display
+def get_login_status():
+    has_full_access, username = check_user_access()
+    status_text = f"Full access for {username}" if has_full_access else \
+                  f"Limited access for {username}" if username else \
+                  "Limited access (not logged in)"
+    return status_text, has_full_access
+
 # Create the Gradio interface
 with gr.Blocks(title="Access Control Demo") as demo:
     gr.Markdown("# Hugging Face Spaces Access Control Demo")
@@ -75,24 +144,25 @@ with gr.Blocks(title="Access Control Demo") as demo:
     To test different access levels, log in with different Hugging Face accounts.
     """)
     
-    # Get initial access status
-    has_full_access, username = check_user_access()
-    
     # Display current user and access level
     with gr.Row():
         status_display = gr.Textbox(
-            value=f"Full access for {username}" if has_full_access else 
-                  f"Limited access for {username}" if username else 
-                  "Limited access (not logged in)",
             label="Current Access Status",
             interactive=False
         )
         access_indicator = gr.Checkbox(
-            value=has_full_access,
             label="Full Access",
             interactive=False
         )
     
+    # Add buttons for login management
+    with gr.Row():
+        refresh_btn = gr.Button("Refresh Login Status")
+        admin_btn = gr.Button("Add Me To Authorized Users")
+    
+    # Output for admin actions
+    admin_result = gr.Textbox(label="Admin Action Result", interactive=False)
+    
     # Input form
     with gr.Row():
         input_text = gr.Textbox(
@@ -117,6 +187,27 @@ with gr.Blocks(title="Access Control Demo") as demo:
         outputs=[output_text, status_display, access_indicator]
     )
     
+    # Connect the refresh button to update login status
+    refresh_btn.click(
+        fn=get_login_status,
+        inputs=[],
+        outputs=[status_display, access_indicator]
+    )
+    
+    # Connect the admin button to add the current user to authorized users
+    admin_btn.click(
+        fn=add_current_user_to_authorized,
+        inputs=[],
+        outputs=[admin_result, access_indicator]
+    )
+    
+    # Initialize the status display
+    demo.load(
+        fn=get_login_status,
+        inputs=[],
+        outputs=[status_display, access_indicator]
+    )
+    
     # Additional information section
     gr.Markdown("""
     ## How This Works