update

app.py

@@ -4,7 +4,7 @@ from huggingface_hub import HfApi, CommitOperationAdd, create_commit
 import pandas as pd
 from datetime import datetime
 import tempfile
-from typing import Optional
+from typing import Optional, Tuple
 
 # Initialize Hugging Face client
 hf_api = HfApi(token=os.getenv("HF_TOKEN"))
@@ -47,16 +47,15 @@ def commit_signup(username: str, email: str, current_data: pd.DataFrame) -> Opti
             os.unlink(tmp.name)
             return str(e)  # Return error message
 
-def join_waitlist(user: dict) -> str:
+def join_waitlist(
+    profile: gr.OAuthProfile | None,
+    oauth_token: gr.OAuthToken | None
+) -> str:
     """Add user to the SmolVLM2 iPhone waitlist with retry logic for concurrent updates"""
     
-    if not user or not user.get("username"):
-        gr.Warning("Please log in to Hugging Face first!")
+    if profile is None or oauth_token is None:
         return "Please log in with your Hugging Face account first!"
 
-    username = user["username"]
-    email = user.get("email", "")
-
     for attempt in range(MAX_RETRIES):
         try:
             # Get current waitlist state
@@ -68,11 +67,11 @@ def join_waitlist(user: dict) -> str:
                 current_data = pd.DataFrame(columns=['userid', 'email', 'joined_at'])
 
             # Check if user already registered
-            if username in current_data['userid'].values:
+            if profile.username in current_data['userid'].values:
                 return "You're already on the waitlist! We'll keep you updated."
 
             # Try to commit the update
-            error = commit_signup(username, email, current_data)
+            error = commit_signup(profile.username, profile.email or "", current_data)
             
             if error is None:  # Success
                 return "Thanks for joining the waitlist! We'll keep you updated on SmolVLM2 iPhone app and source code release."
@@ -89,6 +88,12 @@ def join_waitlist(user: dict) -> str:
             if attempt == MAX_RETRIES - 1:  # Last attempt
                 gr.Error(f"An error occurred: {str(e)}")
                 return "Sorry, something went wrong. Please try again later."
+            
+def show_welcome(profile: gr.OAuthProfile | None) -> str:
+    """Show welcome message based on login status"""
+    if profile is None:
+        return "Please sign in with your Hugging Face account to join the waitlist."
+    return f"Welcome {profile.name}! Click the button below to join the waitlist."
 
 # Create the interface
 with gr.Blocks(title="SmolVLM2 Waitlist") as demo:
@@ -102,16 +107,16 @@ with gr.Blocks(title="SmolVLM2 Waitlist") as demo:
     - First look at the source code when available
     """)
 
-    user = gr.State(value=None)
-        
+    gr.LoginButton()
+    welcome_msg = gr.Markdown()
     join_button = gr.Button("Join Waitlist", variant="primary")
-    output = gr.Markdown(visible=False)
-
-    join_button.click(
-        fn=join_waitlist,
-        inputs=user,
-        outputs=output,
-    )
+    status_msg = gr.Markdown()
+    
+    # Show welcome message on load and auth change
+    demo.load(show_welcome, None, welcome_msg)
+    
+    # Handle join button click
+    join_button.click(join_waitlist, None, status_msg)
 
 if __name__ == "__main__":
-    demo.queue().launch(auth=True, auth_message="Please login to join the waitlist")
+    demo.queue().launch()