update utils

utils.py

@@ -0,0 +1,27 @@
+from typing import Tuple
+
+import PIL
+from PIL.Image import Image as PILImage
+
+
+def get_background_dominant_color(img: PILImage, mask: PILImage) -> tuple:
+    negative_img = img.copy()
+    negative_mask = PIL.ImageOps.invert(mask)
+    negative_img.putalpha(negative_mask)
+    negative_img = negative_img.resize((1, 1))
+    r, g, b, a = negative_img.getpixel((0, 0))
+    return r, g, b, 255
+
+
+def apply_background_color(img: PILImage, color: Tuple[int, int, int, int]) -> PILImage:
+    r, g, b, a = color
+    colored_image = PIL.Image.new("RGBA", img.size, (r, g, b, a))
+    colored_image.paste(img, mask=img)
+    return colored_image
+
+
+def make_flatten_background(img, mask):
+    img = PIL.Image.open(img)
+    mask = PIL.Image.open(mask)
+    color = get_background_dominant_color(img, mask)
+    return apply_background_color(img, color)