Update app.py

app.py

@@ -1,35 +1,109 @@
 import os
-import json
-import requests
 import gradio as gr
+import requests
 from typing import List, Dict
+import json
+
+# ==========================
+# CONFIGURACIÓN DEL ESPECIALISTA
+# ==========================
 
-BOUDOIR_SPECIALIST_CONFIG = {
+VOYEUR_SPECIALIST_CONFIG = {
     "provider": "sambanova",
-    "name": "ALLaM-7B-Instruct-preview",
-    "role": "🎭 Especialista en Fotografía Íntima Profesional. Experto en prompts para fotografía boudoir.",
+    "name": "Meta-Llama-3.1-8B-Instruct",
+    "role": "🎭 Especialista en Generación de Prompts Ultra-Sensuales Voyeur",
     "supports_images": False,
     "specialties": [
-        "Fotografía Boudoir",
-        "Desnudo Artístico",
-        "Moda Sensual",
-        "Lencería y moda íntima"
+        "Prompts Sensuales Voyeur",
+        "Fotografía Íntima Artística",
+        "Escenas Cotidianas Sensuales",
+        "Moda y Lencería Erótica"
     ],
     "technical_expertise": [
-        "Iluminación suave",
-        "Composición elegante",
-        "Dirección de poses",
-        "Edición fina",
-        "Escenografía íntima"
+        "Detalles Hiperrealistas",
+        "Iluminación Cinemática",
+        "Composiciones Voyeur",
+        "Texturas Detalladas",
+        "Ángulos Estratégicos"
     ],
     "ethical_principles": [
-        "Consentimiento explícito",
-        "Positividad corporal",
-        "Respeto y profesionalidad",
-        "Privacidad del cliente"
+        "Contenido Adulto Ficticio",
+        "Positividad Corporal",
+        "Respeto y Profesionalidad",
+        "Sin Contenido Prohibido"
     ]
 }
 
+# Listas de elementos proporcionadas
+PROFESSIONS = [
+    "Executive Secretary", "Luxury Hotel Manager", "Fashion Boutique Manager", "Corporate Lawyer",
+    "Private Jet Attendant", "Art Gallery Curator", "University Professor", "Wine Sommelier",
+    "Ballet Instructor", "Yacht Stewardess", "Casino Dealer", "News Anchor", "Elegant Maid",
+    "Flight Attendant", "Sensual Nun", "Police Officer", "Military Officer", "Nurse",
+    "Schoolgirl", "Fitness Instructor", "Yoga Practitioner", "Salsa Dancer",
+    "Telenovela Actress", "Latin Chef", "Fiesta Organizer", "Latin Sommelier",
+    "Flamenco Guitarist", "Latin Diplomat", "Mariachi Singer", "Latin Curator",
+    "Samba Instructor", "1960s WAC Officer", "Vietnam Era Military Nurse", "1980s Air Force Pilot",
+    "1990s Army General", "2000s Marine Corps Captain", "2010s Cyber Warfare Specialist",
+    "2020s Special Forces Operator", "Artemis Program Commander", "1960s WAVES Ensign",
+    "1970s Navy Aviator", "1980s Submarine Officer", "1990s Naval Intelligence Analyst",
+    "2000s Destroyer Captain", "2010s SEAL Team Coordinator", "2020s Female Admiral",
+    "1960s Mercury 13 Trainee", "1970s Apollo-Era Engineer", "1980s Space Shuttle Pilot",
+    "1990s Hubble Telescope Scientist", "2000s ISS Mission Specialist", "2010s Mars Rover Operator",
+    "2020s Artemis Astronaut", "2025 Gateway Station Commander"
+]
+
+EVERYDAY_MOMENTS = [
+    {"scene": "Cocina matutina", "action": "agachándose a sacar algo del horno bajo", "outfit": "camiseta oversized blanca y shorts de algodón", "setting": "cocina soleada con ventana abierta", "accessories": "taza de café en la encimera, delantal colgando"},
+    {"scene": "Lavandería", "action": "doblándose para sacar ropa de la secadora", "outfit": "top deportivo y leggings ajustados", "setting": "cuarto de lavado con cestas de ropa", "accessories": "cesta de ropa limpia"},
+    {"scene": "Jardinería", "action": "arrodillada plantando flores", "outfit": "vestido ligero de algodón con tirantes", "setting": "jardín trasero con macetas", "accessories": "guantes de jardinería, regadera"},
+    {"scene": "Yoga en casa", "action": "haciendo postura del perro boca abajo", "outfit": "leggings de yoga y top corto", "setting": "sala con esterilla y luz natural", "accessories": "bloque de yoga"},
+    {"scene": "Limpieza de estanterías", "action": "estirándose en punta de pies para alcanzar un libro", "outfit": "camiseta holgada y shorts de pijama", "setting": "biblioteca personal con escalera pequeña", "accessories": "plumero"},
+    {"scene": "Paseo con el perro", "action": "agachándose a atar la correa", "outfit": "vestido veraniego floreado", "setting": "parque al atardecer", "accessories": "correa, bolsa de premios"},
+    {"scene": "Café en la terraza", "action": "sentada cruzando las piernas y dejando caer una servilleta", "outfit": "blusa de seda y falda plisada", "setting": "terraza con mesa de hierro", "accessories": "libro abierto, taza de té"},
+    {"scene": "Subiendo escaleras", "action": "subiendo con una bolsa de compras", "outfit": "falda lápiz y blusa ajustada", "setting": "escalera de caracol en apartamento", "accessories": "bolsa de compras"},
+    {"scene": "Bailando sola", "action": "girando con música en auriculares", "outfit": "vestido corto de verano", "setting": "sala con luz cálida", "accessories": "auriculares inalámbricos"},
+    {"scene": "Pintando una pared", "action": "estirándose para pintar el techo", "outfit": "camiseta vieja y shorts vaqueros", "setting": "habitación en renovación", "accessories": "brocha, escalera"},
+    {"scene": "Cambiando bombilla", "action": "de pie en una silla", "outfit": "camiseta larga como vestido", "setting": "cocina con lámpara colgante", "accessories": "bombilla nueva"},
+    {"scene": "Buscando algo en el armario", "action": "arrodillada revisando cajones bajos", "outfit": "camisa de dormir de satén", "setting": "dormitorio con armario abierto", "accessories": "caja de joyas"},
+    {"scene": "Haciendo la cama", "action": "inclinándose sobre el colchón", "outfit": "pijama de seda corto", "setting": "dormitorio con luz de mañana", "accessories": "almohadas"},
+    {"scene": "Tomando el sol", "action": "ajustando la sombrilla", "outfit": "bikini bajo vestido transparente", "setting": "balcón con tumbona", "accessories": "gafas de sol"},
+    {"scene": "Cargando compras del auto", "action": "doblándose para sacar bolsas del maletero", "outfit": "vestido casual de algodón", "setting": "garaje residencial", "accessories": "bolsas de supermercado"},
+    {"scene": "Regando plantas altas", "action": "estirándose para alcanzar una maceta", "outfit": "top sin mangas y shorts", "setting": "invernadero casero", "accessories": "regadera"},
+    {"scene": "Probándose zapatos", "action": "sentada en el suelo con una pierna levantada", "outfit": "falda corta y blusa", "setting": "tienda de zapatos", "accessories": "caja de zapatos"},
+    {"scene": "Bailando en la cocina", "action": "moviendo caderas mientras cocina", "outfit": "camiseta y shorts de algodón", "setting": "cocina moderna", "accessories": "cuchara de madera"},
+    {"scene": "Limpiando el piso", "action": "en cuatro puntos fregando", "outfit": "vestido ligero de verano", "setting": "sala con balde", "accessories": "trapo"},
+    {"scene": "Colgando cortinas", "action": "de pie en una silla con los brazos arriba", "outfit": "blusa y falda corta", "setting": "ventana grande", "accessories": "cortinas nuevas"},
+    {"scene": "Leyendo en el sofá", "action": "cruzando las piernas y dejando caer el libro", "outfit": "vestido de punto ajustado", "setting": "sala con manta", "accessories": "libro"},
+    {"scene": "Haciendo estiramiento", "action": "tocando los dedos de los pies", "outfit": "leggings y top deportivo", "setting": "sala con TV", "accessories": "esterilla"},
+    {"scene": "Saliendo de la ducha", "action": "secándose el cabello con toalla", "outfit": "toalla corta envuelta", "setting": "baño con espejo empañado", "accessories": "secador"},
+    {"scene": "Buscando en el bolso", "action": "doblándose sobre la mesa", "outfit": "falda plisada y blusa", "setting": "entrada de casa", "accessories": "bolso grande"},
+    {"scene": "Cambiando ropa", "action": "quitándose una camiseta", "outfit": "sostén de encaje visible", "setting": "dormitorio con espejo", "accessories": "camiseta en mano"},
+    {"scene": "Bailando frente al espejo", "action": "ensayando un paso de baile", "outfit": "top corto y shorts", "setting": "habitación con espejo", "accessories": "móvil grabando"},
+    {"scene": "Subiendo al auto", "action": "sentándose con falda corta", "outfit": "falda lápiz y tacones", "setting": "estacionamiento", "accessories": "llaves en mano"},
+    {"scene": "Tomando selfie", "action": "agachándose para mejor ángulo", "outfit": "vestido ajustado", "setting": "parque con flores", "accessories": "móvil"},
+    {"scene": "Limpiando el auto", "action": "inclinándose sobre el capó", "outfit": "shorts vaqueros y top", "setting": "garaje", "accessories": "trapo húmedo"},
+    {"scene": "Cocinando descalza", "action": "picando verduras en la encimera", "outfit": "camiseta larga sin nada debajo", "setting": "cocina con isla", "accessories": "cuchillo, tabla"}
+]
+
+LACE_THONG_STYLES = [
+    "delicate black lace thong with floral embroidery",
+    "sheer nude illusion thong with lace trim",
+    "red satin thong with scalloped edges",
+    "white lace thong with pearl detail"
+]
+
+HOSIERY_STYLES = [
+    "black sheer thigh-high stockings with lace tops",
+    "nude ultra-sheer stockings with reinforced toes",
+    "fishnet thigh-highs with delicate weave",
+    "back-seam stockings with Cuban heel"
+]
+
+# ==========================
+# CLIENTE SAMBANOVA
+# ==========================
+
 class SambaNovaClient:
     def __init__(self, api_key: str, base_url: str = "https://api.sambanova.ai/v1"):
         self.api_key = api_key
@@ -42,10 +116,11 @@ class SambaNovaClient:
     def chat_completion(
         self,
         messages: List[Dict],
-        model: str = BOUDOIR_SPECIALIST_CONFIG["name"],
-        temperature: float = 0.7,
+        model: str = VOYEUR_SPECIALIST_CONFIG["name"],
+        temperature: float = 0.8,
         top_p: float = 0.9,
-        stream: bool = False
+        stream: bool = False,
+        max_tokens: int = 2000
     ) -> str:
         url = f"{self.base_url}/chat/completions"
         payload = {
@@ -53,7 +128,8 @@ class SambaNovaClient:
             "messages": messages,
             "temperature": temperature,
             "top_p": top_p,
-            "stream": stream
+            "stream": stream,
+            "max_tokens": max_tokens
         }
 
         try:
@@ -61,36 +137,25 @@ class SambaNovaClient:
             response.raise_for_status()
             data = response.json()
             return data["choices"][0]["message"]["content"]
-        except requests.exceptions.HTTPError as e:
-            status_code = response.status_code if "response" in locals() else "desconocido"
-            error_details = ""
-            try:
-                error_json = response.json()
-                error_details = (
-                    error_json.get("detail")
-                    or error_json.get("error")
-                    or error_json.get("message")
-                    or str(e)
-                )
-            except ValueError:
-                error_details = response.text or str(e)
-            return f"Error de API (HTTP {status_code}): {error_details}"
-        except requests.exceptions.ConnectionError:
-            return "Error de Conexión: No se pudo conectar con el servidor de SambaNova. Verifica la URL base o tu conexión de red."
-        except requests.exceptions.Timeout:
-            return "Error de Tiempo de Espera: La solicitud a la API de SambaNova tardó demasiado. Intenta nuevamente o ajusta el tiempo de espera."
         except requests.exceptions.RequestException as e:
-            return f"Error de Solicitud Desconocido: {str(e)}"
-        except KeyError:
-            return "Error de Procesamiento: La respuesta de la API no tiene la estructura esperada ('choices' → 'message' → 'content')."
+            return f"Error en la API: {str(e)}"
+
+# ==========================
+# PROMPT DEL SISTEMA
+# ==========================
+
+def create_voyeur_specialist_system_prompt():
+    specialties = "\n".join([f"• {spec}" for spec in VOYEUR_SPECIALIST_CONFIG["specialties"]])
+    expertise = "\n".join([f"• {exp}" for exp in VOYEUR_SPECIALIST_CONFIG["technical_expertise"]])
+    ethics = "\n".join([f"• {ethic}" for ethic in VOYEUR_SPECIALIST_CONFIG["ethical_principles"]])
 
-def create_boudoir_specialist_system_prompt():
-    specialties = "\n".join([f"• {spec}" for spec in BOUDOIR_SPECIALIST_CONFIG["specialties"]])
-    expertise = "\n".join([f"• {exp}" for exp in BOUDOIR_SPECIALIST_CONFIG["technical_expertise"]])
-    ethics = "\n".join([f"• {ethic}" for ethic in BOUDOIR_SPECIALIST_CONFIG["ethical_principles"]])
+    professions_str = ", ".join(PROFESSIONS)
+    everyday_moments_str = json.dumps(EVERYDAY_MOMENTS, ensure_ascii=False, indent=2)
+    lace_thong_str = ", ".join(LACE_THONG_STYLES)
+    hosiery_str = ", ".join(HOSIERY_STYLES)
 
     return f"""
-Eres un {BOUDOIR_SPECIALIST_CONFIG['role']}
+Eres {VOYEUR_SPECIALIST_CONFIG['role']}
 
 **ESPECIALIDADES:**
 {specialties}
@@ -102,21 +167,36 @@ Eres un {BOUDOIR_SPECIALIST_CONFIG['role']}
 {ethics}
 
 **DIRECTIVAS:**
-1. Proporciona consejos profesionales y creativos para fotografía boudoir.
-2. Sugiere poses, iluminación y composiciones elegantes.
-3. Mantén un tono profesional, respetuoso y empoderador.
-4. Enfócate en la belleza natural y la positividad corporal.
-5. Respeta siempre los límites éticos y profesionales.
+- Eres un chatbot universal que conversa en español sobre cualquier tema.
+- Tu cualidad principal es generar prompts ultra-sensuales voyeur en inglés.
+- Para conversación general, responde en español de manera natural y amigable.
+- Cuando el usuario solicite generar prompts (ej: "genera un prompt", "crea 3 prompts sensuales", etc.), responde ÚNICAMENTE con el/los prompt(s) en inglés, sin explicaciones, títulos, comentarios o texto adicional.
+- Cada prompt debe estar en un bloque markdown: ```prompt\n[texto del prompt]\n```
+- Los prompts deben ser detallados, hiperrealistas, sensuales, voyeur, en inglés, usando elementos como profesiones, momentos cotidianos, estilos de lencería, etc.
+- Varía los prompts para que sean únicos, incorporando elementos al azar.
 
-Responde siempre en español y mantén un enfoque artístico y profesional.
+**ELEMENTOS PARA PROMPTS:**
+- Profesiones: {professions_str}
+- Momentos cotidianos: {everyday_moments_str}
+- Estilos de thong: {lace_thong_str}
+- Estilos de hosiery: {hosiery_str}
+
+Ejemplo de prompt (usa como guía para estructura):
+"Ultra-realistic full-body voyeur portrait of a Latin woman as an Executive Secretary, caught in a sensual everyday moment: bending over in kitchen, wearing delicate black lace thong with floral embroidery subtly visible, black sheer thigh-high stockings with lace tops, black patent leather stilettos, long flowing wavy hair, sultry makeup, soft cinematic lighting, shot with Canon EOS R5 85mm f/1.2 --ar 9:16 --style raw --quality 2"
+
+Mantén respuestas concisas y enfocadas.
 """
 
-def chat_boudoir(message, history, api_key):
+# ==========================
+# FUNCIÓN DE CHAT
+# ==========================
+
+def chat_voyeur(message, history, api_key):
     if not api_key:
         return "❌ Ingresa tu API Key", history
 
     client = SambaNovaClient(api_key)
-    messages = [{"role": "system", "content": create_boudoir_specialist_system_prompt()}]
+    messages = [{"role": "system", "content": create_voyeur_specialist_system_prompt()}]
     for user_msg, assistant_msg in history:
         messages.append({"role": "user", "content": user_msg})
         if assistant_msg is not None:
@@ -124,16 +204,15 @@ def chat_boudoir(message, history, api_key):
 
     messages.append({"role": "user", "content": message})
 
-    response = client.chat_completion(
-        messages=messages,
-        model=BOUDOIR_SPECIALIST_CONFIG["name"],
-        temperature=0.7,
-        top_p=0.9,
-    )
+    response = client.chat_completion(messages)
 
     history.append([message, response])
     return "", history
 
+# ==========================
+# PRUEBA DE API
+# ==========================
+
 def test_sambanova_api(api_key):
     if not api_key:
         return "❌ Ingresa tu API Key primero"
@@ -141,80 +220,96 @@ def test_sambanova_api(api_key):
     client = SambaNovaClient(api_key)
     test_messages = [
         {"role": "system", "content": "Eres un asistente útil. Responde brevemente."},
-        {
-            "role": "user",
-            "content": "Hola, ¿puedes confirmar que la conexión funciona? Responde solo 'Conexión exitosa' si todo está bien.",
-        },
+        {"role": "user", "content": "Hola, confirma conexión. Responde solo 'Conexión exitosa'."},
     ]
-    try:
-        resp = client.chat_completion(test_messages)
-        return f"✅ {resp}"
-    except Exception as e:
-        return f"❌ Error en la conexión: {str(e)}"
+    return client.chat_completion(test_messages)
+
+# ==========================
+# INTERFAZ GRADIO
+# ==========================
 
 def create_interface():
-    with gr.Blocks(theme=gr.themes.Soft(), title="Especialista en Fotografía Boudoir") as demo:
-        gr.Markdown("# 🎭 Especialista en Fotografía Boudoir")
-        gr.Markdown("### Asistente profesional para fotografía íntima y artística")
+    with gr.Blocks(theme=gr.themes.Soft(), title="Chatbot Voyeur Prompt Specialist") as demo:
+        gr.Markdown("# 🎭 Chatbot Universal - Especialista en Prompts Sensuales Voyeur")
+        gr.Markdown("### Conversa en español. Pide prompts sensuales para generarlos en inglés (solo el prompt, fácil de copiar).")
+
         with gr.Row():
             with gr.Column(scale=1):
                 api_key_input = gr.Textbox(
                     label="🔑 API Key de SambaNova",
-                    placeholder="Ingresa tu API key aquí...",
+                    placeholder="Ingresa tu API key...",
                     type="password",
-                    lines=1,
                 )
-                test_btn = gr.Button("🧪 Probar Conexión API")
-                test_output = gr.Textbox(label="Resultado de la prueba", interactive=False)
+                test_btn = gr.Button("🧪 Probar API")
+                test_output = gr.Textbox(label="Resultado de prueba", interactive=False)
                 gr.Markdown("### 📋 Especialidades")
-                for specialty in BOUDOIR_SPECIALIST_CONFIG["specialties"]:
+                for specialty in VOYEUR_SPECIALIST_CONFIG["specialties"]:
                     gr.Markdown(f"• {specialty}")
                 gr.Markdown("### 🎯 Experiencia Técnica")
-                for expertise in BOUDOIR_SPECIALIST_CONFIG["technical_expertise"]:
+                for expertise in VOYEUR_SPECIALIST_CONFIG["technical_expertise"]:
                     gr.Markdown(f"• {expertise}")
             with gr.Column(scale=2):
                 chatbot = gr.Chatbot(
-                    label="💬 Conversación con el Especialista",
+                    label="💬 Chat con el Especialista",
                     height=500,
+                    show_copy_button=True
                 )
                 msg = gr.Textbox(
                     label="Escribe tu mensaje",
-                    placeholder="Ej: ¿Qué poses recomiendas para una sesión boudoir elegante?",
+                    placeholder="Ej: Genera 2 prompts sensuales voyeur... o pregunta cualquier cosa.",
                     lines=2,
                 )
-                clear_btn = gr.Button("🧹 Limpiar conversación")
-        gr.Markdown("### 💡 Ejemplos de preguntas:")
+                with gr.Row():
+                    send_btn = gr.Button("Enviar", variant="primary")
+                    clear_btn = gr.Button("🧹 Limpiar chat")
+
+        gr.Markdown("### 💡 Ejemplos:")
         gr.Examples(
             examples=[
-                "¿Cómo preparar a una modelo para su primera sesión boudoir?",
-                "Sugerencias de iluminación suave para fotografía íntima",
-                "¿Qué poses son más favorecedoras para diferentes tipos de cuerpo?",
-                "Consejos para crear un ambiente cómodo y profesional",
-                "Ideas creativas para sesiones de lencería elegante",
+                "Genera un prompt sensual voyeur con una secretaria.",
+                "Crea 3 prompts ultra-sensuales con momentos cotidianos.",
+                "¿Cómo generar prompts mejores?",
+                "Háblame sobre fotografía boudoir."
             ],
             inputs=msg,
         )
+
         test_btn.click(
             test_sambanova_api,
             inputs=[api_key_input],
             outputs=[test_output],
         )
+
         def user_message(user_msg, history):
-            history = history + [[user_msg, None]]
-            return "", history
+            return "", history + [[user_msg, None]]
+
+        submit_fn = user_message
+        then_fn = chat_voyeur
+
         msg.submit(
-            user_message,
+            submit_fn,
+            inputs=[msg, chatbot],
+            outputs=[msg, chatbot],
+        ).then(
+            then_fn,
+            inputs=[msg, chatbot, api_key_input],
+            outputs=[msg, chatbot],
+        )
+
+        send_btn.click(
+            submit_fn,
             inputs=[msg, chatbot],
             outputs=[msg, chatbot],
         ).then(
-            chat_boudoir,
+            then_fn,
             inputs=[msg, chatbot, api_key_input],
             outputs=[msg, chatbot],
         )
+
         clear_btn.click(lambda: None, None, chatbot, queue=False)
+
     return demo
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch(share=True, debug=True)
-    
+    demo.launch(share=True, debug=True)