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"))
@@ -66,13 +66,13 @@ def join_waitlist(
 
             # Check if user already registered
             if profile.username in current_data['userid'].values:
-                return "You're already on the waitlist! We'll keep you updated."
+                return "## 😎 You're already on the waitlist! We'll keep you updated."
 
             # Try to commit the update
             error = commit_signup(profile.username, 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."
+                return "## 🎉 Thanks for joining the waitlist! We'll keep you updated on SmolVLM2 iPhone app."
             
             # If we got a conflict error, retry
             if "Conflict" in str(error) and attempt < MAX_RETRIES - 1:
@@ -80,39 +80,45 @@ def join_waitlist(
                 
             # Other error or last attempt
             gr.Error(f"An error occurred: {str(error)}")
-            return "Sorry, something went wrong. Please try again later."
+            return "## 🫠 Sorry, something went wrong. Please try again later."
 
         except Exception as e:
-            print(str(e))
             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"""
+                return "## 🫠 Sorry, something went wrong. Please try again later."
+
+def update_ui(profile: gr.OAuthProfile | None) -> Tuple[str, bool, bool]:
+    """Update UI elements 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."
+        return (
+            "## Please sign in with your Hugging Face account to join the waitlist.",  # welcome message
+            False,  # hide join button
+            False,  # hide status message
+        )
+    return (
+        f"## Welcome {profile.name} 👋 Click the button below 👇 to receive any updates related with the SmolVLM2 iPhone application",  # welcome message
+        True,  # show join button
+        True,  # show status message
+    )
 
 # Create the interface
 with gr.Blocks(title="SmolVLM2 Waitlist") as demo:
     gr.Markdown("""
     # Join SmolVLM2 iPhone Waitlist 📱
-    Sign up to be notified when the SmolVLM2 iPhone app and source code are released.
-    
-    By joining the waitlist, you'll get:
-    - Early access to the SmolVLM2 iPhone app
-    - Updates on development progress
-    - First look at the source code when available
+    ## Please log-in using your Hugging Face Id to continue
     """)
 
     gr.LoginButton()
-    welcome_msg = gr.Markdown()
-    join_button = gr.Button("Join Waitlist", variant="primary")
-    status_msg = gr.Markdown()
+    welcome_msg = gr.Markdown(visible=True)
+    join_button = gr.Button("Join Waitlist", variant="primary", visible=False)
+    status_msg = gr.Markdown(visible=False)
     
-    # Show welcome message on load and auth change
-    demo.load(show_welcome, None, welcome_msg)
+    # Update UI elements on load and auth change
+    demo.load(
+        fn=update_ui,
+        inputs=None,
+        outputs=[welcome_msg, join_button, status_msg]
+    )
     
     # Handle join button click
     join_button.click(join_waitlist, None, status_msg)