app.py

import random
import gradio as gr


# Dataset with essential elements for prompts
fashion_elements = {
    "hairstyles": [
        "long wavy auburn hair with elegant curls",
        "long straight platinum blonde hair flowing down",
        "long curly deep brown hair with soft waves",
        "long layered red hair with voluminous style",
        "long sleek black hair with glossy finish",
        "long beach wave golden blonde hair",
        "long braided chestnut hair with loose ends",
        "intricate updo with honey highlights",
        "long messy mahogany hair",
        "long straight cherry red hair",
        "long braided platinum hair with twists",
        "long updo with strawberry blonde",
        "long ponytail with sandy blonde"
    ],
    "lace_lingerie": [
        "black lace panties",
        "red lace panties with scalloped edges",
        "white lace panties with delicate trim",
        "pink lace panties with rose patterns",
        "blue lace panties with elegant design",
        "green lace panties with leaf motif",
        "purple lace panties with ornate details",
        "champagne lace panties with subtle sheen",
        "rose gold lace panties with metallic threads",
        "burgundy lace panties with intricate patterns"
    ],
    "dresses_and_skirts": [
        "a short black satin skirt and a white blouse",
        "a tight red mini dress with a side slit",
        "a flowy silk dress with a high thigh slit",
        "a denim mini skirt and a crop top",
        "a short tennis skirt and a tight-fitting polo shirt",
        "a pleated wool mini skirt and an oversized sweater"
    ],
    "boudoir_office_poses": [
        "legs semi-open with slight torso twist, hands gently touching skirt or thong, confident and direct gaze or slightly flirty",
        "sitting on desk edge with one leg crossed over the other, skirt delicately lifted with one hand, soft and confident gaze",
        "gently leaning over the desk with semi-open legs, one hand on the table and the other subtly lifting skirt, intimate yet professional gaze",
        "walking with skirt slightly lifted, one leg forward in a natural step, hands adjusting accessories, focused gaze",
        "standing next to the desk with semi-open legs, one hand on the table while the other subtly touches the skirt or dress, confident or slightly flirty gaze",
        "standing with one leg forward and skirt subtly moved, weight on the back leg with the skirt sliding to slightly show the lingerie, hand on hip, direct and confident gaze",
        "leaning on a backward-turned chair, semi-open legs, skirt smoothly lifted with hands on waist or on the fabric, confident expression and direct gaze at the lens",
        "crossed legs with the skirt shifted, ensuring the posture is natural and feminine, soft and suggestive gaze",
        "one leg slightly elevated and resting against a surface, hand touching the skirt or thong, seductive gaze at the camera",
        "hands on the waist subtly lifting the garment to delicately show the thong, upright posture and confident gaze",
        "standing with hands behind back, torso slightly arched, semi-open legs showing the silhouette, fixed gaze at the lens",
        "walking with a soft step, skirt swaying and slightly lifted, one hand playing with the skirt's edge, confident expression",
        "sitting on an office chair, knees crossed, hands resting on the backrest, relaxed and attractive gaze",
        "standing with torso twisted and one leg crossed in front of the other, hand holding the skirt to hint at the thong, mysterious gaze",
        "leaning over the desk, semi-open legs, arms extended forward, intense and penetrating gaze",
        "sitting slightly inclined with one leg stretched and the other bent, hands on the waist, a confident but sweet expression",
        "arching her back while stretching, outfit lifting to reveal thong, looking back with intense desire",
        "leaning back on an office surface, legs subtly spread hinting thong, eyes locked with a subtle ecstatic gesture",
        "kneeling on the office floor, skirt riding up to tease thong, gazing upwards with desire",
        "sitting on a high office stool, legs crossed and skirt hiked up to show thong, direct gaze with a subtle moan-like expression",
        "bending over an office railing, outfit stretching to reveal thong, looking back with playful desire",
        "reclining on a sofa in the office, legs positioned to hint at thong, eyes conveying subtle ecstasy",
        "standing with hands on hips, skirt swaying to reveal thong, direct eye contact with passionate desire",
        "sitting on the edge of a sofa, adjusting stockings to tease thong view, gazing with desire",
        "reaching for something high on a shelf, outfit lifting to hint at thong, turning head with a desirous look",
        "dancing sensually, twirl causing skirt to lift, thong visible momentarily, gazing passionately",
        "applying lipstick in mirror, bending slightly to reveal thong, glancing at viewer with desire",
        "reading a document in a provocative position, legs positioned to hint at thong, looking up with a subtle flush",
        "heating food with playful movements, bending to the microwave, thong hinted, turning with a desirous smile",
        "exercising with a deep stretch pose, outfit hinting thong, eyes meeting viewer",
        "brushing long hair, movement causing outfit to shift, thong tease, intense gaze",
        "talking on the phone with a seductive posture, pacing, skirt swaying to reveal thong, desirous expression",
        "watering office plants with a bend and reach, thong visible, looking back with a subtle gesture",
        "trying on heels, sitting and adjusting, thong hinted, direct seductive look",
        "folding clothes with a suggestive bend, movement revealing thong, eyes with desire",
        "working at the desk with a relaxed posture, leaning back, legs apart subtly hinting thong, gazing",
        "standing tall with one hand on hip, shoulders relaxed, soft gaze",
        "seated sideways on an office chair, legs crossed elegantly, looking over shoulder",
        "lying on side on a sofa with head propped on hand, legs slightly bent, serene expression",
        "leaning against the office wall with one foot up, hands running through hair, subtle smile",
        "curled up on a sofa, hugging knees, eyes gently closed",
        "lying on back on a sofa with arms above head, arching slightly, eyes on camera",
        "seated on the floor with legs extended, torso twisted to camera, soft smile",
        "standing with back to camera, looking over shoulder with a sultry look",
        "resting chin on hand, elbow on knee, casual yet elegant gaze",
        "hands gently framing face, soft tousled hair, relaxed posture",
        "lying on stomach with crossed ankles, chin resting on hands, playful smile",
        "standing with arms above head, elongating body, peaceful expression",
        "seated on edge of a sofa, one leg crossed over the other, eyes downward",
        "leaning forward on a chair, elbows on knees, intense gaze",
        "standing in a flowing robe or blazer, holding belt loosely, inviting expression",
        "sitting with legs tucked under, looking sideways, soft smile",
        "lying diagonally on a sofa, arm stretched out, relaxed and elegant",
        "standing with hands clasped behind back, head tilted, soft gaze",
        "sitting with knees bent, arms wrapped around legs, contemplative look",
        "semi-open legs with weight on one, the other slightly ahead and relaxed, arms gently crossed or hands on waist, confident and direct gaze",
        "hands on hips for a firm stance, legs shoulder-width apart, head slightly tilted with a seductive expression",
        "semi-open with one bent leg, one leg straight with weight, the other with a slightly bent knee and pointed toe, relaxed arms, intense gaze",
        "legs slightly separated, with one leg slightly ahead pointing outward, hands resting on thighs or lower part of the skirt, elegant and confident expression",
        "legs semi-open, with arms crossed or gently hugging the body for a sophisticated and contained sensuality",
        "semi-open legs with a twisted torso, turning the body and head slightly to soften the pose and add dynamism, with a gaze towards the lens or profile",
        "squatting with semi-open legs and hands resting on thighs, creating a visual line that hints at sensuality and elegance",
        "squatting with hands gently on thighs or slightly to the sides, touching the skirt or thong with delicacy for a tactile feel",
        "torso slightly inclined forward or to the side to add dynamism and flatter body lines",
        "head and gaze oriented toward the camera with a soft and confident expression for an intimate connection",
        "squatting and resting on heels, legs slightly separated, arm resting on the floor or thigh, head slightly tilted for a seductive gaze",
        "squatting with arms around legs, gently holding them or ankles, with direct eye contact or a low gaze for vulnerability or mystery",
        "squatting with elevated hands, hands resting behind or elevated to touch hair or neck, body arched and gaze fixed on the lens for a sophisticated and sensual pose",
        "for a strong connection, the gaze should be direct to the lens, or slightly to the side for a subtle mystery",
        "expressions should be natural, confident, and relaxed, avoiding tension",
        "hands should be soft and natural, gently touching garments or accessories",
        "hands resting on hips or thighs",
        "fluid postures that guide attention, avoiding stiffness"
    ],
        "expressions": [
        "seductive gaze with desire",
        "soft and confident gaze",
        "intimate yet professional gaze",
        "focused gaze",
        "confident or slightly flirty gaze",
        "direct and confident gaze",
        "mysterious gaze",
        "soft and suggestive gaze",
        "seductive gaze at the camera",
        "confident gaze",
        "confident expression",
        "attractive gaze",
        "penetrating gaze",
        "sweet expression",
        "intense desire",
        "subtle ecstatic gesture",
        "gazing upwards with desire",
        "subtle moan-like expression",
        "playful desire",
        "subtle ecstasy",
        "passionate desire",
        "gazing with desire",
        "desirous look",
        "passionately gazing",
        "desire",
        "subtle flush",
        "desirous smile",
        "intense gaze",
        "desirous expression",
        "subtle gesture",
        "direct seductive look",
        "eyes with desire",
        "gazing",
        "soft gaze",
        "seductive expression",
        "intense gaze",
        "serene expression",
        "subtle smile",
        "eyes gently closed",
        "eyes on camera",
        "soft smile",
        "sultry look",
        "elegant gaze",
        "relaxed posture",
        "playful smile",
        "peaceful expression",
        "eyes downward",
        "intense gaze",
        "inviting expression",
        "contemplative look",
        "soft gaze",
        "seductive gaze",
        "intense gaze",
        "elegant expression",
        "gaze towards the lens or profile",
        "seductive gaze",
        "natural, confident, and relaxed expression",
        "soft and confident expression",
        "direct eye contact or a low gaze",
        "direct gaze on the lens, or slightly to the side",
        "soft and confident expression for an intimate connection",
        "natural, confident, and relaxed expression, avoiding tension"
    ],
    "poses_thongs": [
        "arching her back while stretching, outfit lifting to reveal thong, looking back with intense desire",
        "leaning back on a surface, legs subtly spread hinting thong, eyes locked with subtle orgasmic gesture",
        "sipping coffee lounging to reveal thong, subtle ecstasy"
    ],
    "lingerie_sets": [
        "black lace set with push-up bra, floral thong, and sheer thigh-high stockings with a lace top",
        "red satin set with underwire bra, bow accents thong, and fishnet thigh-high stockings with a pattern",
        "white mesh set with sheer-cup bra, transparent thong, and silk thigh-high stockings with a smooth finish",
        "pink lace set with demi-cup bra, scalloped-edge thong, and thigh-high stockings with a garter",
        "blue silk set with smooth-cup bra, a sleek thong, and velvet thigh-high stockings with a soft texture",
        "green velvet set with padded bra, a textured thong, and satin thigh-high stockings with a subtle shine",
        "purple satin set with balconette bra, a glossy thong, and mesh thigh-high stockings with a net pattern",
        "yellow chiffon set with a light-coverage bra, a light thong, and thigh-high stockings with a light, airy feel",
        "black mesh set with transparent-strap bra, a net-pattern thong, and opaque thigh-high stockings with a matte look",
        "red lace set with embroidered bra, an embroidered thong, and sheer thigh-high stockings with a red tint",
        "white satin set with pearl-accented bra, a pearl-detailed thong, and fishnet thigh-high stockings with a diamond pattern",
        "pink silk set with delicate-lace-trim bra, a delicate-strap thong, and shiny silk thigh-high stockings",
        "blue velvet set with a plush bra, a plush thong, and lace thigh-high stockings with a floral design",
        "green lace set with a floral-pattern bra, a leaf-motif thong, and plush velvet thigh-high stockings",
        "purple mesh set with sheer-panel bra, sheer-panel thong, and luxurious satin thigh-high stockings",
        "yellow satin set with a shiny bra, a glossy thong, and transparent mesh thigh-high stockings",
        "black chiffon set with a flowy-edged bra, a flowy-edged thong, and flowy chiffon thigh-high stockings",
        "red mesh set with a bold bra, a bold-line thong, and opaque thigh-high stockings with a bold look",
        "white lace set with a romantic bra, a romantic thong, and sheer thigh-high stockings for a subtle touch",
        "pink velvet set with a soft-textured bra, a soft-pile thong, and fun fishnet thigh-high stockings",
        "blue satin set with a sleek-line bra, a sleek thong, and elegant silk thigh_high stockings",
        "green silk set with an elegant-feel bra, an elegant-drape thong, and romantic lace thigh_high stockings",
        "purple chiffon set with an airy bra, an airy thong, and deep velvet thigh_high stockings",
        "yellow lace set with a bright bra, a bright thong, and bright satin thigh_high stockings",
        "black satin set with a luxurious bra, a luxurious thong, and edgy mesh thigh_high stockings",
        "red velvet set with a deep-hued bra, a deep-colored thong, and airy chiffon thigh_high stockings",
        "white mesh set with a modern bra, a modern-grid thong, and clean opaque thigh_high stockings",
        "pink chiffon set with a feminine bra, a feminine-ruffle thong, and blush-pink sheer thigh_high stockings",
        "blue lace set with a wave-pattern bra, a wave-pattern thong, and playful fishnet thigh_high stockings",
        "green satin set with a natural-color bra, a natural-hued thong, and natural-color silk thigh_high stockings",
        "purple velvet set with a royal-style bra, a royal-vibe thong, and ornate lace thigh_high stockings",
        "yellow mesh set with a sunny-transparency bra, a sunny thong, and sunny velvet thigh_high stockings",
        "black lace set with a gothic bra, a gothic-element thong, and thigh-high stockings with a shadow-black lace top",
        "red silk set with a passionate bra, a passionate thong, and passionate-red mesh thigh-high stockings",
        "white chiffon set with a pure-white bra, a pure-white thong, and pure-white chiffon thigh-high stockings",
        "pink satin set with a blush-pink bra, a blush-pink thong, and soft opaque thigh-high stockings",
        "blue velvet set with a deep-blue bra, a deep-blue thong, and sheer thigh-high stockings in a calm blue",
        "green lace set with an envious-green bra, an envious-green thong, and green fishnet thigh-high stockings",
        "purple mesh set with a mysterious bra, a mysterious thong, and lavender fishnet thigh-high stockings",
        "yellow chiffon set with a joyful-yellow bra, a joyful-yellow thong, and sunny-yellow silk thigh-high stockings",
        "black silk set with a night-black bra, a night-black thong, and shadow-black lace thigh-high stockings",
        "red satin set with a fire-red bra, a fire-red thong, and deep-red velvet thigh-high stockings",
        "white velvet set with a snow-white bra, a snow-white thong, and white-sheer thigh-high stockings",
        "pink lace set with a rose-pink bra, a rose-pink thong, and pink-rose lace thigh-high stockings",
        "blue mesh set with a sky-blue bra, a sky-blue thong, and sky-blue mesh thigh-high stockings",
        "green chiffon set with a forest-green bra, a forest-green thong, and forest-green chiffon thigh-high stockings",
        "purple satin set with a lavender bra, a lavender thong, and lavender lace thigh-high stockings",
        "yellow velvet set with a sun-yellow bra, a sun-yellow thong, and sun-yellow velvet thigh-high stockings",
        "black mesh set with a shadow-black bra, a shadow-black thong, and shadow-black mesh thigh-high stockings",
        "red chiffon set with a sunset-red bra, a sunset-red thong, and sunset-red chiffon thigh-high stockings"
    ],
    "colors": [
        "deep ruby red", "classic black", "emerald green", "sapphire blue", "royal purple",
        "champagne gold", "silver metallic", "burgundy wine", "chocolate brown", "ivory white",
        "hot pink", "electric blue", "vibrant orange", "forest green", "midnight navy",
        "soft lavender", "peachy coral", "platinum silver", "rose gold", "caramel nude"
    ],
    "outfits": [
        "a sleek fitted mini dress with a subtle texture",
        "a stylish A-line mini skirt paired with a delicate blouse",
        "a form-fitting bodycon dress with elegant draping",
        "a sophisticated sheath mini dress with unique detailing",
        "a playful skater dress with a flattering silhouette",
        "a chic wrap mini dress that accentuates the waist",
        "a trendy slip dress with a luxurious satin finish",
        "a tailored pencil mini skirt with a modern cut",
        "a flowing chiffon mini dress with intricate patterns",
        "a form-fitting leather mini skirt with bold accents",
    ],
    "dresses": [
        "black sheath mini dress with lace inserts",
        "red bodycon dress with plunging neckline",
        "white slip dress with spaghetti straps",
        "pink halter mini dress with ruffles",
        "blue wrap dress with belt tie",
        "green off-shoulder dress with puff sleeves",
        "purple A-line dress with pleats",
        "yellow sundress with floral print",
        "black cocktail dress with sequins",
        "red evening dress with slit",
        "white lace dress with scalloped hem",
        "pink skater dress with full skirt",
        "blue maxi dress with side slits",
        "green shift dress with pockets",
        "purple tunic dress with embroidery",
        "yellow babydoll dress with empire waist",
        "black velvet dress with choker neck",
        "red satin dress with cowl neck",
        "white chiffon dress with layers",
        "pink leather dress with zip front",
        "blue denim dress with buttons",
        "green sequin dress with fringe",
        "purple tulle dress with tutu skirt",
        "yellow mesh dress with underlay",
        "black plaid dress with collar",
        "red latex dress with shine",
        "white satin dress with bow back",
        "pink velvet dress with ruche",
        "blue lace dress with illusion neck",
        "green chiffon dress with drape",
        "purple denim dress with distress",
        "yellow leather dress with seams",
        "black sequin dress with sparkle",
        "red tulle dress with volume",
        "white mesh dress with transparency",
        "pink plaid dress with pattern",
        "blue satin dress with gloss",
        "green lace dress with romance",
        "purple velvet dress with luxury",
        "yellow chiffon dress with flow",
        "black denim dress with casual",
        "red leather dress with edge",
        "white sequin dress with party",
        "pink tulle dress with princess",
        "blue velvet dress with texture",
        "green plaid dress with preppy",
        "purple lace dress with gothic",
        "yellow mesh dress with sporty",
        "black satin dress with elegance",
        "red chiffon dress with boho"
    ],
    "lingerie_details": [
        "delicate lace trim",
        "satin ribbon accents",
        "strategic sheer panels",
        "scalloped edges",
        "embellished with subtle crystals",
        "bow details",
        "plunging neckline",
        "underwire support",
        "removable garter straps"
    ],
    "camera_angles": [
        "extreme low angle from floor level, gently tilted upward, filling the frame closely",
        "worm's-eye view, gently tilted upward, filling the frame closely",
        "low angle capturing the full figure from below, gently tilted upward, filling the frame closely",
        "dramatic low angle emphasizing height and form, gently tilted upward, filling the frame closely",
        "low angle close-up focusing on lower body and outfit from below, gently tilted upward, filling the frame closely",
        "low angle medium shot capturing the figure from below, gently tilted upward, filling the frame closely",
    ],
        "miniskirts": [
        "black leather pleated miniskirt with silver zipper",
        "red velvet ultra-short miniskirt with lace hem",
        "white denim tight miniskirt with distressed edges",
        "pink satin flared miniskirt with bow details",
        "blue plaid schoolgirl miniskirt with buttons",
        "green sequin shiny miniskirt with fringe",
        "purple tulle layered miniskirt with ruffles",
        "yellow chiffon flowy miniskirt with slits",
        "black mesh transparent miniskirt with embroidery",
        "red latex shiny miniskirt with buckles",
        "white lace delicate miniskirt with floral patterns",
        "blue denim cutoff miniskirt with pockets",
        "green velvet wrap miniskirt with ties",
        "pink leather high-waist miniskirt with studs",
        "purple satin asymmetric miniskirt with drape",
        "yellow plaid pleated miniskirt with suspenders",
        "black chiffon sheer miniskirt with lining",
        "red sequin glittering miniskirt with hemline",
        "white tulle ballerina miniskirt with layers",
        "blue leather bodycon miniskirt with seams",
        "green lace romantic miniskirt with scallop",
        "pink denim distressed miniskirt with fray",
        "purple velvet ruched miniskirt with drawstring",
        "yellow satin shiny miniskirt with pleats",
        "black plaid tartan miniskirt with kilt pin",
        "red mesh fishnet miniskirt with overlay",
        "white chiffon boho miniskirt with embroidery",
        "blue sequin party miniskirt with sparkle",
        "green tulle voluminous miniskirt with petticoat",
        "pink lace victorian miniskirt with ribbons",
        "purple denim acid-wash miniskirt with patches",
        "yellow leather moto miniskirt with zippers",
        "black satin elegant miniskirt with slit",
        "red velvet holiday miniskirt with trim",
        "white plaid preppy miniskirt with buttons",
        "blue chiffon summer miniskirt with flow",
        "green sequin festival miniskirt with shine",
        "pink tulle princess miniskirt with bows",
        "purple lace gothic miniskirt with corset",
        "yellow mesh sporty miniskirt with stripes",
        "black denim classic miniskirt with belt",
        "red leather edgy miniskirt with chains",
        "white satin bridal miniskirt with veil",
        "blue velvet luxurious miniskirt with texture",
        "green plaid highland miniskirt with pattern",
        "pink chiffon romantic miniskirt with ruffles",
        "purple sequin disco miniskirt with glitter",
        "yellow tulle fun miniskirt with colors",
        "black lace sexy miniskirt with transparency",
        "red denim casual miniskirt with wash"
    ],
    "stockings": [
        "sheer thigh-high stockings with a lace top",
        "fishnet thigh-high stockings with a pattern",
        "silk thigh-high stockings with a smooth finish",
        "thigh-high stockings with a garter",
        "velvet thigh-high stockings with a soft texture",
        "satin thigh-high stockings with a subtle shine",
        "mesh thigh-high stockings with a net pattern",
        "thigh-high stockings with a light, airy feel",
        "opaque thigh-high stockings with a matte look",
        "sheer thigh-high stockings with a red tint",
        "fishnet thigh-high stockings with a diamond pattern",
        "shiny silk thigh-high stockings",
        "lace thigh-high stockings with a floral design",
        "plush velvet thigh-high stockings",
        "luxurious satin thigh-high stockings",
        "transparent mesh thigh-high stockings",
        "flowy chiffon thigh-high stockings",
        "opaque thigh-high stockings with a bold look",
        "sheer thigh-high stockings for a subtle touch",
        "fun fishnet thigh-high stockings",
        "elegant silk thigh-high stockings",
        "romantic lace thigh-high stockings",
        "deep velvet thigh-high stockings",
        "bright satin thigh-high stockings",
        "edgy mesh thigh-high stockings",
        "airy chiffon thigh-high stockings",
        "clean opaque thigh-high stockings",
        "blush-pink sheer thigh-high stockings",
        "playful fishnet thigh-high stockings",
        "natural-color silk thigh-high stockings",
        "ornate lace thigh-high stockings",
        "sunny velvet thigh-high stockings",
        "thigh-high stockings with a shadow-black lace top",
        "passionate-red mesh thigh-high stockings",
        "pure-white chiffon thigh-high stockings",
        "soft opaque thigh-high stockings",
        "sheer thigh-high stockings in a calm blue",
        "green fishnet thigh-high stockings",
        "lavender fishnet thigh-high stockings",
        "sunny-yellow silk thigh-high stockings",
        "shadow-black lace thigh-high stockings",
        "deep-red velvet thigh-high stockings",
        "white-sheer thigh-high stockings",
        "pink-rose lace thigh-high stockings",
        "sky-blue mesh thigh-high stockings",
        "forest-green chiffon thigh-high stockings",
        "lavender lace thigh-high stockings",
        "sun-yellow velvet thigh-high stockings",
        "shadow-black mesh thigh-high stockings",
        "sunset-red chiffon thigh-high stockings"
    ],
    "lingerie_thongs": [
        "floral thong", "bow accents thong", "transparent thong", "scalloped-edge thong",
        "a sleek thong", "a textured thong", "a glossy thong", "a light thong",
        "a net-pattern thong", "an embroidered thong", "a pearl-detailed thong",
        "a delicate-strap thong", "a plush thong", "a leaf-motif thong",
        "sheer-panel thong", "a shiny thong", "a flowy-edged thong", "a bold-line thong",
        "a romantic thong", "a soft-pile thong", "a sleek thong", "an elegant-drape thong",
        "an airy thong", "a bright thong", "a luxurious thong", "a deep-colored thong",
        "a modern-grid thong", "a feminine-ruffle thong", "a wave-pattern thong",
        "a natural-hued thong", "a royal-vibe thong", "a sunny thong", "a gothic-element thong",
        "a passionate thong", "a pure-white thong", "a blush-pink thong", "a deep-blue thong",
        "an envious-green thong", "a mysterious thong", "a joyful-yellow thong",
        "a night-black thong", "a fire-red thong", "a snow-white thong", "a rose-pink thong",
        "a sky-blue thong", "a sky-blue thong", "a forest-green thong", "a lavender thong", "a sun-yellow thong",
        "a shadow-black thong", "a sunset-red thong"
    ],
    "lingerie_bras": [
        "push-up bra", "underwire bra", "sheer-cup bra", "demi-ccup bra", "smooth-cup bra",
        "padded bra", "balconette bra", "a light-coverage bra", "transparent-strap bra",
        "embroidered bra", "pearl-accented bra", "delicate-lace-trim bra", "a plush bra",
        "a floral-pattern bra", "sheer-panel bra", "a shiny bra", "a flowy-edged bra",
        "a bold bra", "a romantic bra", "a soft-textured bra", "a sleek-line bra",
        "an elegant-feel bra", "an airy bra", "a bright bra", "a luxurious bra",
        "a deep-hued bra", "a modern bra", "a feminine bra", "a wave-pattern bra",
        "a natural-color bra", "a royal-style bra", "a sunny-transparency bra", "a gothic bra",
        "a passionate bra", "a pure-white bra", "a blush-pink bra", "a deep-blue bra",
        "an envious-green bra", "a mysterious bra", "a joyful-yellow bra", "a night-black bra",
        "a fire-red bra", "a snow-white bra", "a rose-pink bra", "a sky-blue bra",
        "a forest-green bra", "a lavender bra", "a sun-yellow bra", "a shadow-black bra",
        "a sunset-red bra"
    ],
    "backgrounds": [
        "a luxurious bedroom", "a modern office", "a chic studio apartment", "a dimly lit lounge",
        "a stylish hotel room", "an industrial loft", "a minimalist living space", "a Parisian balcony",
        "a vintage boudoir", "a serene garden setting"
    ],
    "lighting": [
        "soft natural window light", "dramatic chiaroscuro lighting", "warm golden hour glow",
        "studio lighting with softboxes", "ambient moody lighting", "rim lighting to accentuate silhouette",
        "cinematic lighting with subtle color grading", "ethereal backlight", "dynamic spotlighting"
    ]
}

class FashionPromptGenerator:
    def __init__(self):
        self.fashion_elements = fashion_elements

    def generate_prompt(self, celebrity_name):
        primary_color = random.choice(self.fashion_elements["colors"])
        outfit = random.choice(self.fashion_elements["outfits"])
        thong = random.choice(self.fashion_elements["lingerie_thongs"])
        bra = random.choice(self.fashion_elements["lingerie_bras"]) if random.random() > 0.5 else None
        lingerie_item = f"{primary_color} {thong}"
        if bra:
            lingerie_item += f" and {primary_color} {bra}"
        lingerie_detail = random.choice(self.fashion_elements["lingerie_details"])
        stocking = f"{primary_color} {random.choice(self.fashion_elements['stockings'])}"
        pose = random.choice(self.fashion_elements["boudoir_office_poses"])
        hairstyle = random.choice(self.fashion_elements["hairstyles"])
        background = random.choice(self.fashion_elements["backgrounds"])
        lighting = random.choice(self.fashion_elements["lighting"])
        camera_angle = random.choice(self.fashion_elements["camera_angles"])
        expression = random.choice(self.fashion_elements["expressions"])

        prompt = f"""
Ultra-realistic sensual photograph of the beautiful celebrity  (({celebrity_name}:1.2)), captured from an **extreme low angle from floor level**, **gently tilted upward** to emphasize the skirt's texture, the visible {primary_color} {thong}, and then ascending towards the face with impeccable makeup. The camera is positioned slightly below the garment level, near the floor or knees, to highlight details and add drama and depth, emphasizing textures and shapes. The framing is tight and close, completely filling the frame.

The composition frames from the lower part of her body up to the face or makeup, using the pose and gaze to guide the viewer's attention.

She wears a meticulously designed {primary_color} {outfit}, featuring {lingerie_item} with {lingerie_detail}. Also wearing {stocking}. {pose}, with {hairstyle}. Expression: {expression}.

**Capture Technique:**
- **Angle and Framing:** {camera_angle} - **Crucially, the camera is placed slightly below garment level (near the floor or knees) and tilted gently upward.**
- **Distance:** Close-up, capturing fine details in the fabric, lace, or finishes without losing the context of the ensemble. A macro or zoom prime lens (between 85mm and 135mm) is used to capture minute textures.
- **Lighting:** {lighting}, utilizing soft, directional light or diffusers. Side or backlighting is ideal to accentuate reliefs in lace or embroidery, avoiding harsh shadows or distracting reflections.
- **Pose:** Poses with slight inclinations or twists, legs softly crossed, hands gently touching the skirt or lingerie to create dynamism and visual tactile sensitivity.

The image has a perfect depth of field, with sharp focus on the details and a softly blurred background. Expertly color graded for a polished, cinematic quality. Ultra HD, 16K.

Negative prompt: bad framing, subject too small, empty space, cropped poorly, cut off limbs, out of frame, border, frame, outline, harsh shadows, uneven lighting, dark areas, (bad anatomy, deformed, ugly), blurry, noisy, oversaturated, underexposed, overexposed, text, watermark, signature, grain, (anime, cartoon, 3d, painting, illustration), monochrome, black and white, sepia, vintage, low quality, plastic, doll, airbrushed, photoshopped, fake, surreal
"""
        return prompt.strip()

    def generate_five_prompts(self, celebrity_name):
        return [self.generate_prompt(celebrity_name) for _ in range(5)]


def create_gradio_interface():
    generator = FashionPromptGenerator()

    with gr.Blocks(title="FINAL BATUTO", theme=gr.themes.Soft(), css="""
        .copy-btn {
            margin-top: 8px;
        }
        .prompt-box {
            border: 1px solid #e2e8f0;
            border-radius: 8px;
            padding: 16px;
            margin-bottom: 16px;
            background-color: #f8fafc;
        }
        .prompt-title {
            font-weight: bold;
            margin-bottom: 8px;
            color: #4a5568;
        }
        .dark .prompt-box {
            background-color: #2d3748;
            border-color: #4a5568;
        }
        .dark .prompt-title {
            color: #e2e8f0;
        }
    """) as demo:
        gr.Markdown("# ✨ FINAL BATUTO: Fashion Prompt Generator ✨")
        gr.Markdown("Enter a celebrity's name to generate 5 unique and detailed fashion photography prompts.")

        with gr.Row():
            with gr.Column(scale=3):
                celebrity_input = gr.Textbox(
                    label="Celebrity Name",
                    placeholder="Ex: Scarlett Johansson, Zendaya, Ana de Armas...",
                    info="The name of the celebrity who will appear in the prompts."
                )
            with gr.Column(scale=1):
                generate_btn = gr.Button("Generate 5 Prompts", variant="primary", size="lg")
                clear_btn = gr.Button("Clear All", variant="secondary")

        gr.Markdown("## Generated Prompts")

        output_prompts = []
        copy_buttons = []
        prompt_columns = [] # Define prompt_columns here

        for i in range(5):
            with gr.Column(visible=False) as prompt_col:
                gr.Markdown(f"### Prompt {i+1}", elem_classes="prompt-title")
                with gr.Row():
                    prompt_output = gr.Textbox(
                        label="",
                        lines=15,
                        max_lines=20,
                        interactive=False,
                        elem_id=f"prompt_output_{i}",
                        show_copy_button=True,
                        elem_classes="prompt-box"
                    )
                output_prompts.append(prompt_output)
                prompt_columns.append(prompt_col) # Append to the list

        def generate_and_display_prompts(celebrity_name):
            if not celebrity_name.strip():
                return [gr.update(visible=False)] * 5 + [""] * 5
            prompts = generator.generate_five_prompts(celebrity_name)
            return [gr.update(visible=True)] * 5 + prompts

        def clear_all():
            return [gr.update(visible=False)] * 5 + [""] * 5

        generate_btn.click(
            fn=generate_and_display_prompts,
            inputs=celebrity_input,
            outputs=prompt_columns + output_prompts
        )

        clear_btn.click(
            fn=clear_all,
            inputs=None,
            outputs=prompt_columns + output_prompts
        )

        gr.Markdown("---")
        gr.Markdown("### Examples to try:")
        gr.Examples(
            examples=["Zendaya", "Ana de Armas", "Florence Pugh", "Sydney Sweeney", "Margot Robbie"],
            inputs=celebrity_input,
            outputs=prompt_columns + output_prompts,
            fn=generate_and_display_prompts,
            cache_examples=False,
            label="Click an example to load and generate."
        )

    return demo


if __name__ == "__main__":
    demo = create_gradio_interface()
    demo.launch(share=True)