Update app.py

Fix: Handle missing or empty 'layers' field in person_image

- Added validation to check if 'layers' exist and are non-empty before accessing the first layer in the submit function.
- If 'layers' are missing or empty, set the mask to None and auto-generate it using the AutoMasker.
- Ensured smooth fallback for cases where no manual mask is provided by automatically creating a mask.
- Enhanced stability by preventing 'IndexError' and ensuring graceful handling of input images without layers.

This commit improves error handling and ensures more robust functionality for generating virtual try-on results.

app.py

@@ -133,14 +133,19 @@ def submit_function(
     seed,
     show_type
 ):
-    person_image, mask = person_image["background"], person_image["layers"][0]
-    mask = Image.open(mask).convert("L")
-    if len(np.unique(np.array(mask))) == 1:
-        mask = None
+    # Check if layers exist and are not empty
+    if "layers" in person_image and person_image["layers"]:
+        person_image, mask = person_image["background"], person_image["layers"][0]
+        mask = Image.open(mask).convert("L")
+        if len(np.unique(np.array(mask))) == 1:  # All mask values are the same (empty mask)
+            mask = None
+        else:
+            mask = np.array(mask)
+            mask[mask > 0] = 255  # Convert to binary mask (0 or 255)
+            mask = Image.fromarray(mask)
     else:
-        mask = np.array(mask)
-        mask[mask > 0] = 255
-        mask = Image.fromarray(mask)
+        person_image = person_image["background"]
+        mask = None  # No mask is provided, it will be auto-generated
 
     tmp_folder = args.output_dir
     date_str = datetime.now().strftime("%Y%m%d%H%M%S")
@@ -168,7 +173,6 @@ def submit_function(
     mask = mask_processor.blur(mask, blur_factor=9)
 
     # Inference
-    # try:
     result_image = pipeline(
         image=person_image,
         condition_image=cloth_image,
@@ -177,10 +181,6 @@ def submit_function(
         guidance_scale=guidance_scale,
         generator=generator
     )[0]
-    # except Exception as e:
-    #     raise gr.Error(
-    #         "An error occurred. Please try again later: {}".format(e)
-    #     )
     
     # Post-process
     masked_person = vis_mask(person_image, mask)
@@ -195,7 +195,7 @@ def submit_function(
             conditions = image_grid([person_image, cloth_image], 2, 1)
         else:
             condition_width = width // 3
-            conditions = image_grid([person_image, masked_person , cloth_image], 3, 1)
+            conditions = image_grid([person_image, masked_person, cloth_image], 3, 1)
         conditions = conditions.resize((condition_width, height), Image.NEAREST)
         new_result_image = Image.new("RGB", (width + condition_width + 5, height))
         new_result_image.paste(conditions, (0, 0))