Update app.py

app.py

@@ -18,7 +18,7 @@ def infer(
     num_inference_steps,
     user_info,
 ):
-    if not user_info or not user_info.get("username"):
+    if not user_info or "username" not in user_info:
         return "Please log in to generate images.", None
 
     # Load the pipeline only when this function is called
@@ -68,6 +68,8 @@ with gr.Blocks(css=css) as demo:
         user_info = gr.State()
 
         def update_user_info(profile):
+            if profile is None:
+                return "No user information available"
             return profile
 
         login_button.click(