utils/image_utils.py

"""Image utility functions for data URL conversion."""

import base64


def image_to_base64_data_url(image_data: bytes, mime_type: str) -> str:
    r"""Convert image bytes and MIME type to base64 data URL format.

    Example: (b'\xff\xd8\xff...', 'image/jpeg') -> "data:image/jpeg;base64,/9j/4AAQ..."

    Used by: Pipeline stages and web app to include images in LLM messages.
    """
    encoded = base64.b64encode(image_data).decode("ascii")
    return f"data:{mime_type};base64,{encoded}"