"""HoLo-FuSe CLI: python generate.py --label Cat --steps 16 --cfg 1.6 --seed 0 --out cat.png"""
import argparse
from model import generate

ap = argparse.ArgumentParser()
ap.add_argument("--label", default="Cat", choices=["Cat", "Dog"])
ap.add_argument("--arm", default="hsl", choices=["hsl", "learned", "none"])
ap.add_argument("--steps", type=int, default=16)
ap.add_argument("--cfg", type=float, default=1.6)
ap.add_argument("--seed", type=int, default=0)
ap.add_argument("--out", default="sample.png")
a = ap.parse_args()

img = generate(a.label, a.arm, a.steps, a.cfg, a.seed)[0]
img.save(a.out)
print(f"saved {a.out} ({a.label}, {a.arm}, steps={a.steps}, cfg={a.cfg}, seed={a.seed})")