Update app.py

app.py

@@ -58,7 +58,6 @@ with gr.Blocks(theme=gr.themes.Citrus()) as demo:
     gr.HTML("<img src='https://huggingface.co/spaces/multimodalart/nano-banana/resolve/main/nano_banana_pros.png' style='display: block; margin: 0 auto; max-width: 500px' />")
     gr.HTML("<h3 style='text-align:center'>Hugging Face PRO users can use Google's Nano Banana (Gemini 2.5 Flash Image Preview) on this Space. <a href='https://huggingface.co/pro' target='_blank'>Subscribe to PRO</a></h3>")
 
-    login_button = gr.LoginButton()
     pro_message = gr.Markdown(visible=False)
     main_interface = gr.Column(visible=False)
 
@@ -88,6 +87,8 @@ with gr.Blocks(theme=gr.themes.Citrus()) as demo:
                 output_image = gr.Image(label="Output", interactive=False)
                 use_image_button = gr.Button("♻️ Use this Image for Next Edit")
 
+    login_button = gr.LoginButton()
+    
     # --- Event Handlers ---
     def unified_generator(
         prompt: str,