tiny_image_gen/catalog.py

from dataclasses import dataclass


@dataclass(frozen=True)
class StylePreset:
    suffix: str
    hint: str


STYLE_PRESETS = {
    "Cinematic": StylePreset(
        suffix="cinematic lighting, highly detailed, dramatic composition, rich colors",
        hint="Adds film-like lighting and richer detail.",
    ),
    "Anime": StylePreset(
        suffix="anime style, clean line art, vibrant colors, expressive composition",
        hint="Pushes the output toward anime illustration.",
    ),
    "Fantasy": StylePreset(
        suffix="fantasy art, magical atmosphere, detailed environment, epic scene",
        hint="Adds a fantasy painting look.",
    ),
    "Pixel Art": StylePreset(
        suffix="pixel art, retro game sprite style, sharp edges, limited color palette",
        hint="Pushes toward retro pixel-art aesthetics.",
    ),
    "Photographic": StylePreset(
        suffix="photo realistic, natural light, realistic detail, professional photography",
        hint="Pushes toward a photographic look.",
    ),
}


def style_choices() -> list[str]:
    return list(STYLE_PRESETS.keys())


def default_prompt() -> str:
    return "A futuristic city street at sunset with neon reflections after rain"