import os
import io
import base64
import uuid
from PIL import Image


def encode_image(image_path: str) -> str:
    """Convert an image file to base64 string."""
    with open(image_path, "rb") as image_file:
        return base64.b64encode(image_file.read()).decode("utf-8")


def decode_image(image_path: str) -> Image.Image:
    """Convert a base64 string to a PIL Image."""
    image_data = base64.b64decode(image_path)
    return Image.open(io.BytesIO(image_data))

def save_image(image: Image.Image, directory:str = "images") -> str:
    """Save a PIL Image to disk and return the path."""
    os.makedirs(directory, exist_ok = True)
    image_id = str(uuid.uuid4())
    image_path = os.path.join(directory, f"{image_id}.png")
    image.save(image_path)
    return image_path