Update context_engine.py

context_engine.py

@@ -1,4 +1,4 @@
-# context_engine.py - UPDATED (minor doc + thread-safe lazy loading)
+# context_engine.py - UPDATED to include emoji suggestions & guidance
 from typing import Tuple
 import threading
 
@@ -35,13 +35,18 @@ def _safe_emotion_analysis(text: str) -> Tuple[str, float]:
         print(f"Emotion analysis error: {e}")
         return ("neutral", 0.0)
 
-def get_smart_context(user_text):
+def get_smart_context(user_text: str):
     """
-    Analyzes the user's 'Vibe' and returns a short persona instruction.
+    Returns a short persona instruction block including:
+      - Conversation Mode (Ping-Pong / Standard / Deep Dive)
+      - Emotional context (tone + directive)
+      - Emoji suggestions (range and examples)
+      - Hint for minimum verbosity (guideline only)
+    This string is safe to inject into system prompt.
     """
     try:
         label, confidence = _safe_emotion_analysis(user_text or "")
-        top_emotion = label.lower() if label else "neutral"
+        top_emotion = (label or "neutral").lower()
         confidence = float(confidence or 0.0)
         word_count = len((user_text or "").split())
 
@@ -52,25 +57,56 @@ def get_smart_context(user_text):
         else:
             conversation_mode = "Deep Dive Mode (Detailed)"
 
+        # Emoji mapping and style hints
         if top_emotion == "joy":
-            emotional_context = "User: Positive/Energetic. Vibe: Celebrate — be upbeat but concise."
+            emotional_context = "User: Positive/Energetic. Vibe: Upbeat — be warm and slightly playful."
+            emoji_examples = "😊 🎉 🙂"
+            emoji_range = (1, 2)
         elif top_emotion == "sadness":
-            emotional_context = "User: Low Energy. Vibe: Supportive — patient and gentle."
+            emotional_context = "User: Low Energy. Vibe: Supportive — be gentle and empathetic."
+            emoji_examples = "🤍 🌤️"
+            emoji_range = (0, 1)
         elif top_emotion == "anger":
-            emotional_context = "User: Frustrated. Vibe: De-escalate — calm, solution-first."
+            emotional_context = "User: Frustrated. Vibe: De-escalate — calm, solution-first, avoid sarcasm."
+            emoji_examples = "🙏 🛠️"
+            emoji_range = (0, 1)
         elif top_emotion == "fear":
             emotional_context = "User: Anxious. Vibe: Reassure and clarify."
+            emoji_examples = "🤝 🛡️"
+            emoji_range = (0, 1)
         elif top_emotion == "surprise":
             emotional_context = "User: Curious/Alert. Vibe: Engage and explain."
+            emoji_examples = "🤔 ✨"
+            emoji_range = (0, 2)
         else:
             emotional_context = "User: Neutral/Professional. Vibe: Helpful and efficient."
+            emoji_examples = "💡 🙂"
+            emoji_range = (0, 2)
 
+        # Verbosity hint (guideline)
+        if "Deep Dive" in conversation_mode:
+            min_words_hint = 70
+        elif "Standard" in conversation_mode:
+            min_words_hint = 30
+        else:
+            min_words_hint = 12
+
+        # Build the context block
         return (
             f"\n[PSYCHOLOGICAL PROFILE]\n"
             f"1. Interaction Mode: {conversation_mode}\n"
             f"2. {emotional_context}\n"
-            f"3. Directive: Mirror user's energy; keep follow-ups short and on-topic.\n"
+            f"3. Emoji Suggestions: Use {emoji_range[0]}–{emoji_range[1]} emoji(s). Examples: {emoji_examples}\n"
+            f"4. Minimum Word Guidance: Aim for ~{min_words_hint} words unless user explicitly requests 'short' or 'brief'.\n"
+            f"5. Directive: Mirror user's energy; prefer natural phrasing and avoid robotic one-line replies.\n"
         )
     except Exception as e:
         print(f"Context Error: {e}")
-        return ""
+        # Safe default
+        return (
+            "\n[PSYCHOLOGICAL PROFILE]\n"
+            "1. Interaction Mode: Standard Chat Mode (Balanced)\n"
+            "2. User: Neutral. Vibe: Helpful and efficient.\n"
+            "3. Emoji Suggestions: Use 0–1 emoji. Examples: 🙂\n"
+            "4. Minimum Word Guidance: Aim for ~30 words.\n"
+        )