Update app.py

app.py

@@ -135,42 +135,34 @@ with gr.Blocks(css=css, title="ZenCtrl Inpainting") as demo:
             bg_img = gr.Image(label="Background", visible=True)
 
     # ---------- Example wrapper ---------------------------------
-    from pathlib import Path
-    def pil(path):        # open once, keep handles alive for the table
-        return Image.open(Path(path))
-    
-    def row(subj_path, bg_path, prompt):
-        return [
-            pil(subj_path),                         # goes to subj_img (Image widget)
-            {"image": pil(bg_path), "mask": None},  # → ref_img (sketch widget)
-            prompt                                  # → promptbox (Textbox)
-        ]
-    
-    examples = [
-        row("examples/sofa1.png",  "examples/sofa1_bg.png",  "add the sofa"),
-        row("examples/sofa2.png",  "examples/sofa2_bg.png",  "add this sofa"),
-        row("examples/chair1.png", "examples/chair1_bg.png", "add the chair"),
-        row("examples/console_table.png",
-            "examples/console_table_bg.png",
-            "Scandinavian console table against a gallery-style wall filled with abstract framed art"),
-        row("examples/office_chair.png", "examples/office_chair_bg.png", "office chair"),
-        row("examples/car.png",    "examples/car_bg.png",    "car on the road"),
-    ]
-    
-    # identity returns the filled-in widgets unchanged
-    def identity(a, b, c):
-        return a, b, c
-    
+    def _load_and_show(subj_path, bg_path, prompt_text):
+        out_path = subj_path.replace(".png", "_out.png")
+        return (
+            Image.open(subj_path),
+            Image.open(bg_path),
+            prompt_text,
+            Image.open(out_path)
+        ) 
+
+    # ---------- Examples ----------------------------------------
     gr.Examples(
-        examples          = examples,
-        inputs            = [subj_img, ref_img, promptbox],
-        outputs           = [subj_img, ref_img, promptbox],   # must match `identity`
-        fn                = identity,
-        cache_examples    = False,    # fine – we’re not pre-computing anything
-        label             = "Visual Presets – Subject · Background · Prompt",
-        examples_per_page = "all",
+        examples=[
+            ["examples/sofa1_1.png", "examples/sofa1_bg.png", "add the sofa"], #"examples/sofa1_out.png"],
+            ["examples/sofa2.png", "examples/sofa2_bg.png", "add this sofa"],# "examples/sofa2_out.png"],
+            ["examples/chair1.png", "examples/chair1_bg.png", "add the chair"], #"examples/chair1_out.png"],
+            ["examples/console_table.png", "examples/console_table_bg.png", "Scandinavian console table against a gallery-style wall filled with abstract framed art,"], #"examples/console_table_out.png"],
+            ["examples/office_chair.png", "examples/office_chair_bg.png", "office chair"], #"examples/office_chair_out.png"],
+            ["examples/car.png", "examples/car_bg.png", "car on the road"], #"examples/car_out.png"],
+        ],
+        inputs=[subj_img, ref_img, promptbox],   # three inputs
+        outputs=[subj_img, ref_img, promptbox, output_img],                    # 
+        fn=_load_and_show,
+        label="Visual Presets – Subject · Background · Prompt",
+        examples_per_page="all",
+        cache_examples=False,
     )
 
+
     # ---------- Buttons & interactions --------------------------
     # run_btn.click(
     #     process_image_and_text,