Function encode_image_to_base64

datastudio/utils/vision.py:19–38

Encode PIL Image to base64 data URL string. Args: img: PIL Image object. target_size: Maximum dimension for thumbnail (-1 to skip resizing). fmt: Output format (default: JPEG). Returns: Base64 encoded data URL string.

(img, target_size=-1, fmt="JPEG")

def encode_image_to_base64(img, target_size=-1, fmt="JPEG"):
    """Encode PIL Image to base64 data URL string.

    Args:
        img: PIL Image object.
        target_size: Maximum dimension for thumbnail (-1 to skip resizing).
        fmt: Output format (default: JPEG).

    Returns:
        Base64 encoded data URL string.
    """
    if target_size > 0:
        img.thumbnail((target_size, target_size))
    img_buffer = io.BytesIO()
    img.save(img_buffer, format=fmt)
    image_data = img_buffer.getvalue()
    img_buffer.close()  # Explicitly close BytesIO to free memory
    image_base64 = base64.b64encode(image_data).decode("utf-8")
    del image_data  # Explicitly delete large object
    return f"data:image/jpeg;base64,{image_base64}"


# ==============================================================================