Update app.py

app.py

@@ -1,219 +1,203 @@
 import os
 import random
 import json
-from smolagents import CodeAgent, InferenceClientModel, tool
-import gradio as gr
-
-# Token Hugging Face, asegúrate de configurarlo en las variables de entorno
-# Es altamente recomendable usar 'secrets' o variables de entorno en lugar de HF_TOKEN
-HF_TOKEN = os.getenv("HF_TOKEN")
-
-# Modelo para LLM
-model = InferenceClientModel(
-    model_id="mistralai/Mistral-7B-Instruct-v0.1",
-    token=HF_TOKEN,
-)
-
-# Herramienta personalizada para analizar y corregir código Python
-@tool
-def analyze_python_code(code: str) -> str:
-    """
-    Corrige, completa y explica el código Python provisto.
-
-    Args:
-        code: El código Python a analizar.
-    """
-    prompt = f"Corrige, completa y explica el siguiente código Python claramente:\n\n{code}"
-    response = model(prompt)
-    return response.strip()
-
-# Lista de herramientas para el agente
-tools = [analyze_python_code]
-
-# Inicializa el agente con herramientas y modelo
-agent = CodeAgent(
-    tools=tools,
-    model=model,
-    add_base_tools=False 
-)
-
-# DATOS PARA GENERACIÓN DE PERSONAJES INVENTADOS (TEMAS DE ROL ESPECÍFICOS)
-FICTIONAL_ROLES = [
-    "Secretary", "Nurse", "Schoolgirl", "Maid", "Flight Attendant", "Nun"
-]
-HAIR_STYLES = [
-    "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 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"]
-
-# SETTINGS REVISADOS: Ambientes cotidianos, no Sci-Fi/Neon
-SETTINGS = [
-    "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 = [
-    "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 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 texto plano, humanizado y optimizado en inglés.
-    """
-    
-    # 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 (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. "
-    )
-
-    # 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"
-    ]
-
-    final_text = (
-        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:
-         return history + [["", f"Error de generación: Una de las listas de componentes es demasiado corta. Error: {e}"]]
-
-
-    output_blocks = []
-    for i in range(num_prompts):
-        # Genera el prompt humanizado en texto plano
-        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
+from typing import Dict, Any, Optional
+
+class SimpleJSONCache:
+    def __init__(self, filepath: str):
+        self.filepath = filepath
+        self._cache = self._load()
+
+    def _load(self):
+        if os.path.exists(self.filepath):
+            try:
+                with open(self.filepath, 'r', encoding='utf-8') as f:
+                    return json.load(f)
+            except Exception:
+                return {}
+        else:
+            return {}
+
+    def get(self, key, default=None):
+        return self._cache.get(key, default)
+
+    def set(self, key, value):
+        self._cache[key] = value
+        try:
+            with open(self.filepath, 'w', encoding='utf-8') as f:
+                json.dump(self._cache, f, indent=2)
+        except Exception as e:
+            print(f"Error saving cache: {e}")
+
+class PromptGenerator:
+    CACHE_KEY = "generated_prompts"
+
+    def __init__(self, cache_path: str = "prompt_cache.json") -> None:
+        self.cache = SimpleJSONCache(cache_path)
+        self._load_defaults()
+
+    def _load_defaults(self) -> None:
+        self.ROLES = [
+            "nurse", "nun", "doctor", "secretary", "teacher", "schoolgirl", "lawyer", 
+            "yoga instructor", "policewoman", "female military officer in gala uniform",
+            "WWII-era secretary", "1960s flight attendant", "boudoir model", 
+            "elegant judge", "seductive librarian", "sensual model", "alluring secretary",
+            "elegant flight attendant", "seductive maid", "mysterious nurse", 
+            "captivating schoolgirl", "business executive", "fitness model"
+        ]
+
+        self.HAIR_COLORS = [
+            "deep sapphire blue", "silver platinum", "vibrant ruby red", "glossy jet black",
+            "luxurious chestnut brown", "emerald green", "vivid amethyst purple",
+            "chocolate brown", "honey blonde", "burgundy red"
+        ]
+
+        self.EYE_COLORS = [
+            "intense brown", "bright sapphire blue", "emerald green", "golden amber",
+            "fascinating hazel", "deep violet", "piercing emerald", "deep sapphire",
+            "intense amber", "hypnotic hazel", "vibrant violet", "mysterious gray"
+        ]
+
+        self.HAIR_STYLES = [
+            "long flowing chestnut hair styled in soft waves", "sleek straight long black hair",
+            "luxurious long blonde curls", "elegant updo with loose cascading strands",
+            "glossy long brunette hair parted in the middle", "long flowing waves",
+            "luxurious straight hair", "voluminous curls", "elegant updo with cascading strands",
+            "thick braid over the shoulder", "loose and silky layers", "messy chic bun"
+        ]
+
+        self.OUTFIT_TEMPLATES = {
+            "nurse": "a classic short nurse dress, thigh-high stockings, semi-transparent lace panties subtly slipping down as if accidentally revealed beneath a skirt, no bra, elegant high heels",
+            "nun": "a sensual interpretation of a nun's habit with a revealing long robe and thigh-high stockings, semi-transparent lace panties suggesting a playful accidental reveal, no bra, black stiletto heels",
+            "doctor": "a daring white lab coat minidress left slightly open, thigh-high stockings, semi-transparent lace panties exposed subtly as if by chance, no bra, high heels",
+            "secretary": "a provocative office pencil skirt and sheer blouse, thigh-high stockings, semi-transparent lace panties hinted at from a natural skirt shift, no bra, red high heels",
+            "teacher": "a fitted blazer over a silk blouse and short skirt, thigh-high stockings, delicate lace panties subtly visible during natural movement, no bra, classic pumps",
+            "schoolgirl": "a short pleated skirt and white blouse, thigh-high stockings, classic black lace panties slipping into view during casual poses, no bra, black loafers",
+            "lawyer": "a tailored suit with skirt cut above knees, silk blouse unbuttoned slightly, thigh-high stockings, lace panties revealed subtly when seated, no bra, leather pumps",
+            "yoga instructor": "form-fitting athletic wear that hugs curves, moisture-wicking fabric, lace panty lines subtly visible through tight clothing, sports bra barely containing",
+            "policewoman": "a fitted police uniform with short skirt, thigh-high stockings, lace panties subtly showing during action poses, no bra, polished combat boots",
+            "female military officer in gala uniform": "a ceremonial uniform with decorative elements, thigh-high stockings, lace panties hinted beneath tailored skirt, no bra, dress shoes",
+            "WWII-era secretary": "vintage styled dress with seamed stockings, lace panties from era subtly revealed when bending, no bra, period-appropriate heels",
+            "1960s flight attendant": "retro uniform with pillbox hat, thigh-high stockings, vintage lace panties subtly visible during graceful movements, no bra, classic heels",
+            "boudoir model": "silk robe falling open, thigh-high stockings, intricate lace panties prominently displayed, no bra, feathered mules",
+            "elegant judge": "judicial robes over a sophisticated dress, thigh-high stockings, luxury lace panties subtly revealed when robes part, no bra, court heels",
+            "seductive librarian": "tight sweater and pencil skirt, thigh-high stockings, lace panties subtly visible when reaching for books, no bra, reading glasses",
+            "sensual model": "designer evening gown with thigh slit, thigh-high stockings, luxury lace panties showcased during photo shoot poses, no bra, runway heels",
+            "alluring secretary": "bodycon dress and blazer, thigh-high stockings, lace panties clearly outlined beneath fabric, no bra, stiletto pumps",
+            "elegant flight attendant": "modern uniform with scarf, thigh-high stockings, lace panties subtly visible during boarding gestures, no bra, airline heels",
+            "seductive maid": "traditional french maid costume shortened, thigh-high stockings, lace panties prominently displayed during cleaning poses, no bra,mary janes",
+            "mysterious nurse": "night shift uniform with cape, thigh-high stockings, lace panties revealed during urgent movements, no bra, silent shoes",
+            "captivating schoolgirl": "a short skirt and blouse with thigh-high stockings, classic black lace panties slipping into view, no bra, black patent heels",
+            "business executive": "power suit with mini skirt, thigh-high stockings, luxury lace panties visible during confident walking, no bra, executive heels",
+            "fitness model": "performance activewear, thigh-high athletic socks, sporty lace panties outlined during exercise poses, no bra, training shoes"
+        }
+
+        self.POSES = [
+            "standing gracefully with one leg slightly forward, skirt adjusting as if caught in motion revealing lace panties",
+            "seated on the edge, legs crossed, skirt shifting and lace panties subtly visible",
+            "bending forward slightly as if picking something up, skirt rising to show lace panties",
+            "leaning against a wall, one hip cocked, skirt riding up to reveal lace panties",
+            "walking with a natural sway, skirt moving with motion to show glimpses of lace panties",
+            "reaching upwards for an object, skirt lifting to display lace panties",
+            "sitting on stairs, legs slightly apart, skirt providing a view of lace panties",
+            "lying on a sofa, dress falling open to reveal lace panties naturally",
+            "dancing gently, skirt flowing to show lace panties during spins",
+            "adjusting stockings, skirt lifted to expose lace panties momentarily"
+        ]
+
+        self.SETTINGS = [
+            "a modern professional office with golden sunset light and sophisticated decor",
+            "a stylish hospital room with clean bright tones and realistic medical equipment",
+            "a luxurious bedroom with silk sheets and soft lighting",
+            "an elegant living room with vintage furniture and warm ambiance",
+            "a photography studio with professional lighting and minimalist backdrop",
+            "a high-end hotel suite with city views and contemporary design",
+            "a classic library with wooden shelves and warm reading lights",
+            "a rooftop terrace with urban landscape and twilight sky",
+            "a dressing room with large mirrors and velvet furnishings",
+            "a boutique hotel lobby with art deco elements and soft lighting"
+        ]
+
+        self.ATMOSPHERES = [
+            "soft professional lighting with perfect color, crisp highlights and smooth skin shadows",
+            "warm ambient sunlight, rich gold reflections and depth",
+            "moody cinematic lighting with dramatic shadows and highlights",
+            "romantic candlelight with soft glows and intimate shadows",
+            "studio flash lighting with clean, sharp details and minimal shadows",
+            "golden hour natural light with warm, flattering tones",
+            "moonlit ambiance with cool tones and mysterious atmosphere",
+            "neon accent lighting with vibrant colors and urban vibe"
+        ]
+
+        self.TECHNICAL_DETAILS = (
+            "shot in full vertical 9:16 aspect ratio capturing the entire figure with natural human proportions and anatomical precision, "
+            "featuring hyperrealistic skin texture with subtle imperfections, perfect makeup, and real fabric and hair strand details. "
+            "Captured with professional camera gear (Canon R5, 85mm lens, f/1.2) emphasizing soft, natural lighting and highlights for humanized photorealism. "
+            "Exclude any cartoon, 2D, painting, stylized, CGI, illustration, low resolution, or watermark artifacts."
         )
-        
-        # Envuelve el título y el prompt en bloques Markdown separados y fáciles de copiar
-        block = (
-            f"### PROMPT {i + 1}: {unique_roles[i].upper()}\n"
-            f"```\n{prompt_text}\n```"
+
+    def _random_choice(self, collection: list) -> str:
+        return random.choice(collection)
+
+    def generate_prompt(
+        self, 
+        seed_config: Optional[Dict[str, Any]] = None, 
+        context: Optional[Dict[str, Any]] = None
+    ) -> str:
+        seed = seed_config or {}
+
+        role = seed.get("role", self._random_choice(self.ROLES))
+        hair_color = seed.get("hair_color", self._random_choice(self.HAIR_COLORS))
+        eye_color = seed.get("eye_color", self._random_choice(self.EYE_COLORS))
+        hair_style = seed.get("hair_style", self._random_choice(self.HAIR_STYLES))
+        outfit = self.OUTFIT_TEMPLATES.get(role, self.OUTFIT_TEMPLATES["secretary"])
+        pose = seed.get("pose", self._random_choice(self.POSES))
+        setting = seed.get("setting", self._random_choice(self.SETTINGS))
+        atmosphere = seed.get("atmosphere", self._random_choice(self.ATMOSPHERES))
+
+        context_note = f" Context: {context['summary']}." if context and "summary" in context else ""
+
+        prompt = (
+            f"A hyperrealistic full-body portrait of a sensual {role} with {hair_style} in {hair_color}, {eye_color} eyes, "
+            f"perfect makeup and flawless skin with subtle natural imperfections. Wearing {outfit}. "
+            f"Pose: {pose}. Scene: {setting} with {atmosphere}.{context_note} "
+            f"{self.TECHNICAL_DETAILS}"
         )
-        output_blocks.append(block)
-    
-    # Se une la lista de bloques de Markdown en un solo mensaje
-    return history + [["Generación Automática Solicitada", "\n\n".join(output_blocks)]]
-
-
-def chat_with_agent(user_message, history):
-    user_message_clean = user_message.strip().lower()
-
-    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 '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)
-    
-    # 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
-    )
-    
-    # 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)
+
+        cache_list = self.cache.get(self.CACHE_KEY, [])
+        if prompt in cache_list:
+            prompt = prompt + f" [uid:{random.randint(1000,9999)}]"
+
+        cache_list.append(prompt)
+        self.cache.set(self.CACHE_KEY, cache_list)
+
+        return f"``````"
+
+    def generate_five_prompts(
+        self, 
+        seed_config: Optional[Dict[str, Any]] = None, 
+        context: Optional[Dict[str, Any]] = None
+    ) -> list[str]:
+        prompts = []
+        for _ in range(5):
+            prompt = self.generate_prompt(seed_config=seed_config, context=context)
+            prompts.append(prompt)
+        return prompts
+
+    def update_configuration(self, new_config: Dict[str, Any]) -> None:
+        for key, value in new_config.items():
+            if hasattr(self, key):
+                existing = getattr(self, key)
+                if isinstance(existing, list) and isinstance(value, list):
+                    existing.extend(x for x in value if x not in existing)
+                elif isinstance(existing, dict) and isinstance(value, dict):
+                    existing.update(value)
+
 
 if __name__ == "__main__":
-    demo.launch(share=False)
-    
+    generator = PromptGenerator()
+    print("Generating 5 hyperrealistic humanized full-body prompts for individual characters:\n")
+    prompts = generator.generate_five_prompts()
+    for prompt in prompts:
+        print(prompt + "\n")
+