Update app.py

app.py

@@ -136,12 +136,13 @@ with gr.Blocks(css=css, title="ZenCtrl Inpainting") as demo:
 
     # ---------- Example wrapper ---------------------------------
     def _load_and_show(subj_path, bg_path, prompt_text):
-        """
-        Return pre-rendered result corresponding to the subject image.
-        Converts e.g. examples/sofa1.png → examples/sofa1_out.png
-        """
         out_path = subj_path.replace(".png", "_out.png")
-        return (Image.open(out_path), ) 
+        return (
+            Image.open(subj_path),
+            Image.open(bg_path),
+            prompt_text,
+            Image.open(out_path)
+        ) 
 
     # ---------- Examples ----------------------------------------
     gr.Examples(
@@ -154,7 +155,7 @@ with gr.Blocks(css=css, title="ZenCtrl Inpainting") as demo:
             ["examples/car.png", "examples/car_bg.png", "car on the road"], #"examples/car_out.png"],
         ],
         inputs=[subj_img, ref_img, promptbox],   # three inputs
-        outputs=[output_img],                    # 
+        outputs=[subj_img, ref_img, promptbox, output_img],                    # 
         fn=_load_and_show,
         label="Visual Presets – Subject · Background · Prompt",
         examples_per_page="all",