Update

app.py

@@ -27,12 +27,24 @@ def encode_image_as_base64(image: Image.Image) -> str:
     return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
 
-def make_example(image_path: Path) -> EditorValue:
+def make_example(image_path: Path, mask_path: Path | None) -> EditorValue:
     background_image = Image.open(image_path)
     background_image = background_image.convert("RGB")
     background = np.array(background_image)
 
-    mask = np.zeros_like(background)
+    if mask_path:
+        mask_image = Image.open(mask_path)
+        mask_image = mask_image.convert("RGB")
+        mask = np.array(mask_image)
+        mask = mask[:, :, 0]
+        mask = np.where(mask == 255, 0, 255)  # noqa: PLR2004
+    else:
+        mask = np.zeros_like(background)
+        mask = mask[:, :, 0]
+
+    if background.shape[0] != mask.shape[0] or background.shape[1] != mask.shape[1]:
+        msg = "Background and mask must have the same shape"
+        raise ValueError(msg)
 
     layer = np.zeros((background.shape[0], background.shape[1], 4), dtype=np.uint8)
     layer[:, :, 3] = mask
@@ -198,7 +210,7 @@ with gr.Blocks(css=css) as demo:
             )
             image_and_mask_examples = gr.Examples(
                 examples=[
-                    make_example(path)
+                    make_example(path, None)
                     for path in Path("./examples/scenes").glob("*.png")
                 ],
                 label="Room examples",