internals/util/image.py

import io

from PIL import Image


def to_bytes(image: Image.Image) -> bytes:
    with io.BytesIO() as output:
        image.save(output, format="JPEG")
        return output.getvalue()


def resize_image_to512(image: Image.Image) -> Image.Image:
    iw, ih = image.size
    if iw > ih:
        image = image.resize((512, int(512 * ih / iw)))
    else:
        image = image.resize((int(512 * iw / ih), 512))
    return image