Update app.py

app.py

@@ -18,7 +18,7 @@ def infer(
     num_inference_steps,
     user_info,
 ):
-    if user_info is None:
+    if not user_info or not user_info.get("username"):
         return "Please log in to generate images.", None
 
     # Load the pipeline only when this function is called
@@ -62,20 +62,18 @@ with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # Text-to-Image Gradio Template")
 
-        # Add the LoginButton
+        # Login Button inside the Blocks context
         login_button = gr.LoginButton()
+        login_button.activate()
         user_info = gr.State()
 
-        # Function to update user_info upon login
         def update_user_info(profile):
-            if profile is None:
-                return gr.Error("Please log in to access the application.")
             return profile
 
-        # Update user_info when login_button is clicked
-        login_button.click(update_user_info, inputs=None, outputs=user_info)
+        login_button.click(
+            update_user_info, inputs=None, outputs=user_info
+        )
 
-        # Main interface
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",