tools/background.py

from rembg import remove
from PIL import Image
import io


def remove_background(image_path: str):

    with open(image_path, "rb") as f:
        input_data = f.read()

    output = remove(input_data)

    img = Image.open(io.BytesIO(output)).convert("RGBA")

    return img