Actualizar app.py

app.py

@@ -31,7 +31,6 @@ def analyze_python_code(code: str) -> str:
 tools = [analyze_python_code]
 
 # Inicializa el agente con herramientas y modelo
-# Se mantiene add_base_tools=False para evitar el error de dependencia 'ddgs'.
 agent = CodeAgent(
     tools=tools,
     model=model,
@@ -92,136 +91,123 @@ EXPRESSIONS = [
 
 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 texto plano en inglés con enfoque en Low-Angle, lencería de encaje
-    y contacto visual seductor.
+    Genera un prompt en texto plano, humanizado y optimizado en inglés.
     """
     
-    # 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"
+    # 1. Main prompt (Humanized and integrated)
+    main_prompt = (
+        f"A beautiful perfect cosplay of a {role} (Invented Character). Full body shot, Low-Angle (Worm's Eye View). The character is captured while {activity} in a {setting}. "
+        f"The shot is taken from a very low position looking upward, creating strong visual lines toward the underwear area. "
+        f"She has {hair_style} and {eye_color} eyes, with a perfect athlete body proportion. "
+        f"Her suggestive attire is {outfit_theme}. Underneath, she wears a transparent {lingerie_color} lace thong and matching lace bra, "
+        f"accentuated by thigh-high stockings with {stocking_detail} and {heel_detail}. "
+        f"The character is posed in a {pose} and maintains {expression}, always looking at the viewer with desire. "
     )
 
-    # 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"
-    ]
+    # 2. Technical and Style Modifiers (Optimized for flow)
+    technical_and_style = (
+        f"Technical Details: Aspect Ratio: 9:16. Shot Type: Full-Body Shot (Plano entero). Camera: Hasselblad H6D-400c with 80mm f/2.8 lens. "
+        f"Lighting: Rembrandt lighting, ARRI SkyPanel S360-C illumination. Aperture: f/4 (Reduced Depth of Field). "
+        f"Focus: Selective depth of field focusing on the area of interest. "
+        f"Style: High-fashion editorial photography, 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. "
+        f"Physics: Enhanced Human Proportions, Realistic skin imperfections (pores, scars, blemishes), Sub-surface scattering reflections on skin, Realistic fabric sheen. "
+    )
 
-    # 4. Negative prompt (coalesced)
+    # 3. Negative prompt (Clean list)
     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%)"
+        f"{main_prompt} "
+        f"{technical_and_style} "
+        f"Negative Prompt: {', '.join(negative_prompt_keywords)}"
     )
 
     return final_text
 
+# Nueva función para generar los prompts sin entrada de usuario "Ok"
+def generate_five_prompts(history):
+    prompts = []
+    num_prompts = 5
+    
+    # Selección única de componentes para asegurar 5 PROMPTS NO REPETIDOS
+    try:
+        # Componentes del personaje
+        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:
+         # Simplemente devuelve un error sin agregar al historial si la generación falla
+         return history + [["", f"Error de generación: Una de las listas de componentes es demasiado corta. Error: {e}"]]
+
+
+    for i in range(num_prompts):
+        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 % len(LINGERIE_COLORS)],
+            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
+        )
+        # Se genera la salida para el chat
+        prompts.append(f"### PROMPT {i + 1} ({unique_roles[i].upper()}):\n{prompt_text}\n")
+    
+    # Se añade la salida al historial como si el agente hubiera respondido
+    # La parte del usuario se deja vacía o se puede poner una etiqueta
+    return history + [["Generación Automática Solicitada", "\n\n".join(prompts)]]
+
+
 def chat_with_agent(user_message, history):
     user_message_clean = user_message.strip().lower()
 
-    if user_message_clean == "ok":
-        prompts = []
-        num_prompts = 5
-        
-        # Selección única de componentes para asegurar 5 PROMPTS NO REPETIDOS
-        try:
-            # Componentes del personaje
-            # 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 {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_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 % 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
-            )
-            # 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)]]
-
-    elif user_message_clean.startswith("codigo:") or user_message_clean.startswith("code:"):
+    if user_message_clean.startswith("codigo:") or user_message_clean.startswith("code:"):
         code_to_analyze = user_message.split(":", 1)[1].strip()
         analysis_result = analyze_python_code(code_to_analyze)
         return history + [[user_message, analysis_result]]
 
     else:
-        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."]]
+        return history + [[user_message, "Por favor, escribe 'codigo:' seguido del código Python para analizarlo y corregirlo, o haz clic en **'Generar Prompts'**."]]
 
 with gr.Blocks() as demo:
     gr.Markdown("## Generador de Prompts Hiperrealistas y Análisis de Código Python")
     chatbot = gr.Chatbot(label="Historial de Conversación", height=400)
-    msg = gr.Textbox(label="Escribe 'Ok' para prompts o 'codigo:' seguido de código Python")
-    clear = gr.Button("Limpiar")
-
+    
+    # Contenedor para el botón y la caja de texto
+    with gr.Row():
+        msg = gr.Textbox(label="Escribe 'codigo:' seguido del código Python para análisis", scale=4)
+        generate_btn = gr.Button("Generar Prompts", scale=1)
+        clear_btn = gr.Button("Limpiar", scale=1)
+        
+    # Lógica de envío de la caja de texto (solo para código)
     msg.submit(chat_with_agent, [msg, chatbot], chatbot, queue=False).then(
         lambda: gr.update(value="", interactive=True), None, msg, queue=False
     )
-    clear.click(lambda: None, None, chatbot, queue=False)
+    
+    # Lógica del nuevo botón (para generación automática)
+    generate_btn.click(generate_five_prompts, [chatbot], chatbot, queue=False)
+    
+    # Lógica del botón de limpiar
+    clear_btn.click(lambda: None, None, chatbot, queue=False)
 
 if __name__ == "__main__":
     demo.launch(share=False)