# Wytnij tło modelem segmentacji (rembg) i zapisz wycięty obraz RGBA.
# Kanał alfa wycięcia = maska mebla (alpha > 10). Użycie:
#   python scripts/rembg_mask.py <wejscie> <wyjscie.png> [sesja]
# Sesja domyślna: u2net. Model pobierany raz do ~/.u2net.
import sys
from rembg import remove, new_session


def main() -> None:
    src, dst = sys.argv[1], sys.argv[2]
    session_name = sys.argv[3] if len(sys.argv) > 3 else "u2net"
    with open(src, "rb") as f:
        data = f.read()
    out = remove(data, session=new_session(session_name))
    with open(dst, "wb") as f:
        f.write(out)


if __name__ == "__main__":
    main()