Upload 3 files

README.md

@@ -4,7 +4,7 @@ emoji: 🖼🖼️📦
 colorFrom: purple
 colorTo: red
 sdk: gradio
-sdk_version: 6.17.3
+sdk_version: 6.10.0
 python_version: "3.12"
 app_file: app.py
 license: mit

dc.py

@@ -880,6 +880,13 @@ def get_gsd(repo_id, vae, task, controlnet_model):
         return gsd
 
 
+def _editor_first_layer(editor_value):
+    if not isinstance(editor_value, dict):
+        return None
+    layers = editor_value.get("layers") or []
+    return layers[0] if layers else None
+
+
 #@spaces.GPU
 def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
           model_name=load_diffusers_format_model[0], lora1=None, lora1_wt=1.0, lora2=None, lora2_wt=1.0,
@@ -901,12 +908,15 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
           gpu_duration=59, translate=False, recom_prompt=True, progress=gr.Progress(track_tqdm=True)):
     MAX_SEED = np.iinfo(np.int32).max
 
-    image_mask = image_control_dict['layers'][0] if isinstance(image_control_dict, dict) and not image_mask else image_mask
-    image_control = image_control_dict['background'] if isinstance(image_control_dict, dict) else None
-    mask_ip1 = image_ip1_dict['layers'][0] if isinstance(image_ip1_dict, dict) and not mask_ip1 else mask_ip1
-    image_ip1 = image_ip1_dict['background'] if isinstance(image_ip1_dict, dict) else None
-    mask_ip2 = image_ip2_dict['layers'][0] if isinstance(image_ip2_dict, dict) and not mask_ip1 else mask_ip1
-    image_ip2 = image_ip2_dict['background'] if isinstance(image_ip2_dict, dict) else None
+    if not image_mask:
+        image_mask = _editor_first_layer(image_control_dict)
+    image_control = image_control_dict.get('background') if isinstance(image_control_dict, dict) else None
+    if not mask_ip1:
+        mask_ip1 = _editor_first_layer(image_ip1_dict)
+    image_ip1 = image_ip1_dict.get('background') if isinstance(image_ip1_dict, dict) else None
+    if not mask_ip2:
+        mask_ip2 = _editor_first_layer(image_ip2_dict)
+    image_ip2 = image_ip2_dict.get('background') if isinstance(image_ip2_dict, dict) else None
     style_prompt = None
     style_json = None
     hires_before_adetailer = False

utils.py

@@ -578,7 +578,12 @@ def create_mask_now(img, invert):
 
     time.sleep(0.5)
 
-    transparent_image = img["layers"][0]
+    layers = (img.get("layers") or []) if isinstance(img, dict) else []
+    if not layers:
+        background = img.get("background") if isinstance(img, dict) else None
+        return background, None
+
+    transparent_image = layers[0]
 
     # Extract the alpha channel
     alpha_channel = np.array(transparent_image)[:, :, 3]