Rewrite styles and prompts

app.py

@@ -4,7 +4,7 @@ import random
 
 import gradio as gr
 
-from lib import Config, async_call, download_repo_files, generate
+from lib import Config, async_call, download_repo_files, generate, read_file
 
 # the CSS `content` attribute expects a string so we need to wrap the number in quotes
 refresh_seed_js = """
@@ -33,17 +33,6 @@ aspect_ratio_js = """
 """
 
 
-def read_file(path: str) -> str:
-    with open(path, "r", encoding="utf-8") as file:
-        return file.read()
-
-
-def random_fn():
-    prompts = read_file("data/prompts.json")
-    prompts = json.loads(prompts)
-    return gr.Textbox(value=random.choice(prompts))
-
-
 def create_image_dropdown(images, locked=False):
     if locked:
         return gr.Dropdown(
@@ -59,18 +48,18 @@ def create_image_dropdown(images, locked=False):
         )
 
 
-def gallery_fn(images, image, ip_image):
+async def gallery_fn(images, image, ip_image):
     return (
         create_image_dropdown(images, locked=image is not None),
         create_image_dropdown(images, locked=ip_image is not None),
     )
 
 
-def image_prompt_fn(images):
+async def image_prompt_fn(images):
     return create_image_dropdown(images)
 
 
-def image_select_fn(images, image, i):
+async def image_select_fn(images, image, i):
     # -2 is the lock icon, -1 is None
     if i == -2:
         return gr.Image(image)
@@ -79,6 +68,12 @@ def image_select_fn(images, image, i):
     return gr.Image(images[i][0]) if i > -1 else None
 
 
+async def random_fn():
+    prompts = read_file("data/prompts.json")
+    prompts = json.loads(prompts)
+    return gr.Textbox(value=random.choice(prompts))
+
+
 async def generate_fn(*args):
     if len(args) > 0:
         prompt = args[0]
@@ -159,10 +154,13 @@ with gr.Blocks(
 
                     with gr.Row():
                         styles = json.loads(read_file("data/styles.json"))
+                        style_ids = list(styles.keys())
+                        style_ids = [sid for sid in style_ids if not sid.startswith("_")]
                         style = gr.Dropdown(
                             value=Config.STYLE,
                             label="Style",
-                            choices=[("None", None)] + [(s["name"], s["id"]) for s in styles],
+                            choices=[("None", "none")]
+                            + [(styles[sid]["name"], sid) for sid in style_ids],
                         )
                         embeddings = gr.Dropdown(
                             elem_id="embeddings",

data/prompts.json

@@ -1,154 +1,48 @@
 [
-  "Stunning sunset over a futuristic city, with towering skyscrapers and flying vehicles, golden hour lighting and dramatic clouds, high detail, moody atmosphere",
-  "Mystical forest with glowing mushrooms and a babbling brook, surrounded by towering trees and shrouded in mist, ethereal, dreamlike, stylized",
-  "Majestic dragon, perched atop a cliff overlooking a fiery landscape, with smoke and ash rising into the air, intense, detailed, scales, dynamic, epic",
-  "Serene beach scene with crystal clear water and white sand, tropical palm trees swaying in the breeze, perfect paradise, seascape",
-  "Post-apocalyptic wasteland with rusted and abandoned vehicles, dust storms and towering dust clouds, gritty, dark, dramatic, apocalyptic, stylized",
-  "Mystical underwater world with vibrant coral and exotic sea creatures, sun beams shining through the water, mysterious, magical, otherworldly",
-  "Snowy winter wonderland with a lone cabin in the distance, surrounded by frosty trees and fresh snowfall, peaceful, serene, detailed, winter landscape",
-  "Mysterious and abandoned temple in the jungle, surrounded by lush vegetation and tall trees, atmospheric, ancient, moody",
-  "Vibrant and bustling city street with busy traffic, bright lights, and towering skyscrapers, chaotic, fast-paced, cityscape, high detail",
-  "Gothic cathedral in a stormy night, with lightning illuminating the sky and rain pouring down, dramatic, atmospheric, high detail, moody",
-  "Fantasy castle on a hilltop, surrounded by rolling hills and a beautiful sunset, magical, serene, high detail, romantic, stylized",
-  "Vast desert with sand dunes and a lone oasis in the distance, hot sun and clear blue sky, peaceful, serene, desertscape, high resolution",
-  "Ethereal aurora borealis over a snowy mountain range, with a full moon shining in the background, mystical, peaceful, serene, winter landscape, high detail",
-  "Giant robots fighting in a futuristic city, with buildings falling and explosions all around, intense, fast-paced, dramatic, stylized, futuristic",
-  "Medieval market square with vendors selling goods, colorful banners, and bustling crowds, lively, busy, historic, high detail, architectural",
-  "Abandoned prison on a cliff, with a stormy sea below, eerie, moody, atmospheric, architectural, stylized",
-  "Beautiful waterfall in a lush jungle, with sunlight shining through the trees, serene, peaceful, tropical, jungle landscape, high detail",
-  "Surreal carnival scene with bright lights, strange creatures, and a full moon, dreamlike, vibrant, stylized, surreal, high detail",
-  "Breathtaking mountain range with a clear river running through it, surrounded by tall trees and misty clouds, serene, peaceful, mountain landscape, high detail",
-  "Historic battlefield with armies clashing and smoke rising, intense, fast-paced, dramatic, historic, high detail",
-  "Enchanted forest with glowing fireflies and a babbling brook, surrounded by towering trees and shrouded in mist, magical, ethereal, dreamlike, stylized",
-  "Sunken ship in a vibrant coral reef, with schools of colorful fish swimming by, mysterious, magical, high detail",
-  "Futuristic city with towering skyscrapers and flying vehicles, set against a vibrant sunset, futuristic, serene, high detail",
-  "Volcanic island with a boiling crater and ash clouds rising into the air, intense, dramatic, natural disaster, high detail, volcanic landscape",
-  "Lonely lighthouse on a rocky coast during a storm, with waves crashing and lightning flashing, moody, atmospheric, seascape, high detail",
-  "Ruined castle on a cliff overlooking the sea, with a stormy sky and crashing waves, eerie, moody, atmospheric",
-  "Vibrant street fair with colorful stalls and bustling crowds, lively, busy, high detail",
-  "Mysterious underground cave with glowing crystals and an underground river, dark, mysterious, underground landscape, high detail",
-  "Glorious sunrise over a cityscape, with the city slowly coming to life and the sky turning orange and pink, serene, peaceful, cityscape, high detail",
-  "Breathtaking view of a snowy mountain range, with crisp clear air and a brilliant blue sky, serene, peaceful, majestic, high detail, winter landscape",
-  "Enchanting waterfall in a lush jungle, surrounded by exotic plants and wildlife, tranquil, serene, high detail, tropical landscape",
-  "Glowing aurora borealis over a frozen lake, with towering mountains in the distance, ethereal, magical, winter landscape, high detail",
-  "Rushing rapids in a crystal clear river, surrounded by towering trees and lush vegetation, energetic, serene, high detail, river landscape",
-  "Vibrant flower field in full bloom, surrounded by rolling hills and a brilliant blue sky, colorful, serene, high detail, spring landscape",
-  "Glimpses of a herd of wild elephants crossing a savanna, surrounded by tall grass and a brilliant orange sunset, majestic, peaceful, high detail, safari landscape",
-  "Tranquil pond surrounded by tall trees, with a beautiful lily pad garden and calm reflection of the sky, serene, peaceful, high detail, water landscape",
-  "Giant redwoods in a misty forest, with towering trees and lush vegetation, peaceful, serene, high detail, forest landscape",
-  "Breathtaking view of a desert landscape, with towering sand dunes and a brilliant blue sky, serene, vast, high detail, desert landscape",
-  "Stunning sunset over an ocean horizon, with orange and pink hues spreading across the sky, peaceful, serene, high detail, seascape",
-  "Massive airship floating above a sprawling metropolis, with towering skyscrapers and busy streets below, futuristic, detailed, high resolution, urban landscape",
-  "Steampunk submarine exploring a coral reef, surrounded by exotic sea creatures and vibrant coral, detailed, surreal, steampunk style",
-  "Medieval castle on a cliff, surrounded by a moat and rolling green hills, majestic, medieval, high detail",
-  "Towering robot statue in a desolate wasteland, with dust storms and abandoned vehicles in the distance, mechanical, dystopian, high detail, post-apocalyptic landscape",
-  "Massive statue of a dragon in a lush jungle, surrounded by exotic plants and tall trees, mysterious, ancient, high detail, tropical landscape",
-  "Ancient temple in a mountain range, surrounded by misty clouds and tall peaks, mysterious, ancient, high detail",
-  "Crashed spaceship in a dense forest, surrounded by tall trees and exotic vegetation, futuristic, detailed, high detail, sci-fi landscape",
-  "Lighthouse on a stormy beach, surrounded by crashing waves and dramatic clouds, intense, detailed, high detail, coastal landscape",
-  "Steam locomotive in a snowy mountain range, surrounded by tall peaks and crisp clear air, nostalgic, detailed, high detail, winter landscape",
-  "Old western town in the desert, surrounded by towering sand dunes and a brilliant blue sky, nostalgic, detailed, high detail, western landscape",
-  "Futuristic cyborg, sleek metal enhancements and glowing circuits, standing in high-tech laboratory, intense, detailed, high resolution, sci-fi portrait",
-  "Medieval knight, armor and shining sword, standing on battle-scarred battlefield, intense, detailed, high detail, portrait",
-  "Powerful sorceress, flowing robes and mystical staff, standing in dark and ominous forest, mysterious, detailed, high detail, fantasy portrait",
-  "Rogue adventurer, backpack and rugged appearance, standing in dense jungle, adventurous, detailed, high detail, portrait",
-  "Wise wizard, long beard and mysterious tome, standing in dimly lit library, wise, detailed, high detail, portrait",
-  "Daring astronaut, space suit and helmet, standing in front of futuristic spaceship, adventurous, detailed, high detail, portrait",
-  "Skilled archer, bow and quiver of arrows, standing in forest clearing, intense, detailed, high detail, portrait",
-  "Daring treasure hunter, map and compass, standing in desolate desert, adventurous, detailed, high detail, portrait",
-  "Cosmic swirl of stars and galaxies, swirling in endless black void, otherworldly, abstract, high detail, space",
-  "Psychedelic mandala of patterns and shapes, kaleidoscopic, trippy, detailed, abstract art",
-  "Fractal landscape of geometric shapes and patterns, complex, intricate, abstract, digital art",
-  "Glowing nebula of vibrant gas and dust, celestial, otherworldly, abstract, space art",
-  "Hypnotic vortex of swirling colors, intense, detailed, abstract, digital art",
-  "Aurora borealis, vibrant lights dancing in the night sky, ethereal, abstract, high detail, nature art",
-  "Phantasmagoric carnival, carnival attractions shifting and changing, dreamlike, abstract, high detail, surreal art",
-  "Radiant nebula, star clusters and gas clouds shining brightly, celestial, otherworldly, abstract, space art",
-  "Iconic Parisian street with quaint cafes and bistros, charming, romantic, high detail, cityscape",
-  "Golden hour New York City skyline, towering skyscrapers and bustling streets, iconic, dramatic",
-  "Beautiful Venice canals with gondolas and bridges, charming, romantic, high detail, cityscape",
-  "Majestic Machu Picchu, set against a backdrop of towering mountains, breathtaking, high detail, landscape",
-  "Breathtaking view of the Grand Canyon, vast and awe-inspiring, high detail, landscape",
-  "Charming Santorini island with pristine beaches and iconic white buildings, Mediterranean, high detail, seascape",
-  "The iconic Great Wall of China, stretching along the countryside, historical, high detail, landscape",
-  "Stunning view of the Sydney Opera House, with the harbor and cityscape in the background, iconic, high detail, cityscape",
-  "The stunning Taj Mahal, set against a backdrop of lush greenery, historic, high detail, landmark",
-  "Breathtaking view of the Serengeti with roaming wildlife, vast, high detail, nature landscape",
-  "Giant rubber duck floating in the ocean with a small island on its back, surrounded by tropical palm trees and crystal clear water, bright and sunny day, calm seas, vivid colors, cinematic lighting, high detail",
-  "Giant hamster wheel in the middle of a city, with skyscrapers and busy streets in the background, centralized, low details, stylized graphics, night time, lit up, neon lights, no shadows",
-  "Humongous teacup and saucer floating in the sky, surrounded by clouds and rainbows, abstract, surreal, dreamlike, stylized oil painting style, vivid colors, detailed, high resolution, wide angled, otherworldly, fantastic",
-  "Giant ice cream cone melting and creating a river through a city, with boats floating down it, dramatic, intense, chaotic, high detail, fast-paced, wide angled, aerial view, colorful, fun, stylized graphics",
-  "Giant snail racing a car, high speed, intense, dynamic, detailed, cartoon style, wide angled, overhead view, vibrant colors, whimsical, absurd, surreal, fun",
-  "A group of giant robots playing a game of soccer, intense, dynamic, high detail, 3D, stylized, futuristic, metallic, robots, absurd, fantastic, wide angled, overhead view, colorful, fun",
-  "A city built entirely out of food, colorful, detailed, stylized, fun, absurd, vivid, delicious, overhead view, centralized, wide angled, vibrant, fantastical",
-  "Giant fruit and vegetable parade, with various different fruits and vegetables marching down a city street, colorful, detailed, stylized, absurd, fun, vivid, delicious, overhead view, centralized, wide angled",
-  "Giant caterpillar riding a bicycle, surreal, absurd, whimsical, stylized, detailed, vivid, high resolution, centralized, overhead view, colorful, fun",
-  "A bustling city made entirely of candy, with gumdrop buildings and sugar-coated streets, bright colors, whimsical, playful, detailed",
-  "A floating city in the clouds, with airships docking at sky-high platforms and clouds serving as roads, futuristic, whimsical, high-altitude, detailed",
-  "An underground city, filled with steam-powered trains, strange creatures, and intricate tunnels and cave systems, dark, detailed, subterranean, steampunk",
-  "A jungle city, with vines and roots serving as roads and buildings made of leaves, colorful, detailed, natural, tropical",
-  "A city made of ice, with ice slides, frozen rivers, and snow-covered buildings, winter, magical, detailed, cold",
-  "A pirate port, with ships setting sail, blacksmiths crafting weapons, and treasure-filled caves, adventurous, detailed, historic, swashbuckling",
-  "A space station, with spaceships coming and going, astronauts on EVA missions, and maintenance robots hard at work, futuristic, high-tech, detailed, intergalactic",
-  "A dragon's lair, with dragons hoarding treasure, sleeping on piles of gold, and shooting fire from their nostrils, mythical, detailed, adventurous, fantastical",
-  "Bowl of steaming hot ramen with a perfect egg in the center, surrounded by thin slices of meat, green onions, and nori, with a flavorful broth and perfect noodles, high detail, focused on texture and steam",
-  "Delectable pizza with melted cheese, juicy tomato sauce, and an array of toppings including pepperoni, mushrooms, and black olives, served hot and fresh, high resolution, stylized, focused on ingredients and melted cheese",
-  "Sizzling hot sirloin steak with a perfect crust, seared to perfection and topped with herbs and spices, served with a side of roasted vegetables and mashed potatoes, juicy, delicious, high detail",
-  "Assorted fruit platter with ripe, juicy strawberries, sweet grapes, tangy citrus, and juicy watermelon, set on a bed of greens and accented with mint leaves, high resolution, vibrant, natural",
-  "Chocolate cake with rich, fudgy frosting and perfectly layered cake, garnished with fresh berries and drizzled with melted chocolate, decadent, sweet, high detail, food photography",
-  "Baked salmon fillet with a perfectly crispy skin and tender, flaky flesh, served with a side of steamed vegetables and quinoa, healthy, flavorful, high detail, food photography",
-  "Bowl of hearty chili with tender chunks of beef, rich tomato sauce, and a mix of spices, topped with grated cheddar cheese and green onions, high detail, focused on texture and heat, comfort food",
-  "Platter of sushi rolls with colorful and flavorful ingredients, including avocado, tuna, salmon, and crab, arranged with precision and beauty, high resolution, Asian-style, focused on texture and color",
-  "Stuffed bell peppers filled with tender ground beef, flavorful rice, and melted cheese, baked to perfection, juicy, flavorful, high detail, food photography",
-  "Tasty tacos filled with seasoned beef, fresh salsa, melted cheese, and crunchy lettuce, served on a warm corn tortilla, Mexican-style, high resolution, focused on texture and flavor, food photography",
-  "Contemporary living room with large windows overlooking a cityscape, neutral color palette, minimalistic design, sleek modern furniture, gallery wall of abstract art, warm lighting, high detail, open floor plan",
-  "Rustic kitchen with exposed brick wall, reclaimed wood cabinetry, large farmhouse sink, industrial lighting fixtures, antique baking tools on open shelving, cast iron cookware, vintage accents, warm and inviting, detailed textures",
-  "Luxurious bathroom with freestanding bathtub, rain shower, heated flooring, marble tiles, brass fixtures, floating vanity with double sink, elegant chandelier, high contrast lighting, spa-like atmosphere, high resolution",
-  "Elegant dining room with crystal chandelier, dark wood table, velvet upholstered chairs, large statement art piece, tall windows with lush garden view, sophisticated color scheme, detailed textures, candlelit ambiance",
-  "Industrial-style office with concrete floors, raw steel beams, large wooden desk, leather office chair, wall of bookshelves, minimalist design, warm lighting, high detail, organized and professional",
-  "Cozy bedroom with four-poster bed, plush bedding, soft lighting, large windows with natural light, statement wallpaper, decorative throw pillows, high resolution textures, intimate and relaxing",
-  "Glamorous dressing room with large mirror, Hollywood lights, plush velvet seating, glass shelves displaying designer shoes, hanging rods for clothes, detailed textures, high contrast lighting, organized and stylish",
-  "Modern nursery with minimalistic design, white crib, rocking chair, wall mounted bookshelves, abstract art, neutral color palette, warm lighting, high detail, cozy and inviting",
-  "Minimalistic home gym with rubber flooring, wall-mounted TV, weight bench, medicine ball, dumbbells, yoga mats, high-tech equipment, high detail, organized and efficient",
-  "Traditional library with floor-to-ceiling bookcases, rolling ladder, large wooden desk, leather armchair, antique rug, warm lighting, high resolution textures, intellectual and inviting atmosphere",
-  "Contemporary glass and steel building with sleek lines and an innovative facade, surrounded by an urban landscape, modern, high resolution",
-  "Sleek and modern shopping center with an emphasis on natural light, an open-air interior, and eco-friendly features, contemporary, high detail, architectural renderings",
-  "High-rise residential building with a unique form and facade, featuring terraces and stunning views, minimalist, high-end, architectural design",
-  "Contemporary office building with a focus on sustainability, including green roofs and walls, modern, cutting-edge, architectural illustration",
-  "Contemporary cultural center with a distinct form, featuring a vibrant public plaza and innovative exhibition spaces, futuristic, visually stunning, architectural drawings",
-  "Stylish and modern apartment building with a clean, minimalist design and a focus on natural light, contemporary, high-end, architectural rendering",
-  "Innovative mixed-use development featuring an interplay of form and function, with cutting-edge technology and sustainability features, contemporary, visually striking, architectural illustration",
-  "Stylish and contemporary hotel with a unique form and facade, featuring luxury amenities and stunning views, modern, visually stunning",
-  "Sleek and modern shopping mall with a focus on sustainable design, featuring natural light and innovative materials, contemporary, cutting-edge, architectural drawings",
-  "Innovative and contemporary transportation hub with a unique form, featuring cutting-edge technology and sustainable features, modern, visually stunning, architectural illustration",
-  "Race car with sleek design, captured in a high speed motion blur, dramatic lighting and shallow depth of field, motorsports, adrenaline, studio lighting",
-  "Vintage hot rod with custom flame paint job, captured in low key lighting with selective focus on the chrome details, classic car, retro, high resolution",
-  "Luxury sports car with aerodynamic curves, shot in a high contrast, high key lighting with shallow depth of field, exotic, detailed, sporty, studio lighting",
-  "Majestic yacht with sleek lines, captured in a serene sunset light, with a shallow depth of field, boating, lifestyle",
-  "Futuristic flying car with smooth lines, shot in a low light high contrast studio setting, science fiction, cutting edge, high detail, moody atmosphere",
-  "Sturdy pickup truck, captured in a dramatic golden hour light with deep shadows, off-roading, rugged, dramatic, high resolution",
-  "Vintage motorcycle with gleaming chrome and polished leather, captured in soft natural light with selective focus on the engine, classic, retro, detailed",
-  "Luxury sports car with aggressive lines, shot in a high contrast, high key lighting with shallow depth of field, detailed, sporty, sleek, studio lighting",
-  "Massive semi-truck with chrome details, captured in a high key lighting with shallow depth of field, transportation, commercial, detailed, high resolution",
-  "Vintage sports car with classic curves, captured in a moody, low key light with selective focus on the grille, classic, retro, moody, detailed",
-  "Oil painting of a bustling harbor town, with fishing boats, seagulls, and a lighthouse in the background, high contrast, dramatic lighting, heavily textured brushstrokes",
-  "Watercolor painting of a rolling countryside, with fields of flowers, a red barn, and a white picket fence, soft and delicate brushstrokes, pastel colors",
-  "Acrylic painting of a mountain landscape, with a stormy sky and a cabin nestled in the forest, high contrast, bold brushstrokes, high-resolution",
-  "Oil painting of a sunset over the ocean, with orange and pink hues, gently rolling waves, and palm trees, heavily textured brushstrokes, dramatic lighting",
-  "Pen and ink drawing of a cityscape, with tall skyscrapers and bustling city life, high-contrast, detailed linework, dramatic lighting",
-  "Oil painting of a enchanted forest, with glowing mushrooms, fireflies, and a unicorn, soft brushstrokes, pastel colors, dream-like atmosphere",
-  "Watercolor painting of a desert landscape, with sand dunes, mountains, and a blazing sun, soft and delicate brushstrokes, warm and vibrant colors",
-  "Acrylic painting of a futuristic city, with neon lights, advanced technology, and flying cars, bold brushstrokes, high-contrast, highly stylized",
-  "Oil painting of a tranquil lake surrounded by mountains, with a cabin on the shore, boats, and a sunset, heavily textured brushstrokes, warm and vibrant colors",
-  "Pen and ink drawing of a mystical underwater world, with schools of fish, coral reefs, and a mermaid, highly detailed linework, high-contrast, stylized",
-  "Watercolor painting of a rolling countryside, with fields of wheat, a red barn, and a white picket fence, soft and delicate brushstrokes, warm and vibrant colors",
-  "Bronze statue of a powerful warrior, with a sword in hand, chiseled muscles, and a determined expression, highly detailed, dramatic lighting, intense gaze",
-  "Marble statue of a serene goddess, with flowing robes, delicate features, and a tranquil expression, highly detailed, soft lighting, grace and beauty",
-  "Wooden sculpture of a majestic animal, with intricate carving, textured fur, and piercing eyes, highly detailed, natural lighting, raw and powerful",
-  "Stone statue of a mythological creature, with wings, horns, and a fierce expression, highly detailed, dramatic lighting, intense and otherworldly",
-  "Bronze statue of a philosopher, with a wise expression, long beard, and a tome, highly detailed, soft lighting, introspective gaze",
-  "Marble statue of a dancers, with fluid movements, intricate details, and grace, highly detailed, dramatic lighting, intense expression",
-  "Wooden sculpture of a seascape, with waves, boats, and sea creatures, intricate carving, textured surface, high-resolution, natural lighting",
-  "Bronze statue of a king, with regal attire, a crown, and a stern expression, highly detailed, dramatic lighting, commanding presence",
-  "Wooden sculpture of a tree, with intricate branches, textured bark, and a strong trunk, highly detailed, natural lighting, grounding presence"
+  "stunning sunset over a futuristic city, with towering skyscrapers, dramatic clouds, golden-hour lighting, atmospheric",
+  "epic dragon perched atop a cliff, detailed claws, fiery landscape, plumes of smoke",
+  "serene beach with crystal clear water and white sand, tropical palm trees swaying in the breeze, paradise",
+  "post-apocalyptic wasteland with rusted and abandoned vehicles, dust storms and towering dust clouds in the distance, dark, gritty, dramatic",
+  "mysterious underwater world with vibrant coral and a school of colorful fish, sun beams shining through the water, magical, enchanting, otherworldly",
+  "snowy winter wonderland with a lone cabin in the distance, surrounded by frosty trees and fresh snowfall, peaceful, serene",
+  "mysterious and abandoned temple in the jungle, surrounded by lush vegetation and tall trees, ancient, atmospheric",
+  "vibrant and bustling city street, busy traffic, bright lights, fast-paced, intense",
+  "gothic cathedral on a stormy night, lightning illuminating the sky, rain pouring down, epic, dramatic, atmospheric",
+  "fantasy castle on a hilltop, surrounded by rolling hills, breathtaking sunset, magical, enchanting, charming, romantic",
+  "vast desert with sand dunes, a lone oasis in the distance, hot sun, blue sky, peaceful, serene",
+  "beautiful waterfall in a lush jungle, sunlight shining through the trees, tropical, peaceful, serene",
+  "enchanted forest with a babbling brook, glowing fireflies, towering trees, shrouded in mist, magical, ethereal, dreamlike, fantasy",
+  "volcanic island with a boiling crater, clouds of ash rise from the peak, intense, dramatic, atmospheric",
+  "lonely lighthouse on a rocky cliff overlooking the sea, stormy sky, crashing waves, ominous, intense, dramatic, atmospheric",
+  "mysterious underground cave with glowing crystals, underground stream, dark, mysterious, otherworldly",
+  "glowing aurora borealis over a frozen lake, with towering mountains in the distance, ethereal, magical, peaceful, serene",
+  "vibrant flower field in the spring, surrounded by rolling hills, brilliant blue sky, full-bloom, colorful, peaceful, serene",
+  "tranquil pond surrounded by tall trees, with a beautiful lily pad garden and calm reflection of the sky, peaceful, serene",
+  "stunning sunset over an ocean horizon, orange and pink hues spread across the sky, peaceful, serene",
+  "abandoned temple in a mountain range, surrounded by misty clouds and tall peaks, mysterious, ancient",
+  "steam locomotive in a snowy mountain range, surrounded by tall peaks, frosted trees, nostalgic",
+  "radiant nebula, star clusters and gas clouds shining brightly, celestial, otherworldly, abstract, space art",
+  "beautiful Santorini island, iconic white buildings, pristine beach, Mediterranean, charming, romantic",
+  "breathtaking Grand Canyon, vast, awe-inspiring, otherworldly, iconic, historic landmark",
+  "breathtaking Machu Picchu, set against a backdrop of towering mountains, iconic, historic landmark",
+  "iconic New York City skyline, with towering skyscrapers, golden-hour lighting, dramatic, atmospheric",
+  "iconic Great Wall of China, stretching along the countryside, historic landmark",
+  "iconic Sydney Opera House, with the harbor and cityscape in the background, stunning, historic landmark",
+  "iconic Taj Mahal, set against a backdrop of lush greenery, stunning, historic landmark",
+  "bowl of steaming hot ramen with a sliced egg, thin slices of meat, green onions, noodles, chopsticks, solo, minimal",
+  "large pizza with melted cheese, seared pepperoni, crispy crust, solo, minimal",
+  "sizzling hot sirloin steak with a perfect crust, seared to perfection, served with a side of roasted vegetables and mashed potatoes, solo, minimal",
+  "three-tier wedding cake, white frosting, colorful accent flowers, fresh berry garnish, elegant, minimal",
+  "baked salmon fillet with a perfectly crispy skin and flaky flesh, side of steamed vegetables and quinoa, healthy, fresh, solo, minimal",
+  "steaming bowl of hearty chili with tender chunks of beef, rich tomato sauce, topped with grated cheddar cheese and green onions, solo, minimal",
+  "platter of sushi rolls, tuna, salmon, california maki, rainbow, colorful, beautiful arrangement, solo, minimal",
+  "stuffed bell pepper filled with browned ground beef, rice, melted cheese, parsley garnish, solo, minimal",
+  "pair of tacos filled with shredded chicken, red onions, cilantro, a drizzle of cream, white plate, blue tablecloth, solo, minimal",
+  "contemporary living room, floor-to-ceiling windows, neutral color palette, minimalistic design, modern furniture, wall-mounted television, ceiling fan, recessed lighting, open floor plan",
+  "rustic kitchen with reclaimed wood cabinetry, large farmhouse sink, industrial lighting fixtures, open shelving, cast iron cookware, exposed brick wall",
+  "luxurious bathroom with freestanding bathtub, marble tiles, brass fixtures, double-sink floating vanity, spa-like atmosphere",
+  "cozy bedroom with a four-poster bed, decorative throw pillows, plush bedding, single large window with natural light, statement wallpaper, relaxing atmosphere",
+  "formula one race car, aerodynamic design, captured in a high speed motion blur, shallow depth-of-field, dramatic lighting, epic, intense",
+  "luxury supercar with aerodynamic curves, high-key lighting, depth-of-field, exotic",
+  "stunning yacht with sleek lines, golden-hour lighting, depth-of-field, breathtaking"
 ]

data/styles.json

@@ -1,638 +1,136 @@
-[
-  {
-    "id": "twri-ads-automotive",
-    "name": "Ads: Automotive",
-    "prompt": "automotive advertisement style {prompt} . sleek, dynamic, professional, commercial, vehicle-focused, high-resolution, highly detailed",
-    "negative_prompt": "noisy, blurry, unattractive, sloppy, unprofessional"
-  },
-  {
-    "id": "twri-ads-corporate",
-    "name": "Ads: Corporate",
-    "prompt": "corporate branding style {prompt} . professional, clean, modern, sleek, minimalist, business-oriented, highly detailed",
-    "negative_prompt": "noisy, blurry, grungy, sloppy, cluttered, disorganized"
-  },
-  {
-    "id": "twri-ads-fashion-editorial",
-    "name": "Ads: Fashion Editorial",
-    "prompt": "fashion editorial style {prompt} . high fashion, trendy, stylish, editorial, magazine style, professional, highly detailed",
-    "negative_prompt": "outdated, blurry, noisy, unattractive, sloppy"
-  },
-  {
-    "id": "twri-ads-food-photography",
-    "name": "Ads: Food Photography",
-    "prompt": "food photography style {prompt} . appetizing, professional, culinary, high-resolution, commercial, highly detailed",
-    "negative_prompt": "unappetizing, sloppy, unprofessional, noisy, blurry"
-  },
-  {
-    "id": "twri-ads-gourmet-food-photography",
-    "name": "Ads: Gourmet Food Photography",
-    "prompt": "gourmet food photo of {prompt} . soft natural lighting, macro details, vibrant colors, fresh ingredients, glistening textures, bokeh background, styled plating, wooden tabletop, garnished, tantalizing, editorial quality",
-    "negative_prompt": "cartoon, anime, sketch, grayscale, dull, overexposed, cluttered, messy plate, deformed"
-  },
-  {
-    "id": "twri-ads-luxury",
-    "name": "Ads: Luxury",
-    "prompt": "luxury product style {prompt} . elegant, sophisticated, high-end, luxurious, professional, highly detailed",
-    "negative_prompt": "cheap, noisy, blurry, unattractive, amateurish"
-  },
-  {
-    "id": "twri-ads-poster",
-    "name": "Ads: Poster",
-    "prompt": "advertising poster style {prompt} . Professional, modern, product-focused, commercial, eye-catching, highly detailed",
-    "negative_prompt": "noisy, blurry, amateurish, sloppy, unattractive"
-  },
-  {
-    "id": "twri-ads-real-estate",
-    "name": "Ads: Real Estate",
-    "prompt": "real estate photography style {prompt} . professional, inviting, well-lit, high-resolution, property-focused, commercial, highly detailed",
-    "negative_prompt": "dark, blurry, unappealing, noisy, unprofessional"
-  },
-  {
-    "id": "twri-ads-retail",
-    "name": "Ads: Retail",
-    "prompt": "retail packaging style {prompt} . vibrant, enticing, commercial, product-focused, eye-catching, professional, highly detailed",
-    "negative_prompt": "noisy, blurry, amateurish, sloppy, unattractive"
-  },
-  {
-    "id": "twri-art-abstract",
-    "name": "Art: Abstract",
-    "prompt": "abstract style {prompt} . non-representational, colors and shapes, expression of feelings, imaginative, highly detailed",
-    "negative_prompt": "realistic, photographic, figurative, concrete"
-  },
-  {
-    "id": "twri-art-abstract-expressionism",
-    "name": "Art: Abstract Expressionism",
-    "prompt": "abstract expressionist painting {prompt} . energetic brushwork, bold colors, abstract forms, expressive, emotional",
-    "negative_prompt": "realistic, photorealistic, low contrast, plain, simple, monochrome"
-  },
-  {
-    "id": "twri-art-art-deco",
-    "name": "Art: Art Deco",
-    "prompt": "art deco style {prompt} . geometric shapes, bold colors, luxurious, elegant, decorative, symmetrical, ornate, detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, modernist, minimalist"
-  },
-  {
-    "id": "twri-art-art-nouveau",
-    "name": "Art: Art Nouveau",
-    "prompt": "art nouveau style {prompt} . elegant, decorative, curvilinear forms, nature-inspired, ornate, detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, modernist, minimalist"
-  },
-  {
-    "id": "twri-art-constructivist",
-    "name": "Art: Constructivist",
-    "prompt": "constructivist style {prompt} . geometric shapes, bold colors, dynamic composition, propaganda art style",
-    "negative_prompt": "realistic, photorealistic, low contrast, plain, simple, abstract expressionism"
-  },
-  {
-    "id": "twri-art-cubist",
-    "name": "Art: Cubist",
-    "prompt": "cubist artwork {prompt} . geometric shapes, abstract, innovative, revolutionary",
-    "negative_prompt": "anime, photorealistic, 35mm film, deformed, glitch, low contrast, noisy"
-  },
-  {
-    "id": "twri-art-expressionist",
-    "name": "Art: Expressionist",
-    "prompt": "expressionist {prompt} . raw, emotional, dynamic, distortion for emotional effect, vibrant, use of unusual colors, detailed",
-    "negative_prompt": "realism, symmetry, quiet, calm, photo"
-  },
-  {
-    "id": "twri-art-graffiti",
-    "name": "Art: Graffiti",
-    "prompt": "graffiti style {prompt} . street art, vibrant, urban, detailed, tag, mural",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic"
-  },
-  {
-    "id": "twri-art-hyperrealism",
-    "name": "Art: Hyperrealism",
-    "prompt": "hyperrealistic art {prompt} . extremely high-resolution details, photographic, realism pushed to extreme, fine texture, incredibly lifelike",
-    "negative_prompt": "simplified, abstract, unrealistic, impressionistic, low resolution"
-  },
-  {
-    "id": "twri-art-impressionist",
-    "name": "Art: Impressionist",
-    "prompt": "impressionist painting {prompt} . loose brushwork, vibrant color, light and shadow play, captures feeling over form",
-    "negative_prompt": "anime, photorealistic, 35mm film, deformed, glitch, low contrast, noisy"
-  },
-  {
-    "id": "twri-art-pointillism",
-    "name": "Art: Pointillism",
-    "prompt": "pointillism style {prompt} . composed entirely of small, distinct dots of color, vibrant, highly detailed",
-    "negative_prompt": "line drawing, smooth shading, large color fields, simplistic"
-  },
-  {
-    "id": "twri-art-pop-art",
-    "name": "Art: Pop Art",
-    "prompt": "pop Art style {prompt} . bright colors, bold outlines, popular culture themes, ironic or kitsch",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, minimalist"
-  },
-  {
-    "id": "twri-art-psychedelic",
-    "name": "Art: Psychedelic",
-    "prompt": "psychedelic style {prompt} . vibrant colors, swirling patterns, abstract forms, surreal, trippy",
-    "negative_prompt": "monochrome, black and white, low contrast, realistic, photorealistic, plain, simple"
-  },
-  {
-    "id": "twri-art-renaissance",
-    "name": "Art: Renaissance",
-    "prompt": "renaissance style {prompt} . realistic, perspective, light and shadow, religious or mythological themes, highly detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, modernist, minimalist, abstract"
-  },
-  {
-    "id": "twri-art-steampunk",
-    "name": "Art: Steampunk",
-    "prompt": "steampunk style {prompt} . antique, mechanical, brass and copper tones, gears, intricate, detailed",
-    "negative_prompt": "deformed, glitch, noisy, low contrast, anime, photorealistic"
-  },
-  {
-    "id": "twri-art-surrealist",
-    "name": "Art: Surrealist",
-    "prompt": "surrealist art {prompt} . dreamlike, mysterious, provocative, symbolic, intricate, detailed",
-    "negative_prompt": "anime, photorealistic, realistic, deformed, glitch, noisy, low contrast"
-  },
-  {
-    "id": "twri-art-typography",
-    "name": "Art: Typography",
-    "prompt": "typographic art {prompt} . stylized, intricate, detailed, artistic, text-based",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic"
-  },
-  {
-    "id": "twri-art-watercolor",
-    "name": "Art: Watercolor",
-    "prompt": "watercolor painting {prompt} . vibrant, beautiful, painterly, detailed, textural, artistic",
-    "negative_prompt": "anime, photorealistic, 35mm film, deformed, glitch, low contrast, noisy"
-  },
-  {
-    "id": "twri-futuristic-biomechanical",
-    "name": "Futuristic: Biomechanical",
-    "prompt": "biomechanical style {prompt} . blend of organic and mechanical elements, futuristic, cybernetic, detailed, intricate",
-    "negative_prompt": "natural, rustic, primitive, organic, simplistic"
-  },
-  {
-    "id": "twri-futuristic-biomechanical-cyberpunk",
-    "name": "Futuristic: Biomechanical Cyberpunk",
-    "prompt": "biomechanical cyberpunk {prompt} . cybernetics, human-machine fusion, dystopian, organic meets artificial, dark, intricate, highly detailed",
-    "negative_prompt": "natural, colorful, deformed, sketch, low contrast, watercolor"
-  },
-  {
-    "id": "twri-futuristic-cybernetic",
-    "name": "Futuristic: Cybernetic",
-    "prompt": "cybernetic style {prompt} . futuristic, technological, cybernetic enhancements, robotics, artificial intelligence themes",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, historical, medieval"
-  },
-  {
-    "id": "twri-futuristic-cybernetic-robot",
-    "name": "Futuristic: Cybernetic Robot",
-    "prompt": "cybernetic robot {prompt} . android, AI, machine, metal, wires, tech, futuristic, highly detailed",
-    "negative_prompt": "organic, natural, human, sketch, watercolor, low contrast"
-  },
-  {
-    "id": "twri-futuristic-cyberpunk-cityscape",
-    "name": "Futuristic: Cyberpunk Cityscape",
-    "prompt": "cyberpunk cityscape {prompt} . neon lights, dark alleys, skyscrapers, futuristic, vibrant colors, high contrast, highly detailed",
-    "negative_prompt": "natural, rural, deformed, low contrast, black and white, sketch, watercolor"
-  },
-  {
-    "id": "twri-futuristic-general",
-    "name": "Futuristic: General",
-    "prompt": "futuristic style {prompt} . sleek, modern, ultramodern, high tech, detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, vintage, antique"
-  },
-  {
-    "id": "twri-futuristic-retro-cyberpunk",
-    "name": "Futuristic: Retro Cyberpunk",
-    "prompt": "retro cyberpunk {prompt} . 80's inspired, synthwave, neon, vibrant, detailed, retro futurism",
-    "negative_prompt": "modern, desaturated, black and white, realism, low contrast"
-  },
-  {
-    "id": "twri-futuristic-retro-futurism",
-    "name": "Futuristic: Retro Futurism",
-    "prompt": "retro-futuristic {prompt} . vintage sci-fi, 50s and 60s style, atomic age, vibrant, highly detailed",
-    "negative_prompt": "contemporary, realistic, rustic, primitive"
-  },
-  {
-    "id": "twri-futuristic-sci-fi",
-    "name": "Futuristic: Sci-Fi",
-    "prompt": "sci-fi style {prompt} . futuristic, technological, alien worlds, space themes, advanced civilizations",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, historical, medieval"
-  },
-  {
-    "id": "twri-futuristic-vaporwave",
-    "name": "Futuristic: Vaporwave",
-    "prompt": "vaporwave style {prompt} . retro aesthetic, cyberpunk, vibrant, neon colors, vintage 80s and 90s style, highly detailed",
-    "negative_prompt": "monochrome, muted colors, realism, rustic, minimalist, dark"
-  },
-  {
-    "id": "twri-game-bubble-bobble",
-    "name": "Game: Bubble Bobble",
-    "prompt": "Bubble Bobble style {prompt} . 8-bit, cute, pixelated, fantasy, vibrant, reminiscent of Bubble Bobble game",
-    "negative_prompt": "realistic, modern, photorealistic, violent, horror"
-  },
-  {
-    "id": "twri-game-cyberpunk",
-    "name": "Game: Cyberpunk",
-    "prompt": "cyberpunk game style {prompt} . neon, dystopian, futuristic, digital, vibrant, detailed, high contrast, reminiscent of cyberpunk genre video games",
-    "negative_prompt": "historical, natural, rustic, low detailed"
-  },
-  {
-    "id": "twri-game-fighting",
-    "name": "Game: Fighting",
-    "prompt": "fighting game style {prompt} . dynamic, vibrant, action-packed, detailed character design, reminiscent of fighting video games",
-    "negative_prompt": "peaceful, calm, minimalist, photorealistic"
-  },
-  {
-    "id": "twri-game-gta",
-    "name": "Game: GTA",
-    "prompt": "GTA-style artwork {prompt} . satirical, exaggerated, pop art style, vibrant colors, iconic characters, action-packed",
-    "negative_prompt": "realistic, black and white, low contrast, impressionist, cubist, noisy, blurry, deformed"
-  },
-  {
-    "id": "twri-game-mario",
-    "name": "Game: Mario",
-    "prompt": "Super Mario style {prompt} . vibrant, cute, cartoony, fantasy, playful, reminiscent of Super Mario series",
-    "negative_prompt": "realistic, modern, horror, dystopian, violent"
-  },
-  {
-    "id": "twri-game-minecraft",
-    "name": "Game: Minecraft",
-    "prompt": "Minecraft style {prompt} . blocky, pixelated, vibrant colors, recognizable characters and objects, game assets",
-    "negative_prompt": "smooth, realistic, detailed, photorealistic, noise, blurry, deformed"
-  },
-  {
-    "id": "twri-game-pokemon",
-    "name": "Game: Pokémon",
-    "prompt": "Pokémon style {prompt} . vibrant, cute, anime, fantasy, reminiscent of Pokémon series",
-    "negative_prompt": "realistic, modern, horror, dystopian, violent"
-  },
-  {
-    "id": "twri-game-retro",
-    "name": "Game: Retro",
-    "prompt": "retro game art {prompt} . 16-bit, vibrant colors, pixelated, nostalgic, charming, fun",
-    "negative_prompt": "realistic, photorealistic, 35mm film, deformed, glitch, low contrast, noisy"
-  },
-  {
-    "id": "twri-game-retro-arcade",
-    "name": "Game: Retro Arcade",
-    "prompt": "retro arcade style {prompt} . 8-bit, pixelated, vibrant, classic video game, old school gaming, reminiscent of 80s and 90s arcade games",
-    "negative_prompt": "modern, ultra-high resolution, photorealistic, 3D"
-  },
-  {
-    "id": "twri-game-rpg-fantasy",
-    "name": "Game: RPG Fantasy",
-    "prompt": "role-playing game (RPG) style fantasy {prompt} . detailed, vibrant, immersive, reminiscent of high fantasy RPG games",
-    "negative_prompt": "sci-fi, modern, urban, futuristic, low detailed"
-  },
-  {
-    "id": "twri-game-strategy",
-    "name": "Game: Strategy",
-    "prompt": "strategy game style {prompt} . overhead view, detailed map, units, reminiscent of real-time strategy video games",
-    "negative_prompt": "first-person view, modern, photorealistic"
-  },
-  {
-    "id": "twri-game-street-fighter",
-    "name": "Game: Street Fighter",
-    "prompt": "Street Fighter style {prompt} . vibrant, dynamic, arcade, 2D fighting game, highly detailed, reminiscent of Street Fighter series",
-    "negative_prompt": "3D, realistic, modern, photorealistic, turn-based strategy"
-  },
-  {
-    "id": "twri-game-zelda",
-    "name": "Game: Zelda",
-    "prompt": "Legend of Zelda style {prompt} . vibrant, fantasy, detailed, epic, heroic, reminiscent of The Legend of Zelda series",
-    "negative_prompt": "sci-fi, modern, realistic, horror"
-  },
-  {
-    "id": "twri-misc-architectural",
-    "name": "Misc: Architectural",
-    "prompt": "architectural style {prompt} . clean lines, geometric shapes, minimalist, modern, architectural drawing, highly detailed",
-    "negative_prompt": "curved lines, ornate, baroque, abstract, grunge"
-  },
-  {
-    "id": "twri-misc-disco",
-    "name": "Misc: Disco",
-    "prompt": "disco-themed {prompt} . vibrant, groovy, retro 70s style, shiny disco balls, neon lights, dance floor, highly detailed",
-    "negative_prompt": "minimalist, rustic, monochrome, contemporary, simplistic"
-  },
-  {
-    "id": "twri-misc-dreamscape",
-    "name": "Misc: Dreamscape",
-    "prompt": "dreamscape {prompt} . surreal, ethereal, dreamy, mysterious, fantasy, highly detailed",
-    "negative_prompt": "realistic, concrete, ordinary, mundane"
-  },
-  {
-    "id": "twri-misc-dystopian",
-    "name": "Misc: Dystopian",
-    "prompt": "dystopian style {prompt} . bleak, post-apocalyptic, somber, dramatic, highly detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, cheerful, optimistic, vibrant, colorful"
-  },
-  {
-    "id": "twri-misc-fairy-tale",
-    "name": "Misc: Fairy Tale",
-    "prompt": "fairy tale {prompt} . magical, fantastical, enchanting, storybook style, highly detailed",
-    "negative_prompt": "realistic, modern, ordinary, mundane"
-  },
-  {
-    "id": "twri-misc-gothic",
-    "name": "Misc: Gothic",
-    "prompt": "gothic style {prompt} . dark, mysterious, haunting, dramatic, ornate, detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, cheerful, optimistic"
-  },
-  {
-    "id": "twri-misc-grunge",
-    "name": "Misc: Grunge",
-    "prompt": "grunge style {prompt} . textured, distressed, vintage, edgy, punk rock vibe, dirty, noisy",
-    "negative_prompt": "smooth, clean, minimalist, sleek, modern, photorealistic"
-  },
-  {
-    "id": "twri-misc-horror",
-    "name": "Misc: Horror",
-    "prompt": "horror-themed {prompt} . eerie, unsettling, dark, spooky, suspenseful, grim, highly detailed",
-    "negative_prompt": "cheerful, bright, vibrant, light-hearted, cute"
-  },
-  {
-    "id": "twri-misc-kawaii",
-    "name": "Misc: Kawaii",
-    "prompt": "kawaii style {prompt} . cute, adorable, brightly colored, cheerful, anime influence, highly detailed",
-    "negative_prompt": "dark, scary, realistic, monochrome, abstract"
-  },
-  {
-    "id": "twri-misc-lovecraftian",
-    "name": "Misc: Lovecraftian",
-    "prompt": "lovecraftian horror {prompt} . eldritch, cosmic horror, unknown, mysterious, surreal, highly detailed",
-    "negative_prompt": "light-hearted, mundane, familiar, simplistic, realistic"
-  },
-  {
-    "id": "twri-misc-macabre",
-    "name": "Misc: Macabre",
-    "prompt": "macabre style {prompt} . dark, gothic, grim, haunting, highly detailed",
-    "negative_prompt": "bright, cheerful, light-hearted, cartoonish, cute"
-  },
-  {
-    "id": "twri-misc-manga",
-    "name": "Misc: Manga",
-    "prompt": "manga style {prompt} . vibrant, high-energy, detailed, iconic, Japanese comic style",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, Western comic style"
-  },
-  {
-    "id": "twri-misc-metropolis",
-    "name": "Misc: Metropolis",
-    "prompt": "metropolis-themed {prompt} . urban, cityscape, skyscrapers, modern, futuristic, highly detailed",
-    "negative_prompt": "rural, natural, rustic, historical, simple"
-  },
-  {
-    "id": "twri-misc-minimalist",
-    "name": "Misc: Minimalist",
-    "prompt": "minimalist style {prompt} . simple, clean, uncluttered, modern, elegant",
-    "negative_prompt": "ornate, complicated, highly detailed, cluttered, disordered, messy, noisy"
-  },
-  {
-    "id": "twri-misc-monochrome",
-    "name": "Misc: Monochrome",
-    "prompt": "monochrome {prompt} . black and white, contrast, tone, texture, detailed",
-    "negative_prompt": "colorful, vibrant, noisy, blurry, deformed"
-  },
-  {
-    "id": "twri-misc-nautical",
-    "name": "Misc: Nautical",
-    "prompt": "nautical-themed {prompt} . sea, ocean, ships, maritime, beach, marine life, highly detailed",
-    "negative_prompt": "landlocked, desert, mountains, urban, rustic"
-  },
-  {
-    "id": "twri-misc-space",
-    "name": "Misc: Space",
-    "prompt": "space-themed {prompt} . cosmic, celestial, stars, galaxies, nebulas, planets, science fiction, highly detailed",
-    "negative_prompt": "earthly, mundane, ground-based, realism"
-  },
-  {
-    "id": "twri-misc-stained-glass",
-    "name": "Misc: Stained Glass",
-    "prompt": "stained glass style {prompt} . vibrant, beautiful, translucent, intricate, detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic"
-  },
-  {
-    "id": "twri-misc-techwear-fashion",
-    "name": "Misc: Techwear Fashion",
-    "prompt": "techwear fashion {prompt} . futuristic, cyberpunk, urban, tactical, sleek, dark, highly detailed",
-    "negative_prompt": "vintage, rural, colorful, low contrast, realism, sketch, watercolor"
-  },
-  {
-    "id": "twri-misc-tribal",
-    "name": "Misc: Tribal",
-    "prompt": "tribal style {prompt} . indigenous, ethnic, traditional patterns, bold, natural colors, highly detailed",
-    "negative_prompt": "modern, futuristic, minimalist, pastel"
-  },
-  {
-    "id": "twri-misc-zentangle",
-    "name": "Misc: Zentangle",
-    "prompt": "zentangle {prompt} . intricate, abstract, monochrome, patterns, meditative, highly detailed",
-    "negative_prompt": "colorful, representative, simplistic, large fields of color"
-  },
-  {
-    "id": "twri-papercraft-collage",
-    "name": "Papercraft: Collage",
-    "prompt": "collage style {prompt} . mixed media, layered, textural, detailed, artistic",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic"
-  },
-  {
-    "id": "twri-papercraft-flat-papercut",
-    "name": "Papercraft: Flat Papercut",
-    "prompt": "flat papercut style {prompt} . silhouette, clean cuts, paper, sharp edges, minimalist, color block",
-    "negative_prompt": "3D, high detail, noise, grainy, blurry, painting, drawing, photo, disfigured"
-  },
-  {
-    "id": "twri-papercraft-kirigami",
-    "name": "Papercraft: Kirigami",
-    "prompt": "kirigami representation of {prompt} . 3D, paper folding, paper cutting, Japanese, intricate, symmetrical, precision, clean lines",
-    "negative_prompt": "painting, drawing, 2D, noisy, blurry, deformed"
-  },
-  {
-    "id": "twri-papercraft-paper-mache",
-    "name": "Papercraft: Paper Mache",
-    "prompt": "paper mache representation of {prompt} . 3D, sculptural, textured, handmade, vibrant, fun",
-    "negative_prompt": "2D, flat, photo, sketch, digital art, deformed, noisy, blurry"
-  },
-  {
-    "id": "twri-papercraft-paper-quilling",
-    "name": "Papercraft: Paper Quilling",
-    "prompt": "paper quilling art of {prompt} . intricate, delicate, curling, rolling, shaping, coiling, loops, 3D, dimensional, ornamental",
-    "negative_prompt": "photo, painting, drawing, 2D, flat, deformed, noisy, blurry"
-  },
-  {
-    "id": "twri-papercraft-papercut-collage",
-    "name": "Papercraft: Papercut Collage",
-    "prompt": "papercut collage of {prompt} . mixed media, textured paper, overlapping, asymmetrical, abstract, vibrant",
-    "negative_prompt": "photo, 3D, realistic, drawing, painting, high detail, disfigured"
-  },
-  {
-    "id": "twri-papercraft-papercut-shadow-box",
-    "name": "Papercraft: Papercut Shadow Box",
-    "prompt": "3D papercut shadow box of {prompt} . layered, dimensional, depth, silhouette, shadow, papercut, handmade, high contrast",
-    "negative_prompt": "painting, drawing, photo, 2D, flat, high detail, blurry, noisy, disfigured"
-  },
-  {
-    "id": "twri-papercraft-stacked-papercut",
-    "name": "Papercraft: Stacked Papercut",
-    "prompt": "stacked papercut art of {prompt} . 3D, layered, dimensional, depth, precision cut, stacked layers, papercut, high contrast",
-    "negative_prompt": "2D, flat, noisy, blurry, painting, drawing, photo, deformed"
-  },
-  {
-    "id": "twri-papercraft-thick-layered-papercut",
-    "name": "Papercraft: Thick Layered Papercut",
-    "prompt": "thick layered papercut art of {prompt} . deep 3D, volumetric, dimensional, depth, thick paper, high stack, heavy texture, tangible layers",
-    "negative_prompt": "2D, flat, thin paper, low stack, smooth texture, painting, drawing, photo, deformed"
-  },
-  {
-    "id": "twri-photo-alien",
-    "name": "Photo: Alien",
-    "prompt": "alien-themed {prompt} . extraterrestrial, cosmic, otherworldly, mysterious, sci-fi, highly detailed",
-    "negative_prompt": "earthly, mundane, common, realistic, simple"
-  },
-  {
-    "id": "twri-photo-film-noir",
-    "name": "Photo: Film Noir",
-    "prompt": "film noir style {prompt} . monochrome, high contrast, dramatic shadows, 1940s style, mysterious, cinematic",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, vibrant, colorful"
-  },
-  {
-    "id": "twri-photo-glamour",
-    "name": "Photo: Glamour",
-    "prompt": "glamorous photo {prompt} . high fashion, luxurious, extravagant, stylish, sensual, opulent, elegance, stunning beauty, professional, high contrast, detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, distorted, grainy, sketch, low contrast, dull, plain, modest"
-  },
-  {
-    "id": "twri-photo-hdr",
-    "name": "Photo: HDR",
-    "prompt": "HDR photo of {prompt} . High dynamic range, vivid, rich details, clear shadows and highlights, realistic, intense, enhanced contrast, highly detailed",
-    "negative_prompt": "flat, low contrast, oversaturated, underexposed, overexposed, blurred, noisy"
-  },
-  {
-    "id": "twri-photo-iphone-photographic",
-    "name": "Photo: iPhone Photographic",
-    "prompt": "iphone photo {prompt} . large depth of field, deep depth of field, highly detailed",
-    "negative_prompt": "drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly, shallow depth of field, bokeh"
-  },
-  {
-    "id": "twri-photo-long-exposure",
-    "name": "Photo: Long Exposure",
-    "prompt": "long exposure photo of {prompt} . Blurred motion, streaks of light, surreal, dreamy, ghosting effect, highly detailed",
-    "negative_prompt": "static, noisy, deformed, shaky, abrupt, flat, low contrast"
-  },
-  {
-    "id": "twri-photo-neon-noir",
-    "name": "Photo: Neon Noir",
-    "prompt": "neon noir {prompt} . cyberpunk, dark, rainy streets, neon signs, high contrast, low light, vibrant, highly detailed",
-    "negative_prompt": "bright, sunny, daytime, low contrast, black and white, sketch, watercolor"
-  },
-  {
-    "id": "twri-photo-silhouette",
-    "name": "Photo: Silhouette",
-    "prompt": "silhouette style {prompt} . high contrast, minimalistic, black and white, stark, dramatic",
-    "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, color, realism, photorealistic"
-  },
-  {
-    "id": "twri-photo-tilt-shift",
-    "name": "Photo: Tilt-Shift",
-    "prompt": "tilt-shift photo of {prompt} . selective focus, miniature effect, blurred background, highly detailed, vibrant, perspective control",
-    "negative_prompt": "blurry, noisy, deformed, flat, low contrast, unrealistic, oversaturated, underexposed"
-  },
-  {
-    "id": "sai-3d-model",
-    "name": "SAI: 3D Model",
-    "prompt": "professional 3d model {prompt} . octane render, highly detailed, volumetric, dramatic lighting",
-    "negative_prompt": "ugly, deformed, noisy, low poly, blurry, painting"
-  },
-  {
-    "id": "sai-analog-film",
-    "name": "SAI: Analog Film",
-    "prompt": "analog film photo {prompt} . faded film, desaturated, 35mm photo, grainy, vignette, vintage, Kodachrome, Lomography, stained, highly detailed, found footage",
-    "negative_prompt": "painting, drawing, illustration, glitch, deformed, mutated, cross-eyed, ugly, disfigured"
-  },
-  {
-    "id": "sai-anime",
-    "name": "SAI: Anime",
-    "prompt": "anime artwork {prompt} . anime style, key visual, vibrant, studio anime, highly detailed",
-    "negative_prompt": "photo, deformed, black and white, realism, disfigured, low contrast"
-  },
-  {
-    "id": "sai-cinematic",
-    "name": "SAI: Cinematic",
-    "prompt": "cinematic film still {prompt} . shallow depth of field, vignette, highly detailed, high budget, bokeh, cinemascope, moody, epic, gorgeous, film grain, grainy",
-    "negative_prompt": "anime, cartoon, graphic, text, painting, crayon, graphite, abstract, glitch, deformed, mutated, ugly, disfigured"
-  },
-  {
-    "id": "sai-comic-book",
-    "name": "SAI: Comic Book",
-    "prompt": "comic {prompt} . graphic illustration, comic art, graphic novel art, vibrant, highly detailed",
-    "negative_prompt": "photograph, deformed, glitch, noisy, realistic, stock photo"
-  },
-  {
-    "id": "sai-craft-clay",
-    "name": "SAI: Craft Clay",
-    "prompt": "play-doh style {prompt} . sculpture, clay art, centered composition, Claymation",
-    "negative_prompt": "sloppy, messy, grainy, highly detailed, ultra textured, photo"
-  },
-  {
-    "id": "sai-digital-art",
-    "name": "SAI: Digital Art",
-    "prompt": "concept art {prompt} . digital artwork, illustrative, painterly, matte painting, highly detailed",
-    "negative_prompt": "photo, photorealistic, realism, ugly"
-  },
-  {
-    "id": "sai-enhance",
-    "name": "SAI: Enhance",
-    "prompt": "breathtaking {prompt} . award-winning, professional, highly detailed",
-    "negative_prompt": "ugly, deformed, noisy, blurry, distorted, grainy"
-  },
-  {
-    "id": "sai-fantasy-art",
-    "name": "SAI: Fantasy Art",
-    "prompt": "ethereal fantasy concept art of  {prompt} . magnificent, celestial, ethereal, painterly, epic, majestic, magical, fantasy art, cover art, dreamy",
-    "negative_prompt": "photographic, realistic, realism, 35mm film, dslr, cropped, frame, text, deformed, glitch, noise, noisy, off-center, deformed, cross-eyed, closed eyes, bad anatomy, ugly, disfigured, sloppy, duplicate, mutated, black and white"
-  },
-  {
-    "id": "sai-isometric",
-    "name": "SAI: Isometric",
-    "prompt": "isometric style {prompt} . vibrant, beautiful, crisp, detailed, ultra detailed, intricate",
-    "negative_prompt": "deformed, mutated, ugly, disfigured, blur, blurry, noise, noisy, realistic, photographic"
-  },
-  {
-    "id": "sai-line-art",
-    "name": "SAI: Line Art",
-    "prompt": "line art drawing {prompt} . professional, sleek, modern, minimalist, graphic, line art, vector graphics",
-    "negative_prompt": "anime, photorealistic, 35mm film, deformed, glitch, blurry, noisy, off-center, deformed, cross-eyed, closed eyes, bad anatomy, ugly, disfigured, mutated, realism, realistic, impressionism, expressionism, oil, acrylic"
-  },
-  {
-    "id": "sai-lowpoly",
-    "name": "SAI: Lowpoly",
-    "prompt": "low-poly style {prompt} . low-poly game art, polygon mesh, jagged, blocky, wireframe edges, centered composition",
-    "negative_prompt": "noisy, sloppy, messy, grainy, highly detailed, ultra textured, photo"
-  },
-  {
-    "id": "sai-neonpunk",
-    "name": "SAI: Neonpunk",
-    "prompt": "neonpunk style {prompt} . cyberpunk, vaporwave, neon, vibes, vibrant, stunningly beautiful, crisp, detailed, sleek, ultramodern, magenta highlights, dark purple shadows, high contrast, cinematic, ultra detailed, intricate, professional",
-    "negative_prompt": "painting, drawing, illustration, glitch, deformed, mutated, cross-eyed, ugly, disfigured"
-  },
-  {
-    "id": "sai-origami",
-    "name": "SAI: Origami",
-    "prompt": "origami style {prompt} . paper art, pleated paper, folded, origami art, pleats, cut and fold, centered composition",
-    "negative_prompt": "noisy, sloppy, messy, grainy, highly detailed, ultra textured, photo"
-  },
-  {
-    "id": "sai-photographic",
-    "name": "SAI: Photographic",
-    "prompt": "cinematic photo {prompt} . 35mm photograph, film, bokeh, professional, 4k, highly detailed",
-    "negative_prompt": "drawing, painting, crayon, sketch, graphite, impressionist, noisy, blurry, soft, deformed, ugly"
-  },
-  {
-    "id": "sai-pixel-art",
-    "name": "SAI: Pixel Art",
-    "prompt": "pixel-art {prompt} . low-res, blocky, pixel art style, 8-bit graphics",
-    "negative_prompt": "sloppy, messy, blurry, noisy, highly detailed, ultra textured, photo, realistic"
-  },
-  {
-    "id": "sai-texture",
-    "name": "SAI: Texture",
-    "prompt": "texture {prompt} top down close-up",
-    "negative_prompt": "ugly, deformed, noisy, blurry"
+{
+  "_base": {
+    "positive": "professional, perfection, accurate, precision, highly detailed, best quality, masterpiece",
+    "negative": "watermark, trademark, signature, autograph, missing things, anatomically incorrect, structural anomalies, extraneous elements, redundant features, compression artifacts, deformed, mutated, ugly, unattractive, noisy, grainy, blurry, distorted, oversaturated, undersaturated, overexposed, underexposed, amateurish, sloppy, cluttered, low detail, worst quality"
+  },
+  "abstract": {
+    "name": "Abstract",
+    "positive": "({prompt}), in an abstract art style, non-representational colors and shapes, expressive, emotional, imaginative, spontaneous, vibrant",
+    "negative": "({prompt}), discrete, objective, realism, photographic, monochrome, muted"
+  },
+  "anime_shonen": {
+    "name": "Anime: Shonen",
+    "positive": "({prompt}), in a shonen anime style, One Piece, by Eiichiro Oda, manga, action, adventure, heroic, youthful, vibrant, high contrast",
+    "negative": "({prompt}), shojo, seinen, josei, chibi, kawaii, realism, photographic, dull, plain, monochrome, muted, low contrast"
+  },
+  "anime_shojo": {
+    "name": "Anime: Shojo",
+    "positive": "({prompt}), in a shojo anime style, Sailor Moon, by Naoko Takeuchi, manga, kawaii, romantic, emotional, pastel colors, soft lines",
+    "negative": "({prompt}), shonen, seinen, josei, dark, gritty, realism, photographic"
+  },
+  "anime_seinen": {
+    "name": "Anime: Seinen",
+    "positive": "({prompt}), in a seinen anime style, Ghost in the Shell, by Masamune Shirow, manga, adult, mature, dark, gritty, intricate design, dramatic lighting, high contrast",
+    "negative": "({prompt}), shonen, shojo, josei, chibi, kawaii, realism, photographic, dull, plain, low contrast"
+  },
+  "anime_josei": {
+    "name": "Anime: Josei",
+    "positive": "({prompt}), in a josei anime style, Paradise Kiss, by Ai Yazawa, manga, mature, emotional, sophisticated, soft colors, refined lines",
+    "negative": "({prompt}), shonen, shojo, seinen, chibi, dark, gritty, realism, photographic"
+  },
+  "art_deco": {
+    "name": "Art Deco",
+    "positive": "({prompt}), in an art deco style, inspired by Tamara de Lempicka, geometric shapes, bold colors, luxurious, elegant, sleek, streamlined, symmetrical, vibrant, high contrast",
+    "negative": "({prompt}), realism, photographic, organic, natural, monochrome, muted, low contrast"
+  },
+  "biomechanical": {
+    "name": "Biomechanical",
+    "positive": "({prompt}), in a biomechanical style, organic and mechanical, flesh and metal, cybernetic organism, intricate design, futuristic, sci-fi",
+    "negative": "({prompt}), natural, rustic, primitive, medieval"
+  },
+  "cubism": {
+    "name": "Cubism",
+    "positive": "({prompt}), in a cubist style, inspired by Picasso, fragmented shapes and planes, abstract forms, collage",
+    "negative": "({prompt}), discrete, objective"
+  },
+  "cyberpunk": {
+    "name": "Cyberpunk",
+    "positive": "({prompt}), in a cyberpunk style, 2077, synthwave elements, neon lights, digital effects, high-tech, futuristic, dystopian, vibrant, high contrast",
+    "negative": "({prompt}), rustic, primitive, medieval, monochrome, muted, low contrast"
+  },
+  "enhance": {
+    "name": "Enhance",
+    "positive": "({prompt})",
+    "negative": "({prompt})"
+  },
+  "expressionism": {
+    "name": "Expressionism",
+    "positive": "({prompt}), in an expressionist style, energetic brushwork, bold colors, abstract forms, vibrant, expressive, high contrast",
+    "negative": "({prompt}), discrete, objective, realism, photographic, dull, plain, monochrome, muted, low contrast"
+  },
+  "fantasy": {
+    "name": "Fantasy",
+    "positive": "({prompt}), in a fantasy style, digital concept art, by Greg Rutkowski, trending on ArtStation, magical, enchanting, ethereal, dreamlike, graphic, illustration, high contrast",
+    "negative": "({prompt}), realism, photographic, ordinary, mundane, monochrome, muted, low contrast"
+  },
+  "graffiti": {
+    "name": "Graffiti",
+    "positive": "({prompt}), in a graffiti style, street art, creative composition, dynamic lines, spray paint, hip-hop, stylized, bold, vibrant, urban, mural, high contrast",
+    "negative": "({prompt}), dull, plain, monochrome, muted, low contrast"
+  },
+  "line_art": {
+    "name": "Line Art",
+    "positive": "({prompt}), in a line art drawing style, graphic, illustration, sleek, streamlined, centered composition, solo subject, isolated subject, white background, minimalist arrangement",
+    "negative": "({prompt}), off-center, oil, acrylic, watercolor"
+  },
+  "papercraft": {
+    "name": "Papercraft",
+    "positive": "({prompt}), as a papercraft model, Kirigami style, folded paper, papercut, sharp edges, intricate design, 3d, layered, textural, color block, centered composition, minimalist arrangement",
+    "negative": "({prompt}), 2d, flat"
+  },
+  "photography_food": {
+    "name": "Photography: Food",
+    "positive": "({prompt}), food photography style, fresh ingredients, delicious, culinary, real, authentic, macro details, soft natural lighting, high resolution, uhd, centered composition, minimalist arrangement",
+    "negative": "({prompt}), unappetizing, fake, artificial, low resolution"
+  },
+  "photography_hdr": {
+    "name": "Photography: HDR",
+    "positive": "({prompt}), breathtaking HDR photography, high dynamic range, vivid colors, shadows and highlights, dramatic lighting, high contrast, high resolution, uhd",
+    "negative": "({prompt}), flat colors, plain, dull, low dynamic range, low contrast, low resolution"
+  },
+  "photography_iphone": {
+    "name": "Photography: iPhone",
+    "positive": "({prompt}), taken by iPhone ProRAW camera, XDR, Retina, depth-of-field, vivid colors, dynamic range, dramatic lighting, real, authentic, high contrast, high resolution, uhd",
+    "negative": "({prompt}), shallow depth-of-field, bokeh, fake, artificial, low contrast, low resolution"
+  },
+  "photography_iphone_portrait": {
+    "name": "Photography: iPhone Portrait",
+    "positive": "({prompt}), taken by iPhone Portrait Mode, XDR, Retina, shallow depth-of-field, bokeh, vivid colors, dramatic lighting, real, authentic, high contrast, high resolution, uhd",
+    "negative": "({prompt}), fake, artificial, low contrast, low resolution"
+  },
+  "photography_real_estate": {
+    "name": "Photography: Real Estate",
+    "positive": "({prompt}), real estate photography style, on Zillow, inviting, staged, well-lit, real, authentic, high resolution, uhd",
+    "negative": "({prompt}), dark, fake, artificial, low resolution"
+  },
+  "photography_street": {
+    "name": "Photography: Street",
+    "positive": "({prompt}), street photography style, taken on Fujifilm X100V, f2 aperture, 35mm, RAW format, candid, authentic, gritty, urban, high contrast, high resolution, uhd",
+    "negative": "({prompt}), staged, fake, artificial, low contrast, low resolution"
+  },
+  "pointillism": {
+    "name": "Pointillism",
+    "positive": "({prompt}), pointillism style, inspired by Georges Seurat, composed entirely of small colorful dots, intricate design, vibrant",
+    "negative": "({prompt}), line drawing, smooth shading, wide color gamut, dull, plain, monochrome, muted"
+  },
+  "pop_art": {
+    "name": "Pop Art",
+    "positive": "({prompt}), in a pop art style, inspired by Warhol, bright colors, bold outlines, popular culture, kitsch, vibrant, high contrast",
+    "negative": "({prompt}), discrete, objective, dull, plain, monochrome, muted, low contrast"
+  },
+  "render": {
+    "name": "Render",
+    "positive": "({prompt}), Octane render, Unreal Engine, volumetric lighting, ray tracing, ambient occlusion, subsurface scattering, high resolution, uhd",
+    "negative": "({prompt}), glitch, error, painting, sketch, low resolution"
+  },
+  "vaporwave": {
+    "name": "Vaporwave",
+    "positive": "({prompt}), in a vaporwave style, retro aesthetic, neon colors, vibrant, high contrast",
+    "negative": "({prompt}), dark, monochrome, muted, low contrast"
+  },
+  "watercolor": {
+    "name": "Watercolor",
+    "positive": "({prompt}), painted with watercolors, colorful, painterly, artistic, fluid",
+    "negative": "({prompt}), realism, photographic, oil, acrylic, digital"
   }
-]
+}

lib/__init__.py

@@ -1,7 +1,17 @@
 from .config import Config
 from .download import download_repo_files
-from .inference import async_call, generate
+from .inference import generate
 from .loader import Loader
 from .upscaler import RealESRGAN
+from .utils import async_call, load_json, read_file
 
-__all__ = ["Config", "Loader", "RealESRGAN", "async_call", "download_repo_files", "generate"]
+__all__ = [
+    "Config",
+    "Loader",
+    "RealESRGAN",
+    "async_call",
+    "download_repo_files",
+    "generate",
+    "load_json",
+    "read_file",
+]

lib/config.py

@@ -60,7 +60,7 @@ Config = SimpleNamespace(
         "fast_negative",
         "unrealistic_dream",
     ],
-    STYLE="sai-enhance",
+    STYLE="enhance",
     WIDTH=448,
     HEIGHT=576,
     NUM_IMAGES=1,

lib/inference.py

@@ -1,49 +1,24 @@
-import functools
-import inspect
-import json
 import os
 import re
 import time
 from datetime import datetime
 from itertools import product
-from typing import Callable, TypeVar
+from typing import Callable
 
-import anyio
 import numpy as np
 import spaces
 import torch
-from anyio import Semaphore
 from compel import Compel, DiffusersTextualInversionManager, ReturnedEmbeddingsType
 from compel.prompt_parser import PromptParser
 from huggingface_hub.utils import HFValidationError, RepositoryNotFoundError
 from PIL import Image
-from typing_extensions import ParamSpec
 
 from .loader import Loader
+from .utils import load_json
 
 __import__("warnings").filterwarnings("ignore", category=FutureWarning, module="transformers")
 __import__("transformers").logging.set_verbosity_error()
 
-T = TypeVar("T")
-P = ParamSpec("P")
-
-MAX_CONCURRENT_THREADS = 1
-MAX_THREADS_GUARD = Semaphore(MAX_CONCURRENT_THREADS)
-
-with open("./data/styles.json") as f:
-    STYLES = json.load(f)
-
-
-# like the original but supports args and kwargs instead of a dict
-# https://github.com/huggingface/huggingface-inference-toolkit/blob/0.2.0/src/huggingface_inference_toolkit/async_utils.py
-async def async_call(fn: Callable[P, T], *args: P.args, **kwargs: P.kwargs) -> T:
-    async with MAX_THREADS_GUARD:
-        sig = inspect.signature(fn)
-        bound_args = sig.bind(*args, **kwargs)
-        bound_args.apply_defaults()
-        partial_fn = functools.partial(fn, **bound_args.arguments)
-        return await anyio.to_thread.run_sync(partial_fn)
-
 
 def parse_prompt_with_arrays(prompt: str) -> list[str]:
     arrays = re.findall(r"\[\[(.*?)\]\]", prompt)
@@ -64,21 +39,20 @@ def parse_prompt_with_arrays(prompt: str) -> list[str]:
     return prompts
 
 
-def apply_style(prompt, style_id, negative=False):
-    global STYLES
-    if not style_id or style_id == "None":
-        return prompt
-    for style in STYLES:
-        if style["id"] == style_id:
-            if negative:
-                return (
-                    # prepend our negative prompt to the style's negative prompt
-                    f"{prompt}, {style['negative_prompt']}" if prompt else style["negative_prompt"]
-                )
-            else:
-                # inject our positive prompt into the style prompt
-                return style["prompt"].format(prompt=prompt)
-    return prompt
+def apply_style(positive_prompt, negative_prompt, style_id):
+    if style_id.lower() == "none":
+        return (positive_prompt, negative_prompt)
+
+    styles = load_json("./data/styles.json")
+    style = styles.get(style_id)
+    if style is None:
+        return (positive_prompt, negative_prompt)
+
+    style_base = styles.get("_base", {})
+    return (
+        f"{style.get('positive')}, {style_base.get('positive')}".format(prompt=positive_prompt),
+        f"{style.get('negative')}, {style_base.get('negative')}".format(prompt=negative_prompt),
+    )
 
 
 def prepare_image(input, size=None):
@@ -212,7 +186,7 @@ def generate(
     if scale == 4:
         upscaler = loader.upscaler_4x
 
-    # load embeddings and append to negative prompt
+    embeddings_tokens = []
     embeddings_dir = os.path.join(os.path.dirname(__file__), "..", "embeddings")
     embeddings_dir = os.path.abspath(embeddings_dir)
     for embedding in embeddings:
@@ -222,9 +196,7 @@ def generate(
                 pretrained_model_name_or_path=f"{embeddings_dir}/{embedding}.pt",
                 token=f"<{embedding}>",
             )
-            negative_prompt = (
-                f"{negative_prompt}, <{embedding}>" if negative_prompt else f"<{embedding}>"
-            )
+            embeddings_tokens.append(f"<{embedding}>")
         except (EnvironmentError, HFValidationError, RepositoryNotFoundError):
             raise Error(f"Invalid embedding: <{embedding}>")
 
@@ -241,24 +213,27 @@ def generate(
     images = []
     current_seed = seed
 
-    try:
-        styled_negative_prompt = apply_style(negative_prompt, style, negative=True)
-        negative_embeds = compel(styled_negative_prompt)
-    except PromptParser.ParsingException:
-        raise Error("ValueError: Invalid negative prompt")
-
     for i in range(num_images):
         # seeded generator for each iteration
         generator = torch.Generator(device=pipe.device).manual_seed(current_seed)
 
         try:
-            all_positive_prompts = parse_prompt_with_arrays(positive_prompt)
-            prompt_index = i % len(all_positive_prompts)
-            prompt = all_positive_prompts[prompt_index]
-            prompt = apply_style(prompt, style)
-            positive_embeds = compel(prompt)
+            positive_prompts = parse_prompt_with_arrays(positive_prompt)
+            index = i % len(positive_prompts)
+            positive_styled, negative_styled = apply_style(
+                positive_prompts[index],
+                negative_prompt,
+                style,
+            )
+
+            if negative_styled.startswith("(), "):
+                negative_styled = negative_styled[4:]
+
+            if embeddings_tokens:
+                negative_styled += ", " + ", ".join(embeddings_tokens)
+
             positive_embeds, negative_embeds = compel.pad_conditioning_tensors_to_same_length(
-                [positive_embeds, negative_embeds]
+                [compel(positive_styled), compel(negative_styled)]
             )
         except PromptParser.ParsingException:
             raise Error("ValueError: Invalid prompt")

lib/utils.py

@@ -0,0 +1,37 @@
+import functools
+import inspect
+import json
+from typing import Callable, TypeVar
+
+import anyio
+from anyio import Semaphore
+from typing_extensions import ParamSpec
+
+T = TypeVar("T")
+P = ParamSpec("P")
+
+MAX_CONCURRENT_THREADS = 1
+MAX_THREADS_GUARD = Semaphore(MAX_CONCURRENT_THREADS)
+
+
+@functools.lru_cache()
+def load_json(path: str) -> dict:
+    with open(path, "r", encoding="utf-8") as file:
+        return json.load(file)
+
+
+@functools.lru_cache()
+def read_file(path: str) -> str:
+    with open(path, "r", encoding="utf-8") as file:
+        return file.read()
+
+
+# like the original but supports args and kwargs instead of a dict
+# https://github.com/huggingface/huggingface-inference-toolkit/blob/0.2.0/src/huggingface_inference_toolkit/async_utils.py
+async def async_call(fn: Callable[P, T], *args: P.args, **kwargs: P.kwargs) -> T:
+    async with MAX_THREADS_GUARD:
+        sig = inspect.signature(fn)
+        bound_args = sig.bind(*args, **kwargs)
+        bound_args.apply_defaults()
+        partial_fn = functools.partial(fn, **bound_args.arguments)
+        return await anyio.to_thread.run_sync(partial_fn)