print("working***")
from rembg.bg import remove
import numpy as np
import io
from PIL import Image
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
print("working***")
from utils.files_lib import get_resources_data_path

# input_path = 'test.jpg'
# output_path = 'out.png'


input_path = get_resources_data_path("images","re_move_bg.jpeg")
# input_path = get_resources_data_path("images","01.jpg")
# input_path = get_resources_data_path("images","re_move_bg.jpeg")
output_path = get_resources_data_path("images","out-4.png")

f = np.fromfile(input_path)
print("np.from done")
result = remove(f)
print("remove(f) done")
img = Image.open(io.BytesIO(result)).convert("RGBA")
print("output image done")
img.save(output_path)
print("save done")