emojis.py

# emojis.py
# JusticeAI Emoji Intelligence Engine 🧠⚖️
# One-file version — handles emotions, tone, and context adaptively.

import random

# ============================================================
# CORE EMOTIONS 😄😭😡
# ============================================================
EMOJIS = {
    # --- Primary human emotions ---
    "joy": ["🙂", "😊", "😄", "😁", "😆", "😂", "🤣", "😹"],
    "love": ["🥰", "😍", "😘", "💖", "❤️", "💘", "💞", "💓"],
    "surprise": ["😲", "🤯", "😮", "😳", "🙀"],
    "anger": ["😠", "😡", "🤬", "💢", "😤"],
    "fear": ["😨", "😱", "😰", "😧", "😖"],
    "sadness": ["😞", "😢", "😔", "😭", "🥺"],
    "disgust": ["🤢", "🤮", "😒", "😤"],
    "neutral": ["😐", "😑", "🤖", "😶"],

    # ============================================================
    # CONVERSATION FLOW 💬
    # ============================================================
    "greeting": ["👋", "🙋‍♂️", "🙋‍♀️", "🌞"],
    "thinking": ["🤔", "🧠", "💭", "🔍"],
    "ack": ["✅", "👍", "👌", "🙌"],
    "unsure": ["🤷‍♂️", "🤷‍♀️", "😕", "😅"],
    "learning": ["📚", "🧩", "🧠", "🔎", "💡"],
    "success": ["🎉", "🏆", "🌟", "🚀", "🔥"],
    "apology": ["🙏", "😬", "🥺"],
    "encouragement": ["💪", "🌈", "✨", "💫", "🤝"],
    "humor": ["😂", "🤣", "😜", "😎", "🤭", "🤪"],
    "farewell": ["👋", "🌙", "💫", "🌌"],

    # ============================================================
    # CONTEXTUAL EMOJIS 🌍⚙️
    # ============================================================
    "ai": ["🤖", "💻", "🧠", "🔬", "🧮"],
    "justice": ["⚖️", "🕊️", "📜", "🏛️", "👨‍⚖️"],
    "nature": ["🌿", "🌸", "🌍", "🌞", "🌙", "🌊", "🌈"],
    "energy": ["⚡", "🔥", "💥", "🚀"],
    "time": ["⏳", "🕰️", "⌛", "⏱️"],
    "reward": ["🎁", "🎯", "🏅", "💎"],
    "warning": ["⚠️", "🚨", "🛑"],
    "celebration": ["🎊", "🎉", "🥳", "🍾"],
    "philosophy": ["🌌", "🌙", "🔮", "🪞", "🧘‍♂️"],

    # ============================================================
    # SOCIAL / META ✨
    # ============================================================
    "cool": ["😎", "🤙", "🆒", "💥"],
    "nerdy": ["🤓", "🧠", "👓", "📖"],
    "curiosity": ["🕵️‍♂️", "🔎", "🧐"],
    "motivation": ["🚀", "🌟", "🏆", "🔥"],
    "comfort": ["🤗", "💞", "🌸"],
    "celebrate": ["🎊", "🥳", "🎈"],
}

# Merge all keys for validation
ALL_CATEGORIES = set(EMOJIS.keys())

# ============================================================
# SMART SELECTOR FUNCTIONS 🧩
# ============================================================

def get_emoji(category: str, intensity: float = 0.5) -> str:
    """
    Return an emoji from a given category, with intensity (0–1).
    Higher intensity → stronger emotion (later in list).
    """
    if category not in EMOJIS:
        return "🤖"
    emojis = EMOJIS[category]
    index = min(int(len(emojis) * intensity), len(emojis) - 1)
    return emojis[index]


def random_emoji(category: str) -> str:
    """Return a random emoji from a category."""
    return random.choice(EMOJIS.get(category, ["🤖"]))


def merge_emojis(*categories, count=2) -> str:
    """
    Mix emojis from multiple categories.
    Example: merge_emojis("joy", "ai") → 😄🤖
    """
    selected = []
    for cat in categories:
        if cat in EMOJIS:
            selected.append(random.choice(EMOJIS[cat]))
    random.shuffle(selected)
    return " ".join(selected[:count]) if selected else "🤖"


def search_emoji(keyword: str) -> list:
    """
    Search emojis by keyword in category names.
    Returns a flat list of matches.
    """
    result = []
    keyword = keyword.lower()
    for cat, emojis in EMOJIS.items():
        if keyword in cat:
            result.extend(emojis)
    return result or ["🤖"]


def all_emojis() -> list:
    """Return a flattened list of all emojis available."""
    return [e for group in EMOJIS.values() for e in group]


def get_category_for_mood(mood_label: str) -> str:
    """
    Map model mood labels (e.g. 'happy', 'sad', 'angry')
    to an emoji category name.
    """
    mapping = {
        "happy": "joy",
        "joy": "joy",
        "sad": "sadness",
        "anger": "anger",
        "angry": "anger",
        "fear": "fear",
        "disgust": "disgust",
        "surprise": "surprise",
        "neutral": "neutral",
        "love": "love",
        "excited": "success",
    }
    return mapping.get(mood_label.lower(), "neutral")

# ============================================================
# TEST (only runs when executed directly)
# ============================================================
if __name__ == "__main__":
    print("🧠 Total categories:", len(EMOJIS))
    print("🌐 Total emojis:", len(all_emojis()))
    print("🔥 Random test:", merge_emojis("joy", "ai", count=3))
    print("😄 Joy (0.9 intensity):", get_emoji("joy", 0.9))
    print("🕵️ Search 'ai':", search_emoji("ai"))