utils/image_utils.py

import base64
from io import BytesIO
from typing import Any, Dict, List

from PIL import Image


def resize_images(
    images: List[Image.Image],
    config: Dict[str, Any]
):
    images_resized = [image.resize((config["IMG_SIZE"], config["IMG_SIZE"])) for image in images]
    if images_resized[0].mode != 'RGB':
        images_resized = [image.convert('RGB') for image in images_resized]
    return images_resized

def image_to_base64(image: Image.Image):
    """Convert PIL Image to base64 string"""
    buffer = BytesIO()
    image.save(buffer, format='PNG')
    return base64.b64encode(buffer.getvalue()).decode('utf-8')

def base64_to_image(base64_string: str):
    """Convert base64 string to PIL Image"""
    return Image.open(BytesIO(base64.b64decode(base64_string)))