fix mask

demo.py

@@ -705,12 +705,36 @@ console.log("Hello, world!");
 
 
 def create_demo():
+
+    black_background = np.zeros((400, 400), dtype=np.uint8)
+
+    # Load the default mask image if it exists
+    try:
+        mask_image = Image.open("assets/a4c_seg.png").convert("L")
+        mask_image = mask_image.resize((400, 400), Image.Resampling.LANCZOS)
+        # Make it binary (0 or 255)
+        mask_image = ImageOps.autocontrast(mask_image, cutoff=0)
+        mask_image = mask_image.point(lambda p: 255 if p > 127 else 0)
+        mask_array = np.array(mask_image)
+
+        # Create the editor value structure
+        editor_value = {
+            "background": black_background,  # Black background
+            "layers": [mask_array],  # The mask as an editable layer
+            "composite": mask_array,  # The composite image (what's displayed)
+        }
+    except Exception as e:
+        print(f"Error loading mask image: {e}")
+        # Fall back to empty canvas
+        editor_value = black_background
+
     # Define all components first
     mask_input = gr.ImageEditor(
         label="Binary Mask",
         height=400,
         width=400,
         image_mode="L",
+        value=editor_value,
         type="numpy",
         brush=gr.Brush(
             colors=["#ffffff"],
@@ -1019,7 +1043,7 @@ def create_demo():
                             # Fall back to empty canvas
                             editor_value = black_background
 
-                        mask_input.value = editor_value
+                        # mask_input.value = editor_value
                         mask_input.render()
                         class_selection.render()
                         sampling_steps.render()