add-auto generated mask

runway.py

@@ -29,8 +29,8 @@ async def root():
 def prepare_guided_image(original_image: Image, reference_image: Image, mask_image: Image) -> Image:
     """
     Prepare an initial image by softly blending the reference image into the masked area.
-    - Unmasked areas (white in mask, 255) remain fully from the original image.
-    - Masked areas (black in mask, 0) take content from the reference image with soft blending.
+    - Areas to keep (black in mask, 0) remain fully from the original image.
+    - Areas to inpaint (white in mask, 255) take content from the reference image with soft blending.
     
     Args:
         original_image (Image): The original image (RGB).
@@ -43,13 +43,13 @@ def prepare_guided_image(original_image: Image, reference_image: Image, mask_ima
     # Convert images to numpy arrays
     original_array = np.array(original_image)
     reference_array = np.array(reference_image)
-    mask_array = np.array(mask_image) / 255.0  # Normalize to [0, 1] for soft blending
+    mask_array = np.array(mask_image) / 255.0  # Normalize to [0, 1]
 
     # Expand mask to RGB channels
     mask_array = mask_array[:, :, np.newaxis]
 
-    # Softly blend: unmasked areas (1) keep original, masked areas (0) use reference
-    blended_array = original_array * mask_array + reference_array * (1 - mask_array)
+    # Blend: use original where mask=0 (black), reference where mask=1 (white)
+    blended_array = original_array * (1 - mask_array) + reference_array * mask_array
     blended_array = blended_array.astype(np.uint8)
 
     return Image.fromarray(blended_array)
@@ -71,7 +71,7 @@ def soften_mask(mask_image: Image, softness: int = 5) -> Image:
 def generate_rectangular_mask(image_size: tuple, x1: int = 100, y1: int = 100, x2: int = 200, y2: int = 200) -> Image:
     """
     Generate a rectangular mask matching the image dimensions.
-    - White (255) for unmasked areas, black (0) for masked areas.
+    - Black (0) for areas to keep, white (255) for areas to inpaint.
     
     Args:
         image_size (tuple): Tuple of (width, height) of the original image.
@@ -81,12 +81,12 @@ def generate_rectangular_mask(image_size: tuple, x1: int = 100, y1: int = 100, x
     Returns:
         Image: The generated mask in grayscale (L mode).
     """
-    # Create a blank white mask (255 = unmasked)
-    mask = Image.new("L", image_size, 255)
+    # Create a blank black mask (0 = keep)
+    mask = Image.new("L", image_size, 0)
     draw = ImageDraw.Draw(mask)
     
-    # Draw a black rectangle (0 = masked)
-    draw.rectangle([x1, y1, x2, y2], fill=0)
+    # Draw a white rectangle (255 = inpaint)
+    draw.rectangle([x1, y1, x2, y2], fill=255)
     
     return mask