Update app.py

app.py

@@ -138,10 +138,11 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     
     login_button = gr.LoginButton()
     
-    # --- Event Handlers (SIMPLIFIED) ---
-    generate_button.click(
-        unified_image_generator,
-        inputs=[prompt_input, image_input_gallery], # Inputs are now just the prompt and the single gallery
+    # --- Event Handlers ---
+    gr.on(
+        triggers=[generate_button.click, prompt_input.submit],
+        fn=unified_image_generator,
+        inputs=[prompt_input, image_input_gallery],
         outputs=[output_image],
     )