Update app.py

app.py

@@ -193,7 +193,10 @@ def get_image(img_input, mask=False, scale=True):
         img = cv2.imread(img_input)
         img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     else:
-        raise ValueError("Unsupported image input type")
+        raise ValueError(f"Unsupported image input type: {type(img_input)}")
+
+    if img is None:
+        raise ValueError("Failed to load image")
 
     if mask:
         if len(img.shape) == 3:
@@ -209,8 +212,12 @@ def blend_images(bg_img, obj_img, mask_img, method):
     bg_img = get_image(bg_img)
     obj_img = get_image(obj_img)
     
-    if isinstance(mask_img, str):
+    # Handle different input types for mask_img
+    if isinstance(mask_img, dict):
+        mask_img = mask_img.get('composite') or mask_img.get('background')
+    elif isinstance(mask_img, str):
         mask_img = cv2.imread(mask_img, cv2.IMREAD_GRAYSCALE)
+    
     mask_img = get_image(mask_img, mask=True)
 
     if method == "Poisson":
@@ -259,12 +266,19 @@ with gr.Blocks(theme='bethecloud/storj_theme') as iface:
         outputs=output_image
     )
     
+    def create_image_editor_input(image_path):
+        return {
+            "background": image_path,
+            "layers": [],
+            "composite": image_path
+        }
+
     gr.Examples(
-    examples=[
-        ["img1.jpg", "img2.jpg", "mask1.jpg", "Poisson"],
-        ["img3.jpg", "img4.jpg", "mask2.jpg", "Mixed Gradient"],
-        ["img6.jpg", "img9.jpg", "mask3.jpg", "Laplacian"]
-    ],
+        examples=[
+            ["img1.jpg", "img2.jpg", create_image_editor_input("mask1.jpg"), "Poisson"],
+            ["img3.jpg", "img4.jpg", create_image_editor_input("mask2.jpg"), "Mixed Gradient"],
+            ["img6.jpg", "img9.jpg", create_image_editor_input("mask3.jpg"), "Laplacian"]
+        ],
         inputs=[bg_img, obj_img, mask_img, method],
         outputs=output_image,
         fn=blend_images,