rishiproject / app /utils.py
gaurinath's picture
Upload 8 files
41be7ed verified
raw
history blame contribute delete
541 Bytes
import base64
import os
from io import BytesIO
from PIL import Image
from dotenv import load_dotenv
load_dotenv()
def get_env(key: str, default: str = "") -> str:
return os.getenv(key, default)
def image_to_base64_jpeg(image: Image.Image) -> str:
"""Convert a PIL image to base64-encoded JPEG string."""
buf = BytesIO()
image = image.convert("RGB")
image.save(buf, format="JPEG", quality=90)
return base64.b64encode(buf.getvalue()).decode("utf-8")
def load_image(file) -> Image.Image:
return Image.open(file)