utils.py

import io
from PIL import Image
import base64


def add_stats(hits: list) -> None:
    """
    Add stats to the results so that we don't get KeyError when there are no stats

    Parameters:
        hits (list): The list of results from the Genius API
    """

    for i in range(len(hits)):
        if not hits[i]['result']['stats'].get('pageviews'):
            hits[i]['result']['stats'].update({'pageviews': 0})


def img_to_base64(res_img: bytes) -> str:
    """
    Converts a response image to a base64-encoded string.

    Parameters:
    `res_img` (bytes): The response image data as a bytes object.

    Returns:
    str: The base64-encoded image as a string with the "data:image/jpeg;base64," prefix.

    Example usage:
    >>> response = requests.get('https://example.com/image.jpg')
    >>> base64_image = img_to_base64(response.content)
    """

    image = Image.open(io.BytesIO(res_img))

    image = image.convert('RGB')

    buffered = io.BytesIO()
    image.save(buffered, format="JPEG")
    img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
    img_str = 'data:image/jpeg;base64,' + img_str

    return img_str