Update utils.py

utils.py

@@ -2,11 +2,12 @@ from typing import Tuple
 
 import PIL
 from PIL.Image import Image as PILImage
+from PIL import ImageOps
 
 
 def get_background_dominant_color(img: PILImage, mask: PILImage) -> tuple:
     negative_img = img.copy()
-    negative_mask = PIL.ImageOps.invert(mask)
+    negative_mask = PIL.ImageOps.invert(mask).convert("L")
     negative_img.putalpha(negative_mask)
     negative_img = negative_img.resize((1, 1))
     r, g, b, a = negative_img.getpixel((0, 0))
@@ -20,8 +21,10 @@ def apply_background_color(img: PILImage, color: Tuple[int, int, int, int]) -> P
     return colored_image
 
 
-def make_flatten_background(img, mask):
-    img = PIL.Image.open(img)
+def make_flatten_background(img_original, img_cutted, mask):
+    img_original = PIL.Image.open(img_original)
+    img_cutted = PIL.Image.open(img_cutted)
     mask = PIL.Image.open(mask)
-    color = get_background_dominant_color(img, mask)
-    return apply_background_color(img, color)
+    color = get_background_dominant_color(img_original, mask)
+    return apply_background_color(img_cutted, color)
+