fix sign in

app.py

@@ -212,7 +212,7 @@ def main():
         # Add sign-in button at the top
         with gr.Row():
             gr.Markdown("")  # Empty space for alignment
-            login_button = gr.LoginButton(value="🔑 Sign in for PRO benefits", size="sm")
+            login_button = gr.LoginButton(value="🔑 Sign in", size="sm")
             gr.Markdown("")  # Empty space for alignment
         
         pro_message = gr.Markdown(visible=False)
@@ -275,7 +275,7 @@ def main():
                     with gr.Group():
                         gr.Markdown("##  Dataset information")
                         # Dynamic user limit info - default to anonymous user
-                        user_limit_info = gr.Markdown(value="👤 **Anonymous User**: You can generate up to 100 samples per request.", visible=True)
+                        user_limit_info = gr.Markdown(value="👤 **Anonymous User**: You can generate up to 100 samples per request. Use the sign-in button above for PRO benefits (10,000 samples).", visible=True)
                         with gr.Row():
                             with gr.Column():
                                 input_dataset_name = gr.Textbox(label="Input Dataset Name", placeholder="e.g., simplescaling/s1K-1.1")
@@ -371,7 +371,7 @@ def main():
         
         def update_user_limits(oauth_token):
             if oauth_token is None:
-                return "👤 **Anonymous User**: You can generate up to 100 samples per request. [Sign in](javascript:void(0)) for PRO benefits (10,000 samples)."
+                return "👤 **Anonymous User**: You can generate up to 100 samples per request. Use the sign-in button above for PRO benefits (10,000 samples)."
             
             is_pro = verify_pro_status(oauth_token)
             if is_pro:
@@ -386,19 +386,25 @@ def main():
             # Update slider maximum based on user tier
             if oauth_token is None:
                 max_samples = MAX_SAMPLES_FREE
+                button_text = "🔑 Sign in for PRO benefits"
             else:
                 is_pro = verify_pro_status(oauth_token)
                 max_samples = MAX_SAMPLES_PRO if is_pro else MAX_SAMPLES_FREE
+                if is_pro:
+                    button_text = f"✨ Signed in as PRO ({profile.name if profile else 'User'})"
+                else:
+                    button_text = f"👤 Signed in as {profile.name if profile else 'User'}"
             
             slider_update = gr.update(maximum=max_samples)
+            button_update = gr.update(value=button_text)
             
-            return gr.update(visible=True), gr.update(visible=False), oauth_token, limit_msg, slider_update
+            return gr.update(visible=True), gr.update(visible=False), oauth_token, limit_msg, slider_update, button_update
 
 
         # Handle login state changes - LoginButton automatically handles auth state changes
         # The demo.load will handle both initial load and auth changes
         
-        demo.load(control_access, inputs=None, outputs=[main_interface, pro_message, current_oauth_token, user_limit_info, num_output_samples])
+        demo.load(control_access, inputs=None, outputs=[main_interface, pro_message, current_oauth_token, user_limit_info, num_output_samples, login_button])
         demo.queue(max_size=None, default_concurrency_limit=None).launch(show_error=True)
 
 if __name__ == "__main__":