import sys
from rembg import remove, new_session
from PIL import Image

src = "/var/www/safemyident.de/web/img/header-safe-my-ident.png"
img = Image.open(src).convert("RGBA")

for model in ["u2net", "u2net_human_seg", "isnet-general-use"]:
    try:
        sess = new_session(model)
        out = remove(img, session=sess,
                     alpha_matting=True,
                     alpha_matting_foreground_threshold=240,
                     alpha_matting_background_threshold=15,
                     alpha_matting_erode_size=8)
        dst = f"/tmp/smi-shot/person-{model}.png"
        out.save(dst)
        print(f"OK {model} -> {dst}")
    except Exception as e:
        print(f"FAIL {model}: {e}")