Update utils.py

utils.py

@@ -261,23 +261,63 @@ def overlay_ghost_mask(mask_img, background_img):
     bg_width, bg_height = background_img.size
     mask_width, mask_height = mask_img.size
     
+    if bg_width < mask_width or bg_height < mask_height:
+        bg_ratio = bg_width / bg_height
+        mask_ratio = mask_width / mask_height
+        
+        if mask_ratio > bg_ratio:
+            new_bg_height = int(bg_width / mask_ratio)
+            new_bg_width = bg_width
+        else:
+            new_bg_width = int(bg_height * mask_ratio) 
+            new_bg_height = bg_height
+        
+        new_background = Image.new('RGBA', (new_bg_width, new_bg_height), (255, 255, 255, 255))
+        paste_x = (new_bg_width - bg_width) // 2
+        paste_y = (new_bg_height - bg_height) // 2
+        new_background.paste(background_img, (paste_x, paste_y))
+        background_img = new_background
+        bg_width, bg_height = new_bg_width, new_bg_height
+    else:
+        mask_ratio = mask_width / mask_height
+        bg_ratio = bg_width / bg_height
+        
+        if bg_ratio > mask_ratio:
+            new_mask_height = int(mask_width / bg_ratio)
+            new_mask_width = mask_width
+        else:
+            new_mask_width = int(mask_height * bg_ratio)
+            new_mask_height = mask_height
+        
+        new_mask = Image.new('RGBA', (new_mask_width, new_mask_height), (0, 0, 0, 0))
+        paste_x = (new_mask_width - mask_width) // 2
+        paste_y = (new_mask_height - mask_height) // 2
+        new_mask.paste(mask_img, (paste_x, paste_y))
+        mask_img = new_mask
+        mask_width, mask_height = new_mask_width, new_mask_height
+    
     bg_ratio = bg_width / bg_height
     mask_ratio = mask_width / mask_height
     
-    if mask_ratio > bg_ratio:
-        new_mask_width = bg_width
-        new_mask_height = int(bg_width / mask_ratio)
+    if abs(mask_ratio - bg_ratio) < 0.01:
+        mask_resized = mask_img.resize((bg_width, bg_height), Image.Resampling.LANCZOS)
+        result = background_img.copy()
+        result.paste(mask_resized, (0, 0), mask_resized)
     else:
-        new_mask_height = bg_height
-        new_mask_width = int(bg_height * mask_ratio)
-    
-    mask_resized = mask_img.resize((new_mask_width, new_mask_height), Image.Resampling.LANCZOS)
-    
-    x_offset = (bg_width - new_mask_width) // 2
-    y_offset = (bg_height - new_mask_height) // 2
-    
-    result = background_img.copy()
-    result.paste(mask_resized, (x_offset, y_offset), mask_resized)
+        if mask_ratio > bg_ratio:
+            new_mask_width = bg_width
+            new_mask_height = int(bg_width / mask_ratio)
+        else:
+            new_mask_height = bg_height
+            new_mask_width = int(bg_height * mask_ratio)
+        
+        mask_resized = mask_img.resize((new_mask_width, new_mask_height), Image.Resampling.LANCZOS)
+        
+        x_offset = (bg_width - new_mask_width) // 2
+        y_offset = (bg_height - new_mask_height) // 2
+        
+        result = background_img.copy()
+        result.paste(mask_resized, (x_offset, y_offset), mask_resized)
     
     return result