Spaces:
Running
Running
| """ | |
| 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())}") | |