import mimetypes
from base64 import b64decode, b64encode
from io import BytesIO

from PIL import Image


def base64_decode_image_data_url(data_url: str) -> Image:
    _, data = data_url.split("base64,", maxsplit=1)
    byte_data = b64decode(data)
    return Image.open(BytesIO(byte_data))


def base64_encode_image_file(image_file: BytesIO) -> str:
    file_type = image_file.type
    if not file_type:
        file_type = mimetypes.guess_type(image_file.name)[0]
    file_data = image_file.read()
    b64 = b64encode(file_data).decode("utf-8")
    return f"data:{file_type};base64,{b64}"