Update app.py

app.py

@@ -146,20 +146,28 @@ with gr.Blocks(css=css, title="ZenCtrl Inpainting") as demo:
             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(
+    #from PIL import Image
+    def _run_example(subj, bg, prompt):
+        # 1️⃣  Ensure we have PIL.Image objects
+        if isinstance(subj, str):
+            subj = Image.open(subj)
+        if isinstance(bg, str):
+            bg = Image.open(bg)
+    
+        # 2️⃣  Wrap background for pipeline (no mask)
+        adapter_dict = {"image": bg, "mask": None}
+    
+        # 3️⃣  Call your real pipeline
+        gallery_out, _ = process_image_and_text(
             subj,
-            {"image": bg, "mask": None},   # wrap as dict for pipeline
+            adapter_dict,
             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
+    
+        # 4️⃣  Also show the bg in the visible sketch component so user can draw
+        return gallery_out, gr.update(value=bg)
 
     # ---------- Examples ----------------------------------------
     gr.Examples(