| """Style registry — maps style names to LoRA sources. |
| |
| Each style can point to a local .safetensors file or a HuggingFace Hub repo. |
| pipe.load_lora_weights() handles both transparently. |
| """ |
|
|
| STYLES = { |
| "Sunset Coastal Drive": { |
| "source": "samuelsattler/warm-sunset-lora", |
| "weight_name": "pytorch_lora_weights.safetensors", |
| "weight": 1.0, |
| "trigger": "sks", |
| "description": "Golden hour warmth, sun flares, silhouettes, warm color grading", |
| "quality_suffix": "8K, cinematic, golden hour glow, warm volumetric light, lens flare, shallow depth of field", |
| "image_prompt_guidance": ( |
| "SETTING — Coastal sunset drive:\n" |
| "The shared world is a drive along a coastal highway as the sun " |
| "sets. All scenes take place in or around this journey — a car " |
| "cruising along cliffs above the ocean, waves crashing against " |
| "rocks below, wind whipping through open windows, palm trees " |
| "swaying overhead, the sun sinking into the sea on the horizon. " |
| "No humans visible — focus on the car, the road, the ocean, " |
| "and the landscape. Every shot must have motion: wheels turning, " |
| "waves rolling, sun flares shifting, clouds drifting." |
| ), |
| }, |
| "Rainy City Night": { |
| "source": "artificialguybr/filmgrain-redmond-filmgrain-lora-for-sdxl", |
| "weight_name": "FilmGrainRedmond-FilmGrain-FilmGrainAF.safetensors", |
| "weight": 0.8, |
| "trigger": "FilmGrainAF", |
| "description": "35mm film grain, moody color grading, cinematic lighting", |
| "quality_suffix": "8K, cinematic, shot on 35mm film, dramatic rim lighting, high contrast, shallow depth of field", |
| "image_prompt_guidance": ( |
| "SETTING — Rainy city at night:\n" |
| "The shared world is a walk through a rain-soaked city after dark. " |
| "All scenes take place on these streets — rain streaking through " |
| "streetlights, puddles reflecting neon signs, steam rising from " |
| "grates, traffic passing with blurred headlights, wet umbrellas, " |
| "rain hammering awnings, water streaming down windows. " |
| "No humans visible — focus on the environment, the rain, the " |
| "reflections, and the city itself. Every shot must have motion: " |
| "rain falling, cars passing, lights flickering, water flowing " |
| "through gutters." |
| ), |
| }, |
| "Cyberpunk": { |
| "source": "jbilcke-hf/sdxl-cyberpunk-2077", |
| "weight_name": "pytorch_lora_weights.safetensors", |
| "weight": 0.9, |
| "trigger": "cyberpunk-2077", |
| "description": "Neon-lit cityscapes, dark futuristic vibes, glowing signs", |
| "quality_suffix": "8K, cinematic, neon-drenched, volumetric fog, sharp details, high contrast, dramatic lighting", |
| "image_prompt_guidance": ( |
| "SETTING — Cyberpunk nightlife cityscape:\n" |
| "The shared world is a futuristic megacity at night. All scenes " |
| "take place in this neon-drenched urban sprawl — holographic " |
| "billboards flickering on skyscrapers, flying vehicles streaking " |
| "between towers, neon signs buzzing and glitching, rain falling " |
| "through laser grids, steam erupting from vents, LED-lit market " |
| "stalls with flickering displays. " |
| "No humans visible — focus on the city, the machines, the neon, " |
| "and the architecture. Every shot must have motion: vehicles " |
| "flying, signs flickering, rain falling, smoke drifting, lights " |
| "pulsing." |
| ), |
| }, |
| "Watercolour Harbour": { |
| "source": "ostris/watercolor_style_lora_sdxl", |
| "weight_name": "watercolor_v1_sdxl.safetensors", |
| "weight": 1.4, |
| "trigger": "", |
| "description": "Soft watercolor painting style, fluid washes, gentle blending", |
| "quality_suffix": "8K, watercolor painting, soft painterly washes, fluid blending, delicate brushstrokes, atmospheric", |
| "image_prompt_guidance": ( |
| "SETTING — Stormy harbour village:\n" |
| "The shared world is a coastal fishing village during a storm. " |
| "All scenes take place in and around this harbour — waves " |
| "crashing against stone sea walls, fishing boats rocking and " |
| "pulling at their moorings, rain sweeping across the harbour " |
| "in sheets, wind tearing through flags and sails, seabirds " |
| "wheeling against dark clouds, lanterns swinging on posts, " |
| "water pouring off rooftops into cobblestone streets. " |
| "No humans visible — focus on the sea, the boats, the storm, " |
| "and the village. Every shot must have motion: waves surging, " |
| "boats swaying, rain lashing, flags snapping in the wind." |
| ), |
| }, |
| } |
|
|
|
|
| def get_style(name: str) -> dict: |
| """Look up a style by name. Raises KeyError if not found.""" |
| return STYLES[name] |
|
|
|
|
| def style_names() -> list[str]: |
| """Return list of available style names for UI dropdowns.""" |
| return list(STYLES.keys()) |
|
|
