| """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()) | |