Update engines/prompt_intelligence_engine.py

engines/prompt_intelligence_engine.py

@@ -2,78 +2,247 @@ import re
 
 
 class PromptIntelligenceEngine:
+    def __init__(self):
+        self.story_keywords = [
+            "histoire",
+            "conte",
+            "récit",
+            "recit",
+            "cinématique",
+            "cinematique",
+            "immersif",
+            "immersion",
+            "scène",
+            "scene",
+            "suite logique",
+            "ville futuriste",
+            "ville abandonnée",
+            "ville abandonnee",
+            "se réveille",
+            "se reveille",
+            "seul être vivant",
+            "seul etre vivant",
+            "monde vide",
+            "survie",
+            "survit",
+            "découvre progressivement",
+            "decouvre progressivement",
+            "un homme se réveille",
+            "une femme se réveille",
+            "se réveille dans",
+            "se reveille dans",
+            "découvre qu'il est seul",
+            "decouvre qu'il est seul",
+            "se retrouve seul",
+            "réveil dans une ville",
+            "reveil dans une ville",
+            "ville vide",
+            "ville désertée",
+            "ville desertee"
+        ]
+
+        self.anime_keywords = [
+            "anime",
+            "manga",
+            "dragon ball",
+            "dragon ball z",
+            "dragon ball super",
+            "naruto",
+            "one piece",
+            "bleach",
+            "snk",
+            "attaque des titans",
+            "combat",
+            "transformation",
+            "saïyen",
+            "saiyen",
+            "goku",
+            "vegeta",
+            "luffy",
+            "zoro",
+            "guerrier",
+            "ennemi invisible",
+            "nouvelle forme de puissance"
+        ]
+
+        self.tiktok_keywords = [
+            "tiktok",
+            "hook",
+            "cta",
+            "short",
+            "shorts",
+            "réel",
+            "reel",
+            "réels",
+            "reels",
+            "script vidéo",
+            "script video",
+            "vidéo courte",
+            "video courte",
+            "contenu viral",
+            "fait surprenant",
+            "secret",
+            "presque personne ne connaît",
+            "presque personne ne connait",
+            "tu savais",
+            "incroyable",
+            "le cerveau influence",
+            "psychologie",
+            "smartphone",
+            "smartphones",
+            "presque personne ne le sait",
+            "personne ne connaît",
+            "personne ne connait",
+            "personne ne sait",
+            "le cerveau",
+            "influence nos décisions",
+            "influence nos decisions"
+        ]
+
+        self.explanation_keywords = [
+            "explique",
+            "définis",
+            "definis",
+            "c'est quoi",
+            "c est quoi",
+            "qu'est-ce que",
+            "qu est ce que",
+            "comment ça marche",
+            "comment ca marche",
+            "en quelques lignes",
+            "simplement"
+        ]
+
+        self.business_keywords = [
+            "business",
+            "vente",
+            "marketing",
+            "offre",
+            "client",
+            "clients",
+            "stratégie",
+            "strategie",
+            "commercial",
+            "produit",
+            "marque",
+            "branding",
+            "rentable",
+            "attirer plus de clients",
+            "attirer des clients",
+            "magasin",
+            "réparation téléphone",
+            "reparation telephone"
+        ]
+
+        self.emotional_keywords = [
+            "triste",
+            "émotion",
+            "emotion",
+            "amour",
+            "rupture",
+            "touchant",
+            "inspirant",
+            "motivant",
+            "poétique",
+            "poetique",
+            "reprend espoir",
+            "espoir",
+            "période difficile",
+            "periode difficile",
+            "texte touchant",
+            "retrouve progressivement espoir",
+            "petits moments du quotidien",
+            "retrouve l'espoir",
+            "retrouve espoir",
+            "une femme traverse une période difficile",
+            "une femme traverse une periode difficile"
+        ]
+
+        self.latest_info_keywords = [
+            "dernière sortie",
+            "derniere sortie",
+            "dernières sorties",
+            "dernieres sorties",
+            "à jour",
+            "a jour",
+            "actuel",
+            "actualité",
+            "actualite",
+            "en ce moment",
+            "latest",
+            "news",
+            "nouveauté",
+            "nouveaute",
+            "date précise",
+            "date precise",
+            "sortie ps5",
+            "jeux ps5"
+        ]
 
     def normalize(self, text: str) -> str:
         return re.sub(r"\s+", " ", (text or "").strip())
 
-    # =========================
-    # INTENT DETECTION
-    # =========================
+    def _contains_any(self, text: str, keywords) -> bool:
+        return any(keyword in text for keyword in keywords)
 
     def detect_intent(self, prompt: str) -> str:
-        p = (prompt or "").lower()
+        p = self.normalize(prompt).lower()
 
-        if any(k in p for k in [
-            "dernière sortie", "derniere sortie", "actualité", "news",
-            "latest", "à jour", "actuel"
-        ]):
+        if not p:
+            return "general"
+
+        if self._contains_any(p, self.latest_info_keywords):
             return "latest_info"
 
-        if any(k in p for k in [
-            "tiktok", "hook", "cta", "short", "réel", "reel",
-            "vidéo courte", "contenu viral"
-        ]):
+        if self._contains_any(p, self.tiktok_keywords):
             return "tiktok"
 
-        if any(k in p for k in [
-            "anime", "manga", "dragon ball", "naruto", "one piece",
-            "combat", "transformation"
-        ]):
+        if self._contains_any(p, self.anime_keywords):
             return "anime"
 
-        if any(k in p for k in [
-            "histoire", "conte", "récit", "cinématique",
-            "immersif", "scene"
-        ]):
+        if self._contains_any(p, self.story_keywords):
             return "story"
 
-        if any(k in p for k in [
-            "explique", "définis", "c'est quoi",
-            "comment ça marche"
-        ]):
+        if self._contains_any(p, self.explanation_keywords):
             return "explanation"
 
-        if any(k in p for k in [
-            "business", "marketing", "client",
-            "vente", "produit"
-        ]):
+        if self._contains_any(p, self.business_keywords):
             return "business"
 
-        if any(k in p for k in [
-            "émotion", "triste", "amour", "motivant",
-            "espoir", "touchant"
-        ]):
+        if self._contains_any(p, self.emotional_keywords):
             return "emotional"
 
         return "general"
 
-    # =========================
-    # OPTIONAL BOOST
-    # =========================
-
     def enhance_prompt(self, prompt: str, intent: str) -> str:
-        prompt = self.normalize(prompt)
+        clean = self.normalize(prompt)
+
+        if not clean:
+            return clean
 
         if intent == "anime":
-            return f"{prompt}, epic anime battle, dynamic energy, cinematic intensity"
+            return (
+                f"{clean}. "
+                "Même personnage principal du début à la fin, progression nette du combat, "
+                "mise en scène dynamique, vraie montée en puissance, scènes visuellement différentes."
+            )
 
         if intent == "tiktok":
-            return f"{prompt}, viral content, strong hook, high engagement"
+            return (
+                f"{clean}. "
+                "Structure courte, claire, percutante, avec hook fort, montée d'intérêt et fin marquante."
+            )
 
         if intent == "emotional":
-            return f"{prompt}, deep emotion, human moment, cinematic realism"
+            return (
+                f"{clean}. "
+                "Même personne du début à la fin, émotion sincère, progression douce, visuels humains et réalistes."
+            )
 
         if intent == "story":
-            return f"{prompt}, immersive storytelling, cinematic progression"
+            return (
+                f"{clean}. "
+                "Même sujet principal du début à la fin, continuité narrative claire, scènes distinctes et immersives."
+            )
 
-        return prompt
+        return clean