utils.py

import os
import uuid


def generate_id():
    return str(uuid.uuid4())


def ensure_dir(path):
    if not os.path.exists(path):
        os.makedirs(path)


def save_image(image, folder="outputs"):
    ensure_dir(folder)

    file_path = os.path.join(folder, f"{generate_id()}.png")

    image.save(file_path)

    return file_path