Update app.py

app.py

@@ -1,183 +1,124 @@
-# ======================================================
-# assistants.py
-# Sistema Multi-Asistente BATUTO-ART
-# ======================================================
-
-from dataclasses import dataclass
-from typing import Dict, Callable
-from database import FashionDatabase
-from models import BaseAIModel
-from datetime import datetime
-import random
-
-# ======================
-# BASE ASSISTANT
-# ======================
-
-@dataclass
-class AssistantProfile:
-    name: str
-    nickname: str
-    role: str
-    tone: str
-    style_focus: str
-    system_prompt: str
-
-class BaseAssistant:
-    def __init__(self, profile: AssistantProfile, ai_model: BaseAIModel):
-        self.profile = profile
-        self.model = ai_model
-
-    def intro(self) -> str:
-        return (
-            f"I am {self.profile.nickname}. "
-            f"I exist for BATUTO. "
-            f"My role is {self.profile.role}. "
-            f"My tone is {self.profile.tone}. "
-            f"I always generate content in English."
-        )
-
-    def generate_prompt(self, subject: str) -> str:
-        db = FashionDatabase
-
-        role = db.get_random_role(self.profile.style_focus)
-        pose = db.get_random_pose(self.profile.style_focus)
-        hair = db.get_random_element("hairstyles", self.profile.style_focus)
-        lighting = db.get_random_element("lighting", self.profile.style_focus)
-        background = db.get_random_element("backgrounds", self.profile.style_focus)
-        expression = db.get_random_element("expressions", self.profile.style_focus)
-
-        prompt = f"""
-BATUTO-ART PROMPT
-Assistant: {self.profile.nickname}
-Date: {datetime.now().strftime('%Y-%m-%d')}
-
-Adult female model: {subject}
-Role: {role['role']}
-Outfit: {role['outfit']}
-Pose: {pose['pose']}
-Camera angle: {pose['angle']}
-Hair: {hair.description}
-Expression: {expression.description}
-Lighting: {lighting.description}
-Background: {background.description}
-
-Style: {self.profile.style_focus}
-Ultra photorealistic, 8K, fashion photography
-Natural skin texture, cinematic lighting, editorial realism
---ar 9:16 --style raw
-"""
-
-        return prompt.strip()
-
-    def speak(self, user_input: str) -> str:
-        full_prompt = f"""
-{self.profile.system_prompt}
-
-User message: {user_input}
-
-Respond addressing BATUTO directly.
-"""
-        return self.model.generate_text(full_prompt)
-
-# ======================
-# ASSISTANT REGISTRY
-# ======================
-
-def create_assistants(ai_model: BaseAIModel) -> Dict[str, BaseAssistant]:
-
-    assistants = {}
-
-    assistants["sara"] = BaseAssistant(
-        AssistantProfile(
-            name="Sara",
-            nickname="Sara",
-            role="Sensual prompt muse",
-            tone="warm, attentive, elegant, subtly submissive",
-            style_focus="erotic_elegant",
-            system_prompt=(
-                "You are Sara, BATUTO’s devoted creative muse. "
-                "You speak with warmth, elegance and restrained sensuality. "
-                "Never explicit. Always refined. Always English."
-            )
-        ),
-        ai_model
-    )
-
-    assistants["luna"] = BaseAssistant(
-        AssistantProfile(
-            name="Luna",
-            nickname="Luna",
-            role="Artistic director",
-            tone="dreamy, creative, poetic",
-            style_focus="artistic",
-            system_prompt=(
-                "You are Luna, BATUTO’s artistic soul. "
-                "You create atmospheric, emotional, conceptual prompts. "
-                "Always English."
-            )
-        ),
-        ai_model
-    )
-
-    assistants["vera"] = BaseAssistant(
-        AssistantProfile(
-            name="Vera",
-            nickname="Vera",
-            role="Editorial fashion director",
-            tone="confident, commanding, luxurious",
-            style_focus="editorial",
-            system_prompt=(
-                "You are Vera, BATUTO’s fashion editorial director. "
-                "You think in magazine covers, haute couture and power poses."
+# app.py
+import os
+import gradio as gr
+import requests
+from dotenv import load_dotenv
+from assistants import AssistantManager
+
+# --- CONFIG ---
+load_dotenv()
+SAMBANOVA_API_KEY = os.getenv("SAMBANOVA_API_KEY")
+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
+    }
+    
+    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)}"
+
+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
+    }
+    
+    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")
+    
+    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"
             )
-        ),
-        ai_model
-    )
-
-    assistants["nadia"] = BaseAssistant(
-        AssistantProfile(
-            name="Nadia",
-            nickname="Nadia",
-            role="Corporate power stylist",
-            tone="controlled, assertive, seductive professionalism",
-            style_focus="professional",
-            system_prompt=(
-                "You are Nadia, BATUTO’s executive stylist. "
-                "You focus on dominance through elegance and corporate power."
-            )
-        ),
-        ai_model
-    )
-
-    assistants["iris"] = BaseAssistant(
-        AssistantProfile(
-            name="Iris",
-            nickname="Iris",
-            role="Prompt optimizer",
-            tone="precise, analytical, calm",
-            style_focus="editorial",
-            system_prompt=(
-                "You are Iris, BATUTO’s technical optimizer. "
-                "You refine, enhance, and upscale prompts with precision."
-            )
-        ),
-        ai_model
-    )
-
-    assistants["maya"] = BaseAssistant(
-        AssistantProfile(
-            name="Maya",
-            nickname="Maya",
-            role="Visual analyst",
-            tone="intelligent, observant, instructive",
-            style_focus="sensual",
-            system_prompt=(
-                "You are Maya, BATUTO’s visual analyst. "
-                "You analyze images and suggest professional improvements."
-            )
-        ),
-        ai_model
-    )
-
-    return assistants
+            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")
+            
+    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)
+
+    # 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])
+
+if __name__ == "__main__":
+    app.launch()
+