update

app.py

@@ -48,11 +48,11 @@ def commit_signup(username: str, current_data: pd.DataFrame) -> Optional[str]:
 def join_waitlist(
     profile: gr.OAuthProfile | None,
     oauth_token: gr.OAuthToken | None
-) -> str:
+) -> gr.update:
     """Add user to the SmolVLM2 iPhone waitlist with retry logic for concurrent updates"""
     
     if profile is None or oauth_token is None:
-        return "Please log in with your Hugging Face account first!"
+        return gr.update(value="Please log in with your Hugging Face account first!")
 
     for attempt in range(MAX_RETRIES):
         try:
@@ -66,13 +66,19 @@ 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 gr.update(
+                    value="## 😎 You're already on the waitlist! We'll keep you updated.",
+                    visible=True
+                )
 
             # 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."
+                return gr.update(
+                    value="## 🎉 Thanks for joining the waitlist! We'll keep you updated on SmolVLM2 iPhone app.",
+                    visible=True
+                )
             
             # If we got a conflict error, retry
             if "Conflict" in str(error) and attempt < MAX_RETRIES - 1:
@@ -80,25 +86,37 @@ 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 gr.update(
+                value="## 🫠 Sorry, something went wrong. Please try again later.",
+                visible=True
+            )
 
         except Exception as 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."
+                return gr.update(
+                    value="## 🫠 Sorry, something went wrong. Please try again later.",
+                    visible=True
+                )
 
-def update_ui(profile: gr.OAuthProfile | None) -> Tuple[str, bool, bool]:
+def update_ui(profile: gr.OAuthProfile | None) -> Tuple[gr.update, gr.update, gr.update]:
     """Update UI elements based on login status"""
     if profile is None:
         return (
-            "## Please sign in with your Hugging Face account to join the waitlist.",  # welcome message
-            False,  # hide join button
-            False,  # hide status message
+            gr.update(
+                value="## Please sign in with your Hugging Face account to join the waitlist.",
+                visible=True
+            ),
+            gr.update(visible=False),
+            gr.update(visible=False)
         )
     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
+        gr.update(
+            value=f"## Welcome {profile.name} 👋 Click the button below 👇 to receive any updates related with the SmolVLM2 iPhone application",
+            visible=True
+        ),
+        gr.update(visible=True),
+        gr.update(visible=False)
     )
 
 # Create the interface