"""VNAsset — Visual Novel Asset Pipeline CLI."""
import click

from .edit import edit
from .generate import generate


def _parse_seed(ctx, param, value):
    if value is None or value == "":
        return None
    if value.lower() == "random":
        return None
    try:
        return int(value)
    except ValueError:
        raise click.BadParameter("seed must be an integer or 'random'")


@click.group()
def main():
    """VNAsset — fast CLI pipeline for visual novel image assets."""


@main.command()
@click.option("--checkpoint", required=True, help="Path to SDXL checkpoint (.safetensors)")
@click.option("--prompt", required=True, help="Positive prompt")
@click.option("--negative-prompt", default="", help="Negative prompt")
@click.option("--width", default=1024, type=int)
@click.option("--height", default=1024, type=int)
@click.option("--steps", default=20, type=int)
@click.option("--cfg", default=4.5, type=float)
@click.option(
    "--seed", default="random", callback=_parse_seed,
    help="RNG seed (integer or 'random')",
)
@click.option("--output", default="output.png", help="Output image path")
@click.option(
    "--raw", is_flag=True,
    help="Disable ComfyUI-style prompt weighting (use plain diffusers encoding)",
)
def generate_cmd(checkpoint, prompt, negative_prompt, width, height, steps, cfg, seed, output, raw):
    """Generate an image from an SDXL checkpoint."""
    generate(
        checkpoint_path=checkpoint,
        prompt=prompt,
        negative_prompt=negative_prompt,
        width=width,
        height=height,
        steps=steps,
        cfg=cfg,
        seed=seed,
        output_path=output,
        raw=raw,
    )


@main.command()
@click.option("--model", required=True, help="Path to Qwen Image Edit diffusion model (.safetensors)")
@click.option("--input", "input_path", required=True, help="Input image to edit")
@click.option("--prompt", required=True, help="Edit instruction")
@click.option("--steps", default=20, type=int, help="Inference steps")
@click.option("--cfg", default=4.0, type=float, help="CFG scale")
@click.option(
    "--seed", default="random", callback=_parse_seed,
    help="RNG seed (integer or 'random')",
)
@click.option("--output", default="output.png", help="Output image path")
@click.option("--lora", "lora_path", default=None,
              help="Path to LoRA .safetensors (e.g. Lightning 4-step LoRA)")
def edit_cmd(model, input_path, prompt, steps, cfg, seed, output, lora_path):
    """Edit an image using Qwen Image Edit."""
    edit(
        model_path=model,
        input_path=input_path,
        prompt=prompt,
        steps=steps,
        cfg=cfg,
        seed=seed,
        output_path=output,
        lora_path=lora_path,
    )