"""Generate a sprite from a prompt.

    python generate.py "red potion" --seed 3 --scale 8 --out sprite.png
"""

import argparse
import sys
from pathlib import Path

from pixelgen import weights
from pixelgen.model import PixelGen


def main(argv=None):
    ap = argparse.ArgumentParser(description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)
    ap.add_argument("prompt")
    ap.add_argument("--seed", type=int, default=0)
    ap.add_argument("--scale", type=int, default=8)
    ap.add_argument("--out", default="sprite.png")
    ap.add_argument("--weights", default=None, help="load a saved weights file instead of building")
    args = ap.parse_args(argv)

    model = weights.load(args.weights) if args.weights else PixelGen()
    img = model.sprite(args.prompt, seed=args.seed, scale=args.scale)
    out = Path(args.out)
    out.parent.mkdir(parents=True, exist_ok=True)
    img.save(out)
    print(f"wrote {out} ({img.size[0]}x{img.size[1]})")
    return 0


if __name__ == "__main__":
    sys.exit(main())