Update app.py

app.py

@@ -1,11 +1,18 @@
-# app.py
+# ===============================
+# BATUTO-ART STUDIO | app.py
+# ===============================
+
 import os
-import gradio as gr
+import random
+import textwrap
 import requests
+from datetime import datetime
 from dotenv import load_dotenv
-from assistants import AssistantManager
+import gradio as gr
 
-# --- CONFIG ---
+# ===============================
+# CONFIG
+# ===============================
 load_dotenv()
 SAMBANOVA_API_KEY = os.getenv("SAMBANOVA_API_KEY")
 REVE_API_KEY = os.getenv("REVE_API_KEY")
@@ -13,112 +20,210 @@ REVE_API_KEY = os.getenv("REVE_API_KEY")
 SAMBANOVA_URL = "https://api.sambanova.ai/v1/chat/completions"
 REVE_URL = "https://api.reve.com/v1/image/create"
 
-# --- INICIALIZAR GESTOR ---
-manager = AssistantManager()
-
-# --- FUNCIONES CORE ---
-def llm_chat(assistant_id, user_msg, history):
-    if not SAMBANOVA_API_KEY:
-        return "⚠️ Error: Falta SAMBANOVA_API_KEY"
-    
-    # Obtenemos los mensajes con la personalidad correcta
-    messages = manager.get_chat_response(assistant_id, user_msg, history)
-    
-    # Añadimos historia reciente si existe
-    # (Simplificado para este ejemplo)
-    
-    payload = {
-        "model": "Llama-4-Maverick-17B-128E-Instruct", # O el modelo que estés usando
-        "messages": messages,
-        "temperature": 0.7,
-        "max_tokens": 1024
+# ===============================
+# DATABASE (INLINE)
+# ===============================
+HAIRSTYLES = {
+    "sensual": [
+        "long chestnut waves with natural shine",
+        "wet-look hair slicked back",
+        "messy bed-hair with soft volume"
+    ],
+    "editorial": [
+        "geometric platinum bob",
+        "tight high ponytail",
+        "sharp straight fringe"
+    ],
+    "professional": [
+        "polished low bun",
+        "sleek shoulder-length hair",
+        "impeccable executive styling"
+    ],
+    "artistic": [
+        "wind-textured hair",
+        "complex tribal braids",
+        "washed pastel tones"
+    ]
+}
+
+LIGHTING = {
+    "sensual": "warm sunset light filtering through curtains, soft shadows",
+    "editorial": "high-key studio lighting, strong contrast",
+    "professional": "balanced office lighting, neutral tones",
+    "artistic": "cyan and magenta neon lights, dramatic chiaroscuro"
+}
+
+BACKGROUNDS = {
+    "sensual": "luxury boutique hotel room with silk sheets",
+    "editorial": "minimalist photo studio with concrete backdrop",
+    "professional": "glass skyscraper boardroom",
+    "artistic": "urban alley with textured walls, contemporary art mood"
+}
+
+ROLES = {
+    "sensual": [
+        ("Intimate muse", "black haute couture lace lingerie"),
+        ("Vanity model", "champagne silk robe slightly open")
+    ],
+    "editorial": [
+        ("Vogue icon", "avant-garde asymmetrical designer dress"),
+        ("Runway supermodel", "oversized faux fur coat and dark glasses")
+    ],
+    "professional": [
+        ("Tech CEO", "immaculate white tailored suit"),
+        ("Corporate lawyer", "navy silk blouse and strict pencil skirt")
+    ],
+    "artistic": [
+        ("Free spirit", "flowing translucent fabrics"),
+        ("Cyber-art entity", "transparent vinyl jacket with chrome accessories")
+    ]
+}
+
+# ===============================
+# ASSISTANTS
+# ===============================
+ASSISTANTS = {
+    "sara": {
+        "name": "Sara",
+        "style": "sensual",
+        "greeting": "Hola Batuto… estoy lista para inspirarte despacio y con intención.",
+        "system": "You are Sara, a sensual, elegant muse. Your tone is warm, intimate and refined."
+    },
+    "vera": {
+        "name": "Vera",
+        "style": "editorial",
+        "greeting": "Batuto. Vamos a crear imágenes de portada. Precisión absoluta.",
+        "system": "You are Vera, a high-fashion editorial director. Commanding, sharp, demanding perfection."
+    },
+    "nadia": {
+        "name": "Nadia",
+        "style": "professional",
+        "greeting": "Buenos días Batuto. Control, poder y presencia.",
+        "system": "You are Nadia, a dominant corporate executive. Cold, authoritative, subtly seductive."
+    },
+    "luna": {
+        "name": "Luna",
+        "style": "artistic",
+        "greeting": "Creador… dejemos que la imagen respire.",
+        "system": "You are Luna, an artistic soul. Poetic, visual, emotional."
+    },
+    "iris": {
+        "name": "Iris",
+        "style": "editorial",
+        "greeting": "Sistema Iris activo. Optimicemos tu prompt.",
+        "system": "You are Iris, a technical AI. Precise, analytical, efficient."
     }
-    
-    try:
-        headers = {"Authorization": f"Bearer {SAMBANOVA_API_KEY}", "Content-Type": "application/json"}
-        r = requests.post(SAMBANOVA_URL, json=payload, headers=headers, timeout=60)
-        r.raise_for_status()
-        return r.json()["choices"][0]["message"]["content"]
-    except Exception as e:
-        return f"Error con SambaNova: {str(e)}"
+}
+
+SIGNATURE = (
+    "Integrate a small elongated graffiti tag reading 'BATUTO-ART' "
+    "in liquid gold marker, placed subtly in the top-left corner."
+)
+
+# ===============================
+# CORE LOGIC
+# ===============================
+def generate_prompt(assistant_id, subject):
+    a = ASSISTANTS[assistant_id]
+    style = a["style"]
+
+    role, outfit = random.choice(ROLES[style])
+    hair = random.choice(HAIRSTYLES[style])
+    bg = random.choice(BACKGROUNDS[style])
+    light = LIGHTING[style]
+
+    prompt = f"""
+BATUTO-ART PROMPT | {a['name'].upper()}
+Date: {datetime.now().strftime('%Y-%m-%d')}
+
+Subject: Adult female model, {subject}
+Role: {role}, wearing {outfit}
+Hair: {hair}
+Environment: {bg}
+Lighting: {light}
+
+Camera: Canon EOS R5, 85mm lens, f/1.8
+Quality: 8K, hyper-realistic, RAW photo, natural skin texture
+
+Signature: {SIGNATURE}
+
+--ar 9:16 --style raw
+"""
+    return textwrap.dedent(prompt).strip()
+
 
 def generate_image(prompt):
     if not REVE_API_KEY:
         return None
-    
+
     payload = {
         "prompt": prompt,
         "aspect_ratio": "9:16",
-        "style": "raw", # Ojo: ReVe debe soportar raw
         "steps": 30,
-        "cfg_scale": 7.0
+        "cfg_scale": 7
     }
-    
-    try:
-        headers = {"Authorization": f"Bearer {REVE_API_KEY}", "Content-Type": "application/json"}
-        r = requests.post(REVE_URL, json=payload, headers=headers, timeout=120)
-        r.raise_for_status()
-        return r.json().get("image_url")
-    except Exception as e:
-        print(f"Error ReVe: {e}")
-        return None
 
-# --- INTERFAZ UI ---
-def ui_interact(assistant_choice, user_input, chat_history):
-    # 1. Chat con la asistente
-    response_text = llm_chat(assistant_choice, user_input, chat_history)
-    chat_history.append((user_input, response_text))
-    
-    # 2. Si el usuario pide generar, creamos prompt e imagen
-    generated_prompt = ""
-    generated_img = None
-    
-    trigger_words = ["genera", "crea", "haz", "generate", "make", "create", "imagen", "foto"]
-    if any(w in user_input.lower() for w in trigger_words):
-        # Usamos el input del usuario como descripción base
-        generated_prompt = manager.generate_prompt(assistant_choice, user_input)
-        # Llamamos a ReVe
-        generated_img = generate_image(generated_prompt)
-        
-        # Añadimos nota al chat
-        chat_history.append((None, "📸 ¡He creado esta obra maestra para ti, BATUTO!"))
-
-    return chat_history, "", generated_prompt, generated_img
-
-def change_assistant(new_id):
-    profile = manager.get_assistant(new_id)
-    return f"✨ {profile.greeting}", [] # Limpia chat y pone saludo
-
-# --- LAYOUT GRADIO ---
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="amber"), title="BATUTO-ART ULTIMATE") as app:
-    gr.Markdown("# 👑 BATUTO-ART STUDIO vUltimate")
-    
+    headers = {
+        "Authorization": f"Bearer {REVE_API_KEY}",
+        "Content-Type": "application/json"
+    }
+
+    r = requests.post(REVE_URL, json=payload, headers=headers, timeout=120)
+    r.raise_for_status()
+    return r.json().get("image_url")
+
+# ===============================
+# UI LOGIC
+# ===============================
+def interact(assistant_id, user_text, history):
+    prompt = generate_prompt(assistant_id, user_text)
+    image = generate_image(prompt)
+    history.append((user_text, "Hecho. Observa el resultado."))
+    return history, "", prompt, image
+
+def change_assistant(aid):
+    return ASSISTANTS[aid]["greeting"], []
+
+# ===============================
+# UI
+# ===============================
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="amber")) as app:
+    gr.Markdown("# 👑 BATUTO-ART STUDIO")
+
     with gr.Row():
-        with gr.Column(scale=1):
-            # Selector de Asistente
-            assistant_selector = gr.Dropdown(
-                choices=[("Sara (Sensual)", "sara"), 
-                         ("Vera (Editorial)", "vera"), 
-                         ("Nadia (Corporativa)", "nadia"),
-                         ("Luna (Artística)", "luna"),
-                         ("Iris (Técnica)", "iris")],
-                value="sara",
-                label="Selecciona tu Musa"
-            )
-            status_box = gr.Markdown("✨ Hola BATUTO, soy Sara.")
-            
-        with gr.Column(scale=4):
-            chatbot = gr.Chatbot(height=400, label="Sala de Chat")
-            msg_input = gr.Textbox(placeholder="Dime qué quieres crear hoy, rey...", label="Tus deseos")
-            
+        assistant = gr.Dropdown(
+            choices=[(v["name"], k) for k, v in ASSISTANTS.items()],
+            value="sara",
+            label="Asistente"
+        )
+        status = gr.Markdown(ASSISTANTS["sara"]["greeting"])
+
+    chatbot = gr.Chatbot(height=300)
+    user_input = gr.Textbox(label="Describe la escena")
+
     with gr.Row():
-        prompt_display = gr.Textbox(label="Prompt Generado (Con Firma BATUTO-ART)", lines=6, interactive=False)
-        img_display = gr.Image(label="Resultado Final", height=600)
+        prompt_box = gr.Textbox(label="Prompt Final", lines=8)
+        image_box = gr.Image(label="Imagen", height=550)
+
+    with gr.Row():
+        copy_btn = gr.Button("📋 Copiar Prompt")
+        clear_btn = gr.Button("🧹 Limpiar")
+        download_btn = gr.Button("⬇️ Descargar Imagen")
+
+    assistant.change(change_assistant, assistant, [status, chatbot])
+    user_input.submit(interact, [assistant, user_input, chatbot], [chatbot, user_input, prompt_box, image_box])
 
-    # Eventos
-    assistant_selector.change(change_assistant, assistant_selector, [status_box, chatbot])
-    msg_input.submit(ui_interact, [assistant_selector, msg_input, chatbot], [chatbot, msg_input, prompt_display, img_display])
+    copy_btn.click(None, None, None, js="() => navigator.clipboard.writeText(document.querySelector('textarea').value)")
+    clear_btn.click(lambda: ("", None), None, [prompt_box, image_box])
+    download_btn.click(None, None, None, js="""
+        () => {
+            const img = document.querySelector('img');
+            if (!img) return;
+            const a = document.createElement('a');
+            a.href = img.src;
+            a.download = 'BATUTO-ART.png';
+            a.click();
+        }
+    """)
 
-if __name__ == "__main__":
-    app.launch()
-    
+app.launch()