Update app.py

app.py

@@ -31,102 +31,128 @@ def analyze_python_code(code: str) -> str:
 tools = [analyze_python_code]
 
 # Inicializa el agente con herramientas y modelo
-# CORRECCIÓN APLICADA: Se establece add_base_tools=False para evitar el error de dependencia 'ddgs'.
+# Se mantiene add_base_tools=False para evitar el error de dependencia 'ddgs'.
 agent = CodeAgent(
     tools=tools,
     model=model,
     add_base_tools=False 
 )
 
-# DATOS PARA GENERACIÓN DE PERSONAJES INVENTADOS (Mínimo 5 elementos para muestreo único)
+# DATOS PARA GENERACIÓN DE PERSONAJES INVENTADOS (TEMAS DE ROL ESPECÍFICOS)
 FICTIONAL_ROLES = [
-    "Cyber Priestess", "Shadow Weaver", "Neo-Viking Huntress", "Aetherpunk Alchemist", 
-    "Galactic Rogue", "Urban Shaman", "Silent Assassin", "Chrononaut Captain", "Arcane Detective", "Desert Nomad"
+    "Secretary", "Nurse", "Schoolgirl", "Maid", "Flight Attendant", "Nun"
 ]
 HAIR_STYLES = [
-    "long, braided white hair", "short, messy fluorescent green hair", "sleek, silver bun with neon accents", 
-    "crimson dreadlocks", "wavy black hair with blue tips", "voluminous golden curls", "asymmetrical pink bob",
-    "shaved head with glowing runes", "icy blue waterfall hair", "dark purple pigtails"
+    "long wavy auburn hair", "short, tight black bun", "blonde pigtails", 
+    "long cascading brunette hair", "high, professional chignon", "simple black veil over long hair"
 ]
+# OUTFITS REVISADOS para ser ajustados/sugerentes y acordes al rol
 OUTFIT_THEMES = [
-    "a tattered military jacket and tactical gear", "a flowing silk kimono with metallic accents", 
-    "an asymmetrical leather coat and utility harness", "a delicate, bioluminescent gown",
-    "a streamlined obsidian suit with glowing pipes", "a distressed denim outfit with heavy chains",
-    "a high-collared velvet cape and ornate gloves", "a minimalist carbon-fiber bodysuit",
-    "weathered bronze armor with tribal markings", "a torn, ethereal school uniform"
+    "a tight pencil skirt and partially unbuttoned blouse", 
+    "a fitted, short-hemline nurse's uniform dress", 
+    "a classic plaid schoolgirl uniform with a short pleated skirt", 
+    "a short black maid uniform with a tight corset and white lace apron", 
+    "a fitted aeromoza jacket and pencil skirt", 
+    "a severe black habit with exposed décolletage"
 ]
-EYE_COLORS = ["gold", "iridescent blue", "vibrant purple", "deep emerald green", "flaming orange", "silver", "pale white", "ruby red", "electric yellow", "magenta"]
-
+EYE_COLORS = ["gold", "iridescent blue", "vibrant purple", "deep emerald green", "flaming orange", "silver"]
 
+# SETTINGS REVISADOS: Ambientes cotidianos, no Sci-Fi/Neon
 SETTINGS = [
-    "neon-lit office at midnight with hum of computers and scent of rain",
-    "gothic library filled with flickering candlelight and faded parchment scent",
-    "throne room bathed in warm golden light with velvet drapes",
-    "sterile operating room with bright surgical lights and soft beeping sounds",
-    "magical forest with glowing flora and gentle breeze rustling leaves"
+    "vintage wood-paneled private office with an antique desk",
+    "hospital examining room during a quiet moment",
+    "empty classroom after hours, moonlight streaming through the window",
+    "opulent vintage hotel suite with plush velvet furniture",
+    "first-class airplane cabin during a long haul flight",
+    "silent, dimly lit stone chapel corridor"
 ]
 
+# ACTIVITIES REVISADAS: Poses que revelan la ropa interior desde ángulos bajos
 ACTIVITIES = [
-    "reaching for a top-shelf file",
-    "arranging occult tomes on a desk",
-    "casting a subtle spell with glowing magical effects",
-    "serving coffee with graceful movements",
-    "walking confidently down a gothic hallway"
+    "bending over a desk to pick up a dropped pen",
+    "climbing a short set of stairs, lifting the hemline slightly",
+    "adjusting a stocking while sitting casually on a chair",
+    "leaning over a counter to wipe a surface with a playful smirk",
+    "reaching for a top-shelf luggage compartment",
+    "kneeling to pray on a marble floor"
 ]
 
 LINGERIE_COLORS = ["black", "red", "white", "navy blue", "deep burgundy"]
 STOCKING_DETAILS = ["a delicate back seam", "intricate lace pattern", "classic fishnet design", "smooth satin finish", "subtle floral embroidery"]
 HEEL_DETAILS = ["sharp stiletto heels", "elegant kitten heels", "strappy high heels", "sleek patent leather pumps", "classic pointed toe heels"]
 POSES = ["confident stance, one hand on hip", "sitting casually on a desk", "leaning slightly forward", "standing with arms crossed", "walking with a playful smile"]
-EXPRESSIONS = ["a knowing, seductive smile", "a confident, alluring gaze", "a playful smirk", "an intense, focused look", "a soft, inviting smile"]
+# EXPRESSIONS REVISADAS: Enfocadas en el deseo y contacto visual
+EXPRESSIONS = [
+    "a knowing, seductive smile directed at the viewer", 
+    "an intense, lustful gaze locking with the camera", 
+    "a confident, alluring smirk and eye contact", 
+    "a playful wink and inviting stare", 
+    "a soft, inviting smile with direct eye contact"
+]
 
 def generate_prompt(role, hair_style, outfit_theme, eye_color, setting, activity, lingerie_color, stocking_detail, heel_detail, pose, expression, prompt_num):
     """
-    Genera un prompt en formato JSON aplicando los parámetros técnicos de hiperrealismo
-    y el estilo de fotografía editorial de alta moda con un personaje inventado.
+    Genera un prompt en texto plano en inglés con enfoque en Low-Angle, lencería de encaje
+    y contacto visual seductor.
     """
-    # Construcción de la descripción base del personaje para el campo 'prompt'
-    character_description = (
-        f"**{role}** (Invented Character) in a hyper-realistic, high-fashion editorial photograph. "
-        f"Full body shot (Plano entero), featuring **{hair_style}** and **{eye_color} eyes**. "
-        f"Her main attire is **{outfit_theme}**. "
-        f"Underneath, she is wearing **{lingerie_color} lace lingerie** (thong and bra) and thigh-high stockings with {stocking_detail}. "
-        f"Paired with {heel_detail}. Posed in a {pose} and {expression} in a {setting}. "
-        f"The character is {activity}."
+    
+    # 1. Main prompt (Description)
+    main_prompt_start = "A beautiful perfect cosplay of a"
+    main_prompt_details = (
+        f"{main_prompt_start} **{role}** (Invented Character) in a hyper-realistic, high-fashion editorial photograph. "
+        f"Full body shot (Plano entero), **occupying the entire frame** with a perfect body proportion. "
+        f"Featuring **{hair_style}** and **{eye_color} eyes**. "
+        f"Her suggestive main attire is **{outfit_theme}**. "
+        f"Underneath, she is wearing a transparent **{lingerie_color} lace thong** and matching lace bra, "
+        f"accentuated by thigh-high stockings with {stocking_detail}. "
+        f"Paired with {heel_detail}. Posed in a {pose} and **{expression}** in a {setting}. "
+        f"The character is captured while **{activity}**.\n"
     )
 
-    prompt_data = {
-        "title": f"Editorial Shoot - {role} - Prompt {prompt_num}",
-        "metadata": {
-            "aspect_ratio": "9:16",
-            "shot_type": "Full-Body Shot (Plano entero)",
-            "camera": "Hasselblad H6D-400c con 80mm f/2.8 lens",
-            "lighting": "Rembrandt lighting, ARRI SkyPanel S360-C illumination",
-            "aperture": "f/4 (Reduced Depth of Field)",
-            "signature": "BATUTO'ART (Graffiti Tag, Top-Left, 5%)"
-        },
-        "prompt": character_description,
-        "style_modifiers": [
-            "High-fashion editorial photography style",
-            "Hiper-realistic CGI by Blizzard Entertainment",
-            "Fotografía de Alexander Nanitchkov",
-            "Escaneado 3D de personaje AAA",
-            "Unreal Engine 5 Ray Tracing",
-            "Detalles epidérmicos con Mapas 8K"
-        ],
-        "physics_and_effects": {
-            "physiology": "Enhanced Human Proportions (Athlete body, subtle supernatural modifications)",
-            "texture": "Realistic skin imperfections (pores, scars, blemishes), Sub-surface scattering reflections on skin",
-            "atmosphere": "Volumetric atmospheric smoke, Real rain droplets (scale 1:1) (if applicable to setting)",
-            "clothing_detail": "Realistic fabric sheen, Wear and tear imperfections on clothes/props"
-        },
-        "negative_prompt_keywords": [
-            "cartoon", "blurry", "pixelated", "low-resolution", "watermark", "noise", "overexposed", "underexposed",
-            "unnatural shadows", "color banding", "oversaturation", "artificial textures", "disallowed artifacts"
-        ]
-    }
-
-    return json.dumps(prompt_data, indent=4)
+    # 2. Technical and style modifiers
+    style_modifiers = [
+        "High-fashion editorial photography style",
+        "Hiper-realistic CGI by Blizzard Entertainment",
+        "Fotografía de Alexander Nanitchkov", 
+        "Escaneado 3D de personaje AAA",
+        "Unreal Engine 5 Ray Tracing",
+        "Detalles epidérmicos con Mapas 8K"
+    ]
+    
+    # 3. Physics and Effects
+    # La fisiología se mantiene perfecta, pero se elimina la atmósfera sci-fi/neon
+    physics_effects = [
+        "Physiology: Enhanced Human Proportions (Athlete body, subtle supernatural modifications)",
+        "Texture: Realistic skin imperfections (pores, scars, blemishes), Sub-surface scattering reflections on skin",
+        "Clothing: Realistic fabric sheen, Wear and tear imperfections on clothes/props"
+    ]
+
+    # 4. Negative prompt (coalesced)
+    negative_prompt_keywords = [
+        "cartoon", "blurry", "pixelated", "low-resolution", "watermark", "noise", "overexposed", "underexposed",
+        "unnatural shadows", "color banding", "oversaturation", "artificial textures", "disallowed artifacts"
+    ]
+
+    # 5. Combining all parts into a cohesive text block
+    final_text = (
+        f"PROMPT {prompt_num} (Title: Editorial Shoot - {role}):\n\n"
+        f"{main_prompt_details}\n"
+        f"--- METADATA & ANGLES ---\n"
+        f"Aspect Ratio: 9:16. Shot Type: Full-Body Shot (Plano entero) - **Low-Angle (Worm's Eye View)**.\n"
+        f"Angle Focus: Shot from a very low position looking upward, creating visual lines toward the underwear area.\n"
+        f"Camera: Hasselblad H6D-400c with 80mm f/2.8 lens. Aperture: f/4 (Reduced Depth of Field).\n"
+        f"Lighting: Rembrandt lighting, ARRI SkyPanel S360-C illumination. Selective depth of field focusing on the area of interest.\n\n"
+        f"--- STYLES & REFERENCES ---\n"
+        f"{', '.join(style_modifiers)}\n\n"
+        f"--- PHYSICS & DETAILING ---\n"
+        f"{' | '.join(physics_effects)}\n\n"
+        f"--- NEGATIVE PROMPT ---\n"
+        f"Exclude: {', '.join(negative_prompt_keywords)}\n\n"
+        f"--- SIGNATURE ---\n"
+        f"Signature Tag: BATUTO'ART (Graffiti Tag, Top-Left, 5%)"
+    )
+
+    return final_text
 
 def chat_with_agent(user_message, history):
     user_message_clean = user_message.strip().lower()
@@ -138,41 +164,43 @@ def chat_with_agent(user_message, history):
         # Selección única de componentes para asegurar 5 PROMPTS NO REPETIDOS
         try:
             # Componentes del personaje
-            unique_roles = random.sample(FICTIONAL_ROLES, num_prompts)
-            unique_hair_styles = random.sample(HAIR_STYLES, num_prompts)
-            unique_outfit_themes = random.sample(OUTFIT_THEMES, num_prompts)
-            unique_eye_colors = random.sample(EYE_COLORS, num_prompts)
-
-            # Componentes de la escena (también muestreados de forma única para máxima variación)
-            unique_settings = random.sample(SETTINGS, num_prompts)
-            unique_activities = random.sample(ACTIVITIES, num_prompts)
-            unique_lingerie_colors = random.sample(LINGERIE_COLORS, num_prompts)
-            unique_stocking_details = random.sample(STOCKING_DETAILS, num_prompts)
-            unique_heel_details = random.sample(HEEL_DETAILS, num_prompts)
-            unique_poses = random.sample(POSES, num_prompts)
-            unique_expressions = random.sample(EXPRESSIONS, num_prompts)
+            # Usamos min(6, len(LISTA)) para garantizar que no falle si se reduce la lista de datos.
+            unique_roles = random.sample(FICTIONAL_ROLES, min(num_prompts, len(FICTIONAL_ROLES)))
+            unique_hair_styles = random.sample(HAIR_STYLES, min(num_prompts, len(HAIR_STYLES)))
+            unique_outfit_themes = random.sample(OUTFIT_THEMES, min(num_prompts, len(OUTFIT_THEMES)))
+            unique_eye_colors = random.sample(EYE_COLORS, min(num_prompts, len(EYE_COLORS)))
+
+            # Componentes de la escena 
+            unique_settings = random.sample(SETTINGS, min(num_prompts, len(SETTINGS)))
+            unique_activities = random.sample(ACTIVITIES, min(num_prompts, len(ACTIVITIES)))
+            unique_lingerie_colors = random.sample(LINGERIE_COLORS, min(num_prompts, len(LINGERIE_COLORS)))
+            unique_stocking_details = random.sample(STOCKING_DETAILS, min(num_prompts, len(STOCKING_DETAILS)))
+            unique_heel_details = random.sample(HEEL_DETAILS, min(num_prompts, len(HEEL_DETAILS)))
+            unique_poses = random.sample(POSES, min(num_prompts, len(POSES)))
+            unique_expressions = random.sample(EXPRESSIONS, min(num_prompts, len(EXPRESSIONS)))
 
         except ValueError as e:
-             return history + [[user_message, f"Error: Una de las listas de componentes tiene menos de 5 elementos y no se puede generar un set único de 5 prompts. Por favor, asegúrate de que las listas tengan al menos 5 elementos. Error detallado: {e}"]]
+             return history + [[user_message, f"Error: Una de las listas de componentes tiene menos de {num_prompts} elementos y no se puede generar un set único de 5 prompts. Por favor, asegúrate de que las listas tengan al menos {num_prompts} elementos. Error detallado: {e}"]]
 
 
         for i in range(num_prompts):
             # Se llama a la función con el set de componentes único para el índice 'i'
-            prompt_json_str = generate_prompt(
+            prompt_text = generate_prompt(
                 unique_roles[i],
                 unique_hair_styles[i],
                 unique_outfit_themes[i],
                 unique_eye_colors[i],
                 unique_settings[i],
                 unique_activities[i],
-                unique_lingerie_colors[i],
-                unique_stocking_details[i],
-                unique_heel_details[i],
-                unique_poses[i],
-                unique_expressions[i],
+                unique_lingerie_colors[i % len(LINGERIE_COLORS)], # Usar módulo para listas más cortas
+                unique_stocking_details[i % len(STOCKING_DETAILS)], 
+                unique_heel_details[i % len(HEEL_DETAILS)],
+                unique_poses[i % len(POSES)],
+                unique_expressions[i % len(EXPRESSIONS)],
                 i + 1
             )
-            prompts.append(f"### PROMPT {i + 1} GENERADO (JSON):\n```json\n{prompt_json_str}\n```")
+            # Se envuelve la salida de texto plano en un bloque de código monoscape
+            prompts.append(f"\n```\n{prompt_text}\n```")
         
         return history + [[user_message, "\n\n".join(prompts)]]
 
@@ -182,7 +210,7 @@ def chat_with_agent(user_message, history):
         return history + [[user_message, analysis_result]]
 
     else:
-        return history + [[user_message, "Por favor, escribe 'Ok' para generar cinco prompts en **formato JSON**, o 'codigo:' seguido del código Python para analizarlo y corregirlo."]]
+        return history + [[user_message, "Por favor, escribe 'Ok' para generar cinco prompts en **formato de texto plano monoscape**, o 'codigo:' seguido del código Python para analizarlo y corregirlo."]]
 
 with gr.Blocks() as demo:
     gr.Markdown("## Generador de Prompts Hiperrealistas y Análisis de Código Python")
@@ -197,5 +225,4 @@ with gr.Blocks() as demo:
 
 if __name__ == "__main__":
     demo.launch(share=False)
-