app.py

#!/usr/bin/env python3
import random
from typing import List
import gradio as gr

# --- Datos convertidos de JSON Java (estructuras originales completas) ---

BRA_STYLES = [
    "a light and transparent lace bra",
    "a sheer balcony bra with delicate lace details",
    "a push-up bra with intricate lace overlay",
    "an unlined lace bralette",
    "a seamless sheer bra with minimalist design",
    "a silk and lace demi-cup bra"
]

THONG_COLORS = [
    "black",
    "red",
    "white",
    "cream",
    "burgundy",
    "champagne",
    "dark blue",
    "deep purple",
    "silver",
    "gold",
    "pink",
    "black lace thong",
    "red silk thong"
]

THIGH_HIGH_STOCKINGS = [
    "thigh-high silk stockings with perfect details",
    "sheer black stockings with a delicate lace top",
    "seamless nude hosiery with perfect details",
    "fishnet tights that reach her thighs",
    "back-seam stockings with a lace garter"
]

SECRETARY_DRESS = [
    "a tight elastic short dress helping to appreciate transparent inner layers revealing delicate lingerie without riding up"
]

SECRETARY_SKIRT = [
    "a miniskirt paired with a semi-transparent blouse, elegantly revealing her outfit beneath"
]

HIGH_HEELS = [
    "classic high heels in black patent leather",
    "elegant red stiletto heels",
    "sleek nude high heels",
    "platform high heels",
    "sparkling silver high heels"
]

HAIR_STYLES = [
    "long elegant hairstyle with a refined arrangement",
    "sleek straight long hair",
    "soft voluminous curls",
    "a classic updo",
    "a messy bun",
    "a sophisticated chignon",
    # Original from second script, add sensual and elegant styles:
    "long flowing hair that frames her face sensually",
    "elegant waves that cascade over her shoulders seductively",
    "sleek hairstyle that accentuates her neck and features",
    "voluminous curls that enhance her sensual appearance",
    "sophisticated updo that highlights her facial features alluringly"
]

HAIR_COLORS = [
    "chestnut brown with vibrant highlights",
    "platinum blonde",
    "deep raven black",
    "auburn red",
    "honey blonde",
    "strawberry blonde",
    "ash blonde",
    "vibrant blue hair"
]

MAKEUP_STYLES = [
    "perfect professional makeup with subtle contouring and elegant lipstick",
    "natural-looking makeup enhancing her features with soft tones",
    "classic smoky eyes with nude lips",
    "bold red lipstick with minimalist eye makeup",
    "glittery eyeshadow with glossy lips",
    # Extended with second script makeup options:
    "flawless makeup with sensual emphasis on her lips and eyes",
    "perfectly contoured makeup that enhances her natural features seductively",
    "smokey eye makeup with glossy lips that express desire",
    "dewy complexion with accentuated features that radiate lust",
    "professional makeup with sensual emphasis that enhances her allure"
]

CAMERA_EQUIPMENT = [
    "a Canon EOS R5 Cine RAW camera paired with a Canon RF 85mm f/1.2L USM lens",
    "a Sony Alpha 1 with a G Master 50mm f/1.2 lens",
    "a Hasselblad X1D II 50C with an XCD 80mm f/1.9 lens",
    "a RED Komodo 6K with a Sigma Art 35mm f/1.4 lens"
]

LIGHTING_EQUIPMENT = [
    "an ARRI SkyPanel S360-C panel providing smooth, shadow-free professional light",
    "Godox AD600Pro with a large octabox for soft, even lighting",
    "Profoto B10 Plus with a strip softbox for edge lighting",
    "Nanlite Forza 500 with a parabolic softbox for dramatic lighting"
]

ACTIVITIES = [
    "bending over to pick up dropped documents",
    "reaching for a high shelf",
    "sitting cross-legged at her desk",
    "stretching after long hours of work",
    "leaning forward to hand over files",
    "adjusting her stockings discreetly",
    "getting up from her chair gracefully",
    "answering a phone call with a hand resting on her thigh",
    "typing intently on a vintage typewriter",
    "walking confidently towards the camera",
    "filing documents in a tall cabinet",
    "organizing her desk with a focused expression"
]

EXPRESSIONS = [
    "with a seductive glance towards the viewer",
    "smiling coyly while maintaining eye contact",
    "looking over her shoulder with desire",
    "biting her lip subtly with pleasure",
    "with a knowing smile that invites curiosity",
    "maintaining intense eye contact with the spectator",
    "a playful wink",
    "a mischievous grin",
    "a serious yet alluring gaze",
    # Extended with second script expressions of desire/lust
    "looking directly at the viewer with intense desire and lust in her eyes",
    "gazing seductively at the camera with passionate expression",
    "maintaining eye contact with the viewer with unmistakable desire",
    "looking passionately at the spectator with sensual intensity",
    "directing a lustful gaze toward the viewer with undeniable attraction"
]

PHOTOGRAPHY_COMPOSITION = {
    "Perspectives_and_Angles": [
        "extremely low angle perspective from floor level",
        "vertical 9:16 format",
        "subject dominates the entire frame with dramatic perspective",
        "ultra-shallow depth of field",
        "completely blurred background",
        "cinematic worm's-eye view emphasizing legs and figure",
        "mid-thigh up framing"
    ],
    "Visual_Style": [
        "professional, ultra-high-resolution photograph",
        "elegant and raw sensuality",
        "natural, unposed moment",
        "authentic yet erotic atmosphere",
        "sensual shadows and highlighting of delicate textures",
        "cinematic quality with rich tones and perfect contrast"
    ]
}

# Escenarios adicionales como secretaria, mini falda, uniforme escolar del segundo código:

OFFICE_SECRETARY_SCENARIOS = [
    "standing next to a desk, adjusting an executive mini skirt in a 'careless' way hinting at a thong, direct seductive gaze with pleasure, nadir angle from the ground",
    "sitting on an office table, legs accidentally ajar and mini skirt rising showing the thong, ecstatic seductive face, camera on the floor with contrapicado angle",
    "leaning to file documents, mini skirt sliding voyeuristically revealing the thong, intense gaze with an open mouth of pleasure, nadir angle",
    "leaning against a wall, executive mini skirt lifted by 'accident' hinting at the thong, defiant gaze with a look of ecstasy, contrapicado angle",
    "sitting in a swivel chair, crossing legs and letting the thong show carelessly under the mini skirt, seductive expression of pleasure, full frame with nadir perspective",
    "dancing discreetly in the office, mini skirt lifts unintentionally revealing the thong, seductive gaze with climax, camera on the ground with contrapicado angle",
    "delivering reports, leaning forward with the mini skirt rising voyeuristically showing the thong, playful gaze with ecstasy, nadir angle",
    "reclining on a desk, a bent leg lifting the mini skirt carelessly and revealing the thong, seductive face with pleasure, full frame contrapicado",
    "crouching to pick up papers, mini skirt accidentally risen showing the thong, fiery gaze with an open mouth of pleasure, nadir angle, camera on the floor",
    "organizing a bookshelf, mini skirt lifted voyeuristically showing the thong, direct gaze with ecstasy, elegant straight hairstyle, contrapicado angle",
    "sitting with a leg up on the desk, mini skirt unintentionally rising and the thong visible, defiant gaze with pleasure, nadir angle",
    "standing next to an office window, mini skirt lifted by 'wind' hinting at the thong, sensual seductive expression, full frame contrapicado",
    "leaning over a filing cabinet, mini skirt sliding carelessly showing the thong, mischievous gaze with orgasm, camera on the floor nadir angle",
    "sitting on a meeting table, short mini skirt and thong accidentally visible, ecstatic seductive face, nadir angle",
    "leaning on a photocopier, mini skirt lifted voyeuristically showing the thong, expression of pleasure looking at the viewer, full frame contrapicado",
    "filing on a high shelf, mini skirt rises unintentionally showing the thong, seductive facial expression of pleasure, nadir angle",
    "standing taking notes on a whiteboard, mini skirt rising carelessly and the thong visible, ecstatic face, camera on the ground contrapicado",
    "sitting on a desk, crossing legs and letting the thong show under the mini skirt voyeuristically, fiery gaze with an open mouth of pleasure, nadir angle",
    "leaning over a keyboard, mini skirt accidentally lifted showing the thong, seductive gaze with pleasure, full frame contrapicado",
    "with back to camera turning head, executive mini skirt lifted carelessly showing the thong, seductive expression of pleasure, nadir angle"
]

SCHOOLGIRL_UNIFORM_SCENARIOS = [
    "standing in a classroom, slightly lifting the schoolgirl skirt in 'carelessness' to show the thong, direct seductive gaze with pleasure, nadir angle",
    "sitting at a desk, legs accidentally open revealing the thong under the skirt, ecstatic seductive expression, camera on the floor with contrapicado angle",
    "crouching to pick up school supplies, skirt lifted voyeuristically and the thong visible, intense gaze with an open mouth of pleasure, full frame nadir angle",
    "leaning against a hallway wall, skirt unintentionally lifted hinting at the thong, defiant gaze with ecstasy, contrapicado angle",
    "with back to camera in a line, turning head with the skirt lifted carelessly showing the thong, seductive expression of pleasure, camera on the ground nadir angle",
    "sitting on a classroom table, crossed legs accidentally sliding showing the thong, playful gaze with ecstasy, full frame contrapicado",
    "dancing at a school party, skirt lifts up voyeuristically showing the thong, seductive gaze with climax, nadir angle",
    "climbing school stairs, skirt rises unintentionally showing the thong, head turned with a look of pleasure, camera on the floor contrapicado",
    "reclining on a desk, a bent leg lifting the skirt carelessly and revealing the thong, seductive face with pleasure, full frame nadir angle",
    "crouching, skirt accidentally lifted revealing the thong, fiery gaze with an open mouth of pleasure, nadir angle",
    "standing in front of a whiteboard, lifting the skirt with a hand for an 'adjustment' showing the thong, direct gaze with ecstasy, elegant wavy hairstyle, contrapicado angle",
    "kneeling in a classroom, skirt rolled up voyeuristically showing the thong, arched body with a seductive gesture of pleasure, camera on the floor nadir angle",
    "leaning on a desk, skirt unintentionally lifted showing the thong, defiant gaze with facial pleasure, full frame contrapicado",
    "lying on back on a table, legs bent and skirt fallen carelessly showing the thong, lost seductive gaze of pleasure, nadir angle",
    "dancing on a chair, short skirt accidentally lifts up revealing the thong, facial look of pleasure staring forward, low angle contrapicado",
    "sitting in the schoolyard, legs open from 'wind' with a wet skirt and the thong visible, ecstatic seductive expression, camera on the ground nadir angle",
    "spinning with a schoolgirl skirt, skirt flies up voyeuristically revealing the thong, direct gaze with an open mouth of pleasure, contrapicado angle",
    "standing with hands on head, skirt lifted simulating carelessness and the thong exposed, seductive expression of ecstasy, full frame nadir angle",
    "leaning over a desk, skirt rises unintentionally and the thong visible, defiant gaze with intense pleasure, nadir angle",
    "sitting with legs tucked in, skirt stretched carelessly showing the thong, mischievous gaze with a look of pleasure, perfect makeup contrapicado"
]

MINI_SKIRT_LINGERIE_SCENARIOS = [
    "standing, wind simulating carelessness slightly lifts the black pleated silk mini skirt revealing the red lace thong, direct seductive gaze at the viewer with eyes squinted in pleasure, nadir angle from the ground occupying the entire frame",
    "sitting on the edge of the bed, legs opened by 'accident' revealing the thong under the mini skirt, ecstatic expression with half-open lips reflecting pleasure, contrapicado angle camera on the ground",
    "crouching to pick something up from the ground, mini skirt lifted in carelessness showing the thong, intense seductive gaze with an open mouth of pleasure, nadir angle from the floor occupying the entire frame",
    "leaning against the wall, mini skirt lifted by 'carelessness' hinting at the thong, defiant gaze with a look of pleasure, contrapicado camera on the ground"
]

# --- Technical and Artistic Composition ---

ARTISTIC_COMPOSITION_DESC = (
    "(masterpiece, best quality:1.4), Ultra HD 16K (15360 × 8640), "
    "professional color photograph, captured with a Canon EOS R5 Cine RAW camera "
    "and a Canon RF 85mm f/1.2L USM lens, "
    "alternating between low-angle frontal, nadir (from below looking up), "
    "and contrapicado (slightly tilted low angle) perspectives, "
    "highlighting the figure from knees to head. The scene is illuminated by an ARRI SkyPanel S360-C panel, "
    "providing soft, professional light that emphasizes the textures of skin and delicate fabrics "
    "without harsh shadows. Expertly color graded in DaVinci Resolve with a cinematic touch, "
    "perfect depth of field, vertical 9:16 format. No distractions, perfect lighting without harsh shadows or annoying edges, "
    "occupying the entire frame with a sensual and natural atmosphere. (BATUTO-ART watermark:0.4)"
)

# --- Prompt Generation Function ---

def generate_prompt(celebrity_name: str, scenario: str) -> str:
    if not celebrity_name.strip():
        return "⚠️ Please enter a celebrity name to generate prompts."
    
    prompt = (
        f"The renowned celebrity {celebrity_name} is captured in a full-length professional color photograph "
        f"that conveys natural sensuality and elegance through subtle movement. {random.choice(EXPRESSIONS)}, "
        f"with {random.choice(MAKEUP_STYLES)}. Her {random.choice(HAIR_STYLES)} complements "
        f"her alluring appearance. The composition naturally suggests {random.choice(BRA_STYLES)} and "
        f"{random.choice(THONG_COLORS)} with {random.choice(THIGH_HIGH_STOCKINGS)} and {random.choice(HIGH_HEELS)}. "
        f"Scene: {scenario}. "
        f"The lighting is perfectly balanced without harsh shadows or distracting edges, creating a natural "
        f"and sensual atmosphere that occupies the entire frame. {celebrity_name}'s expression radiates "
        f"desire and lust while maintaining sophistication and elegance. "
        f"{ARTISTIC_COMPOSITION_DESC}"
    )

    negative_prompt = (
        "nsfw, nude, explicit content, deformed, bad anatomy, extra limbs, "
        "blurry, low quality, cartoon, anime, child, minor, grotesque, "
        "poor lighting, harsh shadows, annoying edges, vulgar, explicit display, "
        "overexposed, underexposed, poorly framed, distracting elements"
    )
    
    full_prompt = f"{prompt} | Negative prompt: {negative_prompt}"
    return full_prompt


def generate_all_prompts(celebrity_name: str) -> List[str]:
    if not celebrity_name.strip():
        return ["⚠️ Please enter a celebrity name to generate prompts."] * 5
    
    # Combinar y seleccionar escenarios al azar de todas las listas
    all_scenarios = OFFICE_SECRETARY_SCENARIOS + SCHOOLGIRL_UNIFORM_SCENARIOS + MINI_SKIRT_LINGERIE_SCENARIOS + ACTIVITIES
    selected_scenarios = random.sample(all_scenarios, 5)
    return [generate_prompt(celebrity_name, scenario) for scenario in selected_scenarios]


# --- Gradio Interface ---

title_html = """
<h1 style="
    font-family: 'Georgia', serif;
    font-size: 3.5em;
    font-weight: bold;
    text-align: center;
    background: linear-gradient(45deg, #003366, #336699, #6699CC);
    -webkit-background-clip: text;
    -webkit-text-fill-color: transparent;
    text-shadow: 1px 1px 2px rgba(0,0,0,0.7);
    margin-bottom: 10px;">
    SENSUAL CELEBRITY FASHION PROMPTS
</h1>
<p style="text-align: center; font-family:'Georgia', serif; font-size: 1.2em; color:#6699CC; margin-top: -10px;">
    Natural & Elegant Lingerie Suggestions
</p>
"""

with gr.Blocks(theme=gr.themes.Monochrome()) as app:
    gr.HTML(title_html)
    
    with gr.Row():
        name_input = gr.Textbox(
            label="Enter Celebrity Name:",
            placeholder="e.g., Ana de Armas, Margot Robbie, Alexandra Daddario"
        )
        generate_button = gr.Button("✨ Generate 5 Sensual Fashion Prompts", variant="primary")

    prompt_boxes = []
    for i in range(5):
        with gr.Column():
            gr.Markdown(f"### 📝 Prompt {i+1}")
            prompt_txt = gr.Textbox(
                interactive=False,
                lines=10,
                max_lines=15,
                show_copy_button=True
            )
            prompt_boxes.append(prompt_txt)

    generate_button.click(
        generate_all_prompts,
        inputs=name_input,
        outputs=prompt_boxes
    )

if __name__ == "__main__":
    app.launch()