update

app.py

@@ -10,40 +10,53 @@ from typing import Optional, Tuple
 hf_api = HfApi(token=os.getenv("HF_TOKEN"))
 DATASET_REPO = "HuggingFaceTB/smolvlm2-iphone-waitlist"
 MAX_RETRIES = 3
-
 def commit_signup(username: str, current_data: pd.DataFrame) -> Optional[str]:
     """Attempt to commit new signup atomically"""
-    
-    # Add new user with timestamp
-    new_row = pd.DataFrame([{
-        'userid': username,
-        'joined_at': datetime.utcnow().isoformat()
-    }])
-    updated_data = pd.concat([current_data, new_row], ignore_index=True)
-    
-    # Save to temp file
-    with tempfile.NamedTemporaryFile(mode='w', suffix='.csv', delete=False) as tmp:
-        updated_data.to_csv(tmp.name, index=False)
+    try:
+        # Verify current data has required columns
+        if 'userid' not in current_data.columns or 'joined_at' not in current_data.columns:
+            current_data = pd.DataFrame(columns=['userid', 'joined_at'])
         
-        # Create commit operation
-        operation = CommitOperationAdd(
-            path_in_repo="waitlist.csv",
-            path_or_fileobj=tmp.name
-        )
+        # Add new user with timestamp
+        new_row = pd.DataFrame([{
+            'userid': username,
+            'joined_at': datetime.utcnow().isoformat()
+        }])
         
-        try:
-            # Try to create commit
-            create_commit(
-                repo_id=DATASET_REPO,
-                repo_type="dataset",
-                operations=[operation],
-                commit_message=f"Add user {username} to waitlist"
+        # Ensure we're not duplicating any existing users
+        updated_data = pd.concat([
+            current_data[~current_data['userid'].isin([username])],  # Keep all except current user
+            new_row
+        ], ignore_index=True)
+        
+        # Sort by joined_at to maintain chronological order
+        updated_data = updated_data.sort_values('joined_at').reset_index(drop=True)
+        
+        # Save to temp file
+        with tempfile.NamedTemporaryFile(mode='w', suffix='.csv', delete=False) as tmp:
+            updated_data.to_csv(tmp.name, index=False)
+            
+            # Create commit operation
+            operation = CommitOperationAdd(
+                path_in_repo="waitlist.csv",
+                path_or_fileobj=tmp.name
             )
-            os.unlink(tmp.name)
-            return None  # Success
-        except Exception as e:
-            os.unlink(tmp.name)
-            return str(e)  # Return error message
+            
+            try:
+                # Try to create commit
+                create_commit(
+                    repo_id=DATASET_REPO,
+                    repo_type="dataset",
+                    operations=[operation],
+                    commit_message=f"Add user {username} to waitlist"
+                )
+                os.unlink(tmp.name)
+                return None  # Success
+            except Exception as e:
+                os.unlink(tmp.name)
+                return str(e)  # Return error message
+    except Exception as e:
+        return str(e)
 
 def join_waitlist(
     profile: gr.OAuthProfile | None,
@@ -61,11 +74,14 @@ def join_waitlist(
                 current_data = pd.read_csv(
                     f"https://huggingface.co/datasets/{DATASET_REPO}/raw/main/waitlist.csv"
                 )
+                # Ensure data frame has required columns
+                if 'userid' not in current_data.columns or 'joined_at' not in current_data.columns:
+                    current_data = pd.DataFrame(columns=['userid', 'joined_at'])
             except:
                 current_data = pd.DataFrame(columns=['userid', 'joined_at'])
 
             # Check if user already registered
-            if profile.username in current_data['userid'].values:
+            if not current_data.empty and profile.username in current_data['userid'].values:
                 return gr.update(
                     value="## 😎 You're already on the waitlist! We'll keep you updated.",
                     visible=True
@@ -75,6 +91,21 @@ def join_waitlist(
             error = commit_signup(profile.username, current_data)
             
             if error is None:  # Success
+                # Verify the commit worked by reading back the data
+                try:
+                    verification_data = pd.read_csv(
+                        f"https://huggingface.co/datasets/{DATASET_REPO}/raw/main/waitlist.csv"
+                    )
+                    if profile.username not in verification_data['userid'].values:
+                        if attempt < MAX_RETRIES - 1:
+                            continue  # Retry if verification failed
+                        else:
+                            raise Exception("Failed to verify signup")
+                except Exception as e:
+                    if attempt < MAX_RETRIES - 1:
+                        continue
+                    raise e
+
                 return gr.update(
                     value="## 🎉 Thanks for joining the waitlist! We'll keep you updated on SmolVLM2 iPhone app.",
                     visible=True
@@ -98,25 +129,18 @@ def join_waitlist(
                     value="## 🫠 Sorry, something went wrong. Please try again later.",
                     visible=True
                 )
-
-def update_ui(profile: gr.OAuthProfile | None) -> Tuple[gr.update, gr.update, gr.update]:
+def update_ui(profile: gr.OAuthProfile | None) -> Tuple[str, bool, bool]:
     """Update UI elements based on login status"""
     if profile is None:
         return (
-            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)
+            "## Please sign in with your Hugging Face account to join the waitlist.",  # welcome message
+            False,  # hide join button
+            False,  # hide status message
         )
     return (
-        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)
+        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