Update app.py

app.py

@@ -107,20 +107,31 @@ header_html = """
 """
 
 # ────────── Gradio UI ──────────
-with gr.Blocks(css=css, title="ZenCtrl Playground") as demo:
+with gr.Blocks(css=css, title="ZenCtrl Inpainting") as demo:
     raw_state = gr.State()
 
     gr.HTML(header_html)
-    gr.Markdown("""
-**Generate context-aware images of your subject with ZenCtrl’s inpainting playground.**  
-Upload a subject + optional mask, write a prompt, and hit **Generate**.  
-Open *Advanced Settings* to fetch an AI-generated background.
-""")
+    gr.Markdown(
+        "**Generate context-aware images of your subject with ZenCtrl’s inpainting playground.**  \n"
+        "Upload a subject + optional mask, write a prompt, and hit **Generate**.  \n"
+        "Open *Advanced Settings* to fetch an AI-generated background."
+    )
 
     with gr.Row():
         with gr.Column(scale=2, elem_id="col-container"):
             subj_img = gr.Image(type="pil", label="Subject image")
-            ref_img  = gr.Image(type="pil", label="Background / Mask image", tool="sketch", brush_color="#00FF00", sources=["upload", "clipboard"])
+
+            # ⚑ Visible sketch component (what the user draws on)
+            ref_img  = gr.Image(
+                type="pil", label="Background / Mask image",
+                tool="sketch", brush_color="#00FF00"
+            )
+
+            # ⚑ Hidden plain-image component used ONLY for examples
+            ref_img_ex = gr.Image(
+                type="pil", visible=False
+            )
+
             use_detect_ck = gr.Checkbox(False, label="Detect with Florence-SAM")
             detect_box    = gr.Textbox(label="Detection prompt", value="person, chair", visible=False)
             promptbox     = gr.Textbox(label="Generation prompt", value="furniture", lines=2)
@@ -134,29 +145,48 @@ Open *Advanced Settings* to fetch an AI-generated background.
             gallery = gr.Gallery(columns=[1], rows=[1], object_fit="contain", height="auto")
             bg_img  = gr.Image(label="Background", visible=False)
 
+    # ---------- Example wrapper ---------------------------------
+    from PIL import Image
+    def _run_example(subj_path, bg_path, prompt):
+        """Convert the plain bg image to the dict format and run generation."""
+        subj = Image.open(subj_path)
+        bg   = Image.open(bg_path)
+        outputs, _ = process_image_and_text(
+            subj,
+            {"image": bg, "mask": None},   # wrap as dict for pipeline
+            prompt,
+            use_detect=False,
+            detect_prompt=""
+        )
+        # Return generation + also show bg in the sketch component so user can draw
+        return outputs[0], gr.update(value=bg)   # gallery, ref_img update
+
+    # ---------- Examples ----------------------------------------
     gr.Examples(
         examples=[
-            ["./examples/subject1.png", "examples/subject1.png", "Make the toy sit on a marble table", "./examples/subject1.png"],
-            ["./examples/subject1.png", "examples/subject1.png", "Turn the flowers into sunflowers", "./examples/subject1.png"],
-            ["./examples/subject1.png", "examples/subject1.png", "Make this monster ride a skateboard on the beach", "./examples/subject1.png"],
-            ["./examples/subject1.png", "examples/subject1.png", "Make this cat happy", "./examples/subject1.png"],
+            ["examples/subject1.png", "examples/bg1.png", "Make the toy sit on a marble table"],
+            ["examples/subject2.png", "examples/bg2.png", "Turn the flowers into sunflowers"],
+            ["examples/subject3.png", "examples/bg3.png", "Make this monster ride a skateboard on the beach"],
+            ["examples/subject4.png", "examples/bg4.png", "Make this cat happy"],
         ],
-        inputs=[subj_img, ref_img, promptbox],
-        outputs=[gallery],
-        fn=process_image_and_text,
+        inputs=[subj_img, ref_img_ex, promptbox],   # <-- uses the hidden plain image
+        outputs=[gallery, ref_img],                 # update gallery + visible sketch img
+        fn=_run_example,
         examples_per_page="all",
-        label="Presets (Input · Background · Prompt · Output)",
-        cache_examples="lazy"
+        label="Presets (Input · Background · Prompt)",
+        cache_examples="lazy",
     )
 
+    # ---------- Buttons & interactions --------------------------
     run_btn.click(
         process_image_and_text,
         inputs=[subj_img, ref_img, promptbox, use_detect_ck, detect_box],
         outputs=[gallery, raw_state]
     )
-
     bg_btn.click(_gen_bg, inputs=[bgprompt], outputs=[bg_img])
-    use_detect_ck.change(lambda v: gr.update(visible=v), inputs=use_detect_ck, outputs=detect_box)
+    use_detect_ck.change(
+        lambda v: gr.update(visible=v), inputs=use_detect_ck, outputs=detect_box
+    )
 
-# ────────── Launch ──────────
-demo.launch(show_api=False, share=True)
+# ---------------- Launch ---------------------------------------
+demo.launch(show_api=False, share=True)