|
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(base64_string: str) -> Image.Image:
|
|
"""Convert a base64 string to a PIL Image."""
|
|
image_data = base64.b64decode(base64_string)
|
|
return Image.open(io.BytesIO(image_data))
|
|
|
|
|
|
def save_image(image: Image.Image, directory: str = "image_outputs") -> 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
|
|
|
