avatar/sentiment_emoji_map.py

"""
Sentiment Emoji Map - Maps emotion labels to face emojis

This module maps detected emotions to visual emoji representations
Unified for both user and AI (same emojis for all)

Version 2.0: Added new emotions for multi-emotion model support
"""

from typing import Dict, Optional


class EmojiMapper:
    """Maps emotion labels to face emojis - unified for user and AI"""
    
    EMOTION_EMOJIS: Dict[str, str] = {
        # === POSITIVE EMOTIONS (High Arousal) ===
        "ecstasy": "🤩",
        "joy": "😁",
        "happiness": "😊",
        "delight": "😄",
        "elation": "😆",
        "euphoria": "🥳",
        "excitement": "😃",
        "thrill": "😝",
        "enthusiasm": "😀",
        "cheerfulness": "😊",
        
        # === POSITIVE EMOTIONS (Medium Arousal) ===
        "contentment": "😌",
        "satisfaction": "😏",
        "pleasure": "😋",
        "relief": "😮‍💨",
        "serenity": "😇",
        "calm": "😌",
        "relaxed": "😎",
        "pride": "😤",
        "confidence": "😎",
        "triumph": "😏",
        
        # === LOVE & AFFECTION ===
        "love": "🥰",
        "adoration": "😍",
        "affection": "🥰",
        "tenderness": "🤗",
        "caring": "🤗",
        "compassion": "🥹",
        "empathy": "🥹",
        "gratitude": "🙏",
        "thankful": "😊",
        
        # === INTEREST & CURIOSITY ===
        "curiosity": "🧐",
        "interest": "🤔",
        "fascination": "😯",
        "wonder": "😲",
        "awe": "😮",
        "amazement": "😲",
        "intrigue": "🤨",
        
        # === SURPRISE ===
        "surprise": "😮",
        "astonishment": "😱",
        "shock": "😳",
        "startled": "😧",
        
        # === NEUTRAL / THINKING ===
        "neutral": "😐",
        "thinking": "🤔",
        "contemplative": "🤔",
        "pensive": "😔",
        "reflective": "🤔",
        "uncertain": "😕",
        "ambivalent": "😐",
        "indifferent": "😑",
        
        # === CONFUSION ===
        "confused": "😕",
        "confusion": "😕",
        "puzzled": "🤨",
        "perplexed": "😵‍💫",
        "bewildered": "😵",
        "baffled": "🤯",
        
        # === NEGATIVE EMOTIONS (Low Arousal) ===
        "sadness": "😢",
        "sorrow": "😭",
        "grief": "😭",
        "melancholy": "😔",
        "disappointment": "😞",
        "dejection": "😩",
        "despair": "😫",
        "hopelessness": "😿",
        "loneliness": "🥺",
        "hurt": "😢",
        "misery": "😣",
        
        # === FEAR & ANXIETY ===
        "fear": "😨",
        "terror": "😱",
        "horror": "😱",
        "dread": "😰",
        "anxiety": "😰",
        "worry": "😟",
        "nervousness": "😬",
        "apprehension": "😧",
        "panic": "😱",
        
        # === ANGER & FRUSTRATION ===
        "anger": "😠",
        "rage": "🤬",
        "fury": "😡",
        "irritation": "😤",
        "annoyance": "🙄",
        "frustration": "😤",
        "exasperation": "😫",
        "resentment": "😒",
        "hostility": "😠",
        "bitterness": "😖",
        
        # === DISGUST & CONTEMPT ===
        "disgust": "🤢",
        "revulsion": "🤮",
        "contempt": "😒",
        "disdain": "🙄",
        "loathing": "😤",
        "scorn": "😏",
        
        # === SHAME & EMBARRASSMENT ===
        "shame": "😳",
        "embarrassment": "😳",
        "guilt": "😣",
        "regret": "😔",
        "remorse": "😞",
        "humiliation": "🫣",
        
        # === BOREDOM & TIREDNESS ===
        "boredom": "😑",
        "tiredness": "😴",
        "exhaustion": "😩",
        "fatigue": "🥱",
        "weariness": "😮‍💨",
        "sleepy": "😴",
        
        # === ENVY & JEALOUSY ===
        "envy": "😒",
        "jealousy": "🙄",
        
        # === PLAYFUL & SILLY ===
        "playful": "😜",
        "silly": "🤪",
        "mischievous": "😈",
        "teasing": "😏",
        "sarcastic": "😏",
        "witty": "😎",
        "amusement": "😂",
        "funny": "🤣",
        
        # === SPECIAL STATES ===
        "sympathy": "🥹",
        "nostalgia": "🥲",
        "hope": "🙂",
        "optimism": "😊",
        "pessimism": "😕",
        "longing": "🥺",
        "yearning": "😢",
        
        # === NEW IN V2 ===
        "determination": "😤",
        "inspiration": "🤩",  # Changed from ✨ to face emoji
        "anticipation": "🤗",
        "trust": "😊",  # Changed from 🤝 to face emoji
        "acceptance": "😌",
        "sarcasm": "🙄",
        
        # === NEW IN V3 - Context Emotions ===
        "indifferent": "😐",
        "admiration": "🤩",
        "despair": "😰",
        "greeting": "😊",  # Changed from 👋 to face emoji
        "heartbreak": "😭",  # Changed from 💔 to face emoji,
        
        # === LEGACY MAPPINGS ===
        "very_negative": "😭",
        "negative": "😠",
        "slightly_negative": "😕",
        "slightly_positive": "🙂",
        "positive": "😊",
        "very_positive": "🤩",
        
        # Default
        "default": "😐",
        "ready": "😐",
    }
    
    def __init__(self, custom_emojis: Optional[Dict[str, str]] = None):
        """Initialize with default or custom emoji mappings"""
        self.emojis = self.EMOTION_EMOJIS.copy()
        if custom_emojis:
            self.emojis.update(custom_emojis)
    
    def get_emoji(self, emotion: str) -> str:
        """
        Get emoji for a given emotion
        
        Args:
            emotion: Emotion label (e.g., 'joy', 'anger', 'curiosity')
            
        Returns:
            Face emoji string
        """
        emotion_clean = emotion.lower().strip().replace(" ", "_").replace("-", "_")
        return self.emojis.get(emotion_clean, self.emojis.get("default", "😐"))
    
    def get_all_emojis(self) -> Dict[str, str]:
        """Get all emotion→emoji mappings"""
        return self.emojis.copy()


if __name__ == "__main__":
    mapper = EmojiMapper()
    test_emotions = [
        "joy", "happiness", "love", "sadness", "anger",
        "fear", "surprise", "disgust", "curiosity", "neutral"
    ]
    
    print("Emotion → Emoji Mappings:")
    for emotion in test_emotions:
        emoji = mapper.get_emoji(emotion)
        print(f"  {emotion:20} → {emoji}")
    
    print(f"\nTotal emotions: {len(mapper.get_all_emojis())}")