Update app.py

app.py

@@ -185,8 +185,13 @@ def laplacian_blend(img1: np.ndarray, img2: np.ndarray, mask: np.ndarray, depth:
     return np.clip(imgs[-1], 0, 1)
 
 def get_image(img_input, mask=False, scale=True):
+    if img_input is None:
+        raise ValueError("Image input is None")
+    
     if isinstance(img_input, dict):
-        img = img_input.get('composite') or img_input.get('background')
+        img = img_input.get('composite')
+        if img is None:
+            img = img_input.get('background')
     elif isinstance(img_input, np.ndarray):
         img = img_input
     elif isinstance(img_input, str):
@@ -214,7 +219,9 @@ def blend_images(bg_img, obj_img, mask_img, method):
     
     # Handle different input types for mask_img
     if isinstance(mask_img, dict):
-        mask_img = mask_img.get('composite') or mask_img.get('background')
+        mask_img = mask_img.get('composite')
+        if mask_img is None:
+            mask_img = mask_img.get('background')
     elif isinstance(mask_img, str):
         mask_img = cv2.imread(mask_img, cv2.IMREAD_GRAYSCALE)
     
@@ -272,7 +279,7 @@ with gr.Blocks(theme='bethecloud/storj_theme') as iface:
             "layers": [],
             "composite": image_path
         }
-
+    
     gr.Examples(
         examples=[
             ["img1.jpg", "img2.jpg", create_image_editor_input("mask1.jpg"), "Poisson"],