Update app.py

app.py

@@ -62,9 +62,9 @@ def run_multi_image_logic(prompt: str, images: List[str]) -> List[str]:
     return output_images
 
 # --- Gradio App UI ---
-with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
-    gr.Markdown("# Nano Banana for PROs")
-    gr.Markdown("Hugging Face PRO users can use Google's Nano Banana (Gemini 2.5 Flash Image Preview) on this Space. [Subscribe to PRO](https://huggingface.co/pro)")
+with gr.Blocks(theme=gr.themes.Citrus()) as demo:
+    gr.Markdown("# Nano Banana Image Generation")
+    gr.Markdown("Generate or edit images with FAL. **Sign in with Hugging Face to begin.**")
 
     login_button = gr.LoginButton()
     pro_message = gr.Markdown(visible=False)
@@ -96,7 +96,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
             with gr.Column(scale=1):
                 output_gallery = gr.Gallery(label="Output")
                 selected_output_image_state = gr.State()
-                use_image_button = gr.Button("♻️ Use Generated Image for Next Edit")
+                use_image_button = gr.Button("♻️ Use Selected Image for Next Edit")
 
     # --- Event Handlers ---
 
@@ -105,7 +105,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
         single_image: Optional[str],
         multi_images: Optional[List[str]],
         active_tab: str,
-        oauth_token: Optional[gr.OAuthToken] = None,
+        oauth_token: Optional[gr.OAuthToken] = None, # Gradio injects this automatically
     ) -> List[str]:
         if not verify_pro_status(oauth_token):
             raise gr.Error("Access Denied. This service is for PRO users only.")
@@ -116,28 +116,22 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
 
     single_tab.select(lambda: "single", None, active_tab_state)
     multi_tab.select(lambda: "multiple", None, active_tab_state)
-    
+
     generate_button.click(
         unified_generator,
-        inputs=[prompt_input, image_input, gallery_input, active_tab_state, login_button],
+        # CORRECTED: login_button is removed. Gradio will inject the token via the type hint.
+        inputs=[prompt_input, image_input, gallery_input, active_tab_state],
         outputs=[output_gallery],
     )
 
-    # New handlers for the continuous editing loop
+    # Handlers for the continuous editing loop
     def store_selected_image(evt: gr.SelectData):
-        """When an image is selected in the output gallery, store its path in state."""
-        return evt.value['image']
+        return evt.value['image']['path']
 
     def reuse_output_image(selected_image_path):
-        """
-        Takes the path from state and sends it to the single image input.
-        Also forces the UI to switch to the "Single Image" tab.
-        """
         if not selected_image_path:
             gr.Warning("Please select an image from the output gallery first!")
             return None, gr.update()
-        # Output 1: The image path for the gr.Image component
-        # Output 2: The ID of the tab to select for the gr.Tabs component
         return selected_image_path, "single"
 
     output_gallery.select(store_selected_image, None, selected_output_image_state)
@@ -163,7 +157,7 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
                 "To unlock this and many other benefits, please consider upgrading your account.\n\n"
                 "### [**Become a PRO Member Today!**](https://huggingface.co/pro)"
             )
-            return gr.update(visible=False), gr.update(visible=True, value=message)
+            return gr.update(visible=False), gr.update(visible=true, value=message)
 
     demo.load(control_access, inputs=None, outputs=[main_interface, pro_message])