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).convert("L")
    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_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_original, mask)
    return apply_background_color(img_cutted, color)