Create assistants.py

assistants.py

@@ -0,0 +1,183 @@
+# ======================================================
+# 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."
+            )
+        ),
+        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