Handle blending correctly

app.py

@@ -54,10 +54,6 @@ CONFIG_FILE = 'GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py'
 GROUNDINGDINO_CHECKPOINT = "groundingdino_swint_ogc.pth"
 SAM_CHECKPOINT = 'sam_hq_vit_l.pth'
 OUTPUT_DIR = "outputs"
-# FAL_KEY = os.getenv("FAL_KEY")
-# UPLOAD_DIR = "./tmp/images"
-
-# os.makedirs(UPLOAD_DIR, exist_ok=True)
 
 # Global variables for model caching
 _models = {
@@ -364,8 +360,21 @@ def expand_mask(mask, expand, tapered_corners):
     return Image.fromarray(mask_np, mode="L")
 
 def image_blend_by_mask(image_a, image_b, mask, blend_percentage):
-
-    mask = ImageOps.invert(mask.convert('L'))
+    # Ensure images have the same size and mode
+    image_a = image_a.convert('RGB')
+    image_b = image_b.convert('RGB')
+    mask = mask.convert('L')
+    
+    # Resize images if they don't match
+    if image_a.size != image_b.size:
+        image_b = image_b.resize(image_a.size, Image.LANCZOS)
+    
+    # Ensure mask has the same size
+    if mask.size != image_a.size:
+        mask = mask.resize(image_a.size, Image.LANCZOS)
+    
+    # Invert mask
+    mask = ImageOps.invert(mask)
 
     # Mask image
     masked_img = Image.composite(image_a, image_b, mask)
@@ -585,8 +594,6 @@ def generate_image(input_image_path, prompt):
 
     resized_input_img = resize_to_square(input_image_path, 512)
 
-    # resized_input_img_path = resized_input_img.save('input_img.png')
-
     resized_input_img_path = '/tmp/gradio/resized_input_img.png'
 
     resized_input_img.convert("RGBA").save(resized_input_img_path, "PNG")
@@ -597,8 +604,6 @@ def generate_image(input_image_path, prompt):
 
     final_image = blend_details(resized_input_img, ai_gen_image, mask_input_image)
 
-    # final_image = ai_gen_image
-
     return final_image
 
 def create_ui():