Update app.py

app.py

@@ -427,12 +427,16 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
                 run_button = gr.Button("✨ Edit Image", variant="primary", size="lg")
 
             with gr.Column():
-                output_image = gr.Image(
-                    label="Output",
-                    interactive=False,
-                    format="png",
-                    elem_classes=["contain-preview"],
-                )
+                with gr.Column():
+                    output_image = gr.Image(
+                        label="Output",
+                        interactive=False,
+                        format="png",
+                        elem_classes=["contain-preview"],
+                    )
+                    with gr.Row():
+                        use_as_input_btn = gr.Button("↻ Use as Input", size="sm", variant="secondary")
+                        clear_output_btn = gr.Button("✕ Clear", size="sm", variant="secondary")
 
                 with gr.Accordion("➕ Extra LoRAs (optional)", open=False):
                     with gr.Row():
@@ -493,6 +497,19 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
             outputs=[prompt],
         )
 
+    # ── Output image actions ─────────────────────────────────────────────────
+    use_as_input_btn.click(
+        fn=lambda img: img,
+        inputs=[output_image],
+        outputs=[input_image],
+    )
+
+    clear_output_btn.click(
+        fn=lambda: gr.update(value=None),
+        inputs=None,
+        outputs=[output_image],
+    )
+
 if __name__ == "__main__":
     demo.queue(max_size=30).launch(
         mcp_server=True, ssr_mode=False, show_error=True