"""Deterministic paired prompt manifest."""

from __future__ import annotations

import hashlib
import json
from pathlib import Path

CATEGORIES = ("animals", "people", "landscapes", "vehicles", "architecture", "food", "indoor", "abstract")
SUBJECTS = {
    "animals": ["red fox crossing a snowy ridge", "sea turtle above a coral reef", "owl on a mossy branch", "elephant family at dusk", "golden retriever in a farmers market", "penguin colony on blue ice", "dragonfly on a lotus pond", "wild horse in desert wind", "sleeping koala in eucalyptus", "humpback whale breaching offshore"],
    "people": ["violinist backstage before a concert", "ceramic artist shaping a bowl", "runner crossing a bridge at dawn", "grandmother reading beside a window", "astronomer under a meteor shower", "street dancer in a neon alley", "gardener holding heirloom tomatoes", "chef plating a tasting menu", "child flying a paper kite", "sailor repairing a bright red net"],
    "landscapes": ["volcanic valley beneath low clouds", "lavender fields in Provence", "desert canyon after rain", "fjord with a small wooden ferry", "autumn forest reflected in a lake", "rice terraces in morning mist", "coastal cliffs under storm light", "alpine meadow with a footpath", "tropical waterfall behind ferns", "salt flats mirroring the night sky"],
    "vehicles": ["yellow tram turning through old streets", "sailboat in a harbor of lanterns", "red bicycle beside a canal", "electric train entering a tunnel", "vintage airplane in a hangar", "motorcycle on a coastal road", "blue pickup at a rural gas station", "submarine beneath polar ice", "hot air balloon over vineyards", "kayak on a glassy river"],
    "architecture": ["brutalist library in winter", "wooden temple surrounded by cedars", "Art Deco cinema at twilight", "floating pavilion on a lake", "desert observatory with domes", "narrow townhouse courtyard", "futuristic transit station", "stone bridge over a market canal", "greenhouse made of iron and glass", "lighthouse on a basalt shore"],
    "food": ["sourdough loaf cooling on linen", "ramen bowl in a tiny Tokyo shop", "summer peaches on a blue plate", "spice market with open jars", "chocolate cake with raspberry glaze", "breakfast table with coffee and oranges", "handmade dumplings on bamboo", "roasted vegetables on a ceramic platter", "lemon tart beside garden flowers", "night market skewers over charcoal"],
    "indoor": ["sunlit reading nook with a velvet chair", "rainy day kitchen with copper pans", "minimal studio with a drafting table", "old train compartment with luggage", "music room filled with records", "green apartment balcony after watering", "museum gallery with a skylight", "warm cabin bedroom with quilts", "maker workshop with wooden tools", "quiet cafe corner beside a window"],
    "abstract": ["ink spirals colliding with gold leaf", "isometric cubes in coral and teal", "watercolor gradients like a topographic map", "paper cut shapes casting long shadows", "glass ribbons twisting in sunlight", "monochrome circles with one yellow accent", "liquid metal waves on black", "woven threads forming a labyrinth", "pixel mosaic inspired by ocean tides", "floating geometric shards in soft fog"],
}


def build_prompts(count: int = 100) -> list[dict[str, object]]:
    """8カテゴリを循環し、stable IDとpaired seedを付けたpromptを作る。"""
    if count < 1:
        raise ValueError("count must be positive")
    result: list[dict[str, object]] = []
    for index in range(count):
        category = CATEGORIES[index % len(CATEGORIES)]
        variant = index // len(CATEGORIES)
        prompt = f"{SUBJECTS[category][variant % len(SUBJECTS[category])]}, {('wide composition' if variant % 3 == 0 else 'intimate composition' if variant % 3 == 1 else 'cinematic composition')}, {('morning light' if variant % 4 == 0 else 'soft overcast light' if variant % 4 == 1 else 'late afternoon light' if variant % 4 == 2 else 'blue hour light')}, high detail, scene {index + 1}"
        prompt_id = f"pm4-{index + 1:03d}-{hashlib.sha256(prompt.encode()).hexdigest()[:8]}"
        seed = 10_000 + index
        result.append({"prompt_id": prompt_id, "category": category, "prompt": prompt, "seed": seed, "paired_seed": seed, "initial_latent_seed": seed})
    return result


def load_prompt_manifest(path: str | Path) -> list[dict[str, object]]:
    """version-controlled prompt manifestを検証して読む。"""
    entries = json.loads(Path(path).read_text(encoding="utf-8"))
    if not isinstance(entries, list) or len(entries) != 100:
        raise ValueError("prompt manifest must contain exactly 100 entries")
    texts = [str(item["prompt"]) for item in entries]
    if len(set(texts)) != 100:
        raise ValueError("prompt text must be unique")
    for item in entries:
        if {"prompt_id", "category", "prompt", "seed", "initial_latent_seed"} - item.keys():
            raise ValueError("prompt entry schema is incomplete")
        if item["category"] not in CATEGORIES:
            raise ValueError(f"unknown prompt category: {item['category']}")
    if any(sum(item["category"] == category for item in entries) < 10 for category in CATEGORIES):
        raise ValueError("each prompt category requires at least 10 entries")
    return entries


def write_prompt_manifest(path: str | Path, count: int = 100) -> str:
    """prompt manifestをJSONへ書き、sha256を返す。"""
    target = Path(path)
    if count == 100 and Path(path).exists():
        payload = load_prompt_manifest(path)
    else:
        payload = build_prompts(count)
    target.write_text(json.dumps(payload, ensure_ascii=False, indent=2) + "\n")
    return hashlib.sha256(target.read_bytes()).hexdigest()