update: re-use output image

app.py

@@ -68,9 +68,9 @@ def preview_padding(image, side, padding_size):
 def outpaint_with_nano_banana(padded_image, side, custom_prompt):
     """Use Nano Banana to outpaint the padded image"""
     if padded_image is None:
-        return None, "No padded image to outpaint"
+        return None, "No padded image to outpaint", None
     if side is None:
-        return None, "No side information available"
+        return None, "No side information available", None
 
     if custom_prompt and custom_prompt.strip():
         prompt = custom_prompt
@@ -94,11 +94,35 @@ def outpaint_with_nano_banana(padded_image, side, custom_prompt):
             if getattr(part, "inline_data", None):
                 result_image = Image.open(io.BytesIO(part.inline_data.data))
                 w, h = result_image.size
-                return result_image, f"Output size: {w} x {h} pixels"
+                return result_image, f"Output size: {w} x {h} pixels", result_image
         w, h = padded_image.size
-        return padded_image, f"Output size: {w} x {h} pixels (no result from API)"
+        return padded_image, f"Output size: {w} x {h} pixels (no result from API)", padded_image
     except Exception as e:
-        return padded_image, f"Error: {str(e)}"
+        return padded_image, f"Error: {str(e)}", padded_image
+
+
+def reuse_outpainted_image(outpainted_image):
+    """Set the most recent outpainted image as the new input and reset padding state."""
+    if outpainted_image is None:
+        return (
+            None,
+            "No outpainted image available. Run an outpaint first.",
+            None,
+            "No padding applied",
+            None,
+            None,
+        )
+    if not isinstance(outpainted_image, Image.Image):
+        outpainted_image = Image.fromarray(outpainted_image)
+    w, h = outpainted_image.size
+    return (
+        outpainted_image,
+        f"Image size: {w} x {h} pixels",
+        None,
+        "No padding applied",
+        None,
+        None,
+    )
 
 
 def get_image_info(image):
@@ -121,6 +145,7 @@ with gr.Blocks(title="Nano Banana Outpainting") as demo:
     # State variables
     padded_state = gr.State(None)
     side_state = gr.State(None)
+    outpainted_state = gr.State(None)
 
     # ─────────────────────────────
     # ROW 1: The three image panes
@@ -182,6 +207,8 @@ with gr.Blocks(title="Nano Banana Outpainting") as demo:
             btn_outpaint = gr.Button(
                 "🎨 Outpaint with Nano Banana", size="lg", variant="primary"
             )
+            gr.Markdown("### Optional: Reuse the outpainted result")
+            btn_reuse = gr.Button("♻️ Reuse Outpainted Image")
 
         # STEP 2 PANEL (right)
         with gr.Column(scale=1, min_width=360):
@@ -225,7 +252,13 @@ with gr.Blocks(title="Nano Banana Outpainting") as demo:
     btn_outpaint.click(
         fn=outpaint_with_nano_banana,
         inputs=[padded_state, side_state, prompt_textbox],
-        outputs=[output_image, output_info],
+        outputs=[output_image, output_info, outpainted_state],
+    )
+
+    btn_reuse.click(
+        fn=reuse_outpainted_image,
+        inputs=outpainted_state,
+        outputs=[input_image, image_info, padded_preview, padded_info, padded_state, side_state],
     )
 
 if __name__ == "__main__":