llama-index

1
"""Utils for manipulating images."""
2
import base64
3
from io import BytesIO
4
from typing import cast
5

6
from PIL import Image
7

8

9
def img_2_b64(image: Image, format: str = "JPEG") -> str:
10
    """Convert a PIL.Image to a base64 encoded image str."""
11
    buff = BytesIO()
12
    image.save(buff, format=format)
13
    return cast(str, base64.b64encode(buff.getvalue()))
14

15

16
def b64_2_img(data: str) -> Image:
17
    """Convert base64 encoded image str to a PIL.Image."""
18
    buff = BytesIO(base64.b64decode(data))
19
    return Image.open(buff)
20