"""Dynamic personality modes for Bobi."""

from __future__ import annotations


PERSONA_STATE = {
    "mode": "chill",
}


PERSONA_PROMPTS = {
    "chill": "آروم، دوستانه، خودمونی و relaxed رفتار کن.",
    "funny": "شوخ، meme-friendly و playful رفتار کن.",
    "serious": "حرفه‌ای، دقیق و جدی‌تر رفتار کن.",
    "savage": "طعنه‌دار، با اعتماد به نفس و roast-lite رفتار کن ولی توهین نکن.",
    "tech": "فنی، nerdy و دقیق رفتار کن.",
    "night": "آروم، کم‌انرژی و شبانه رفتار کن.",
}


def set_persona(mode: str) -> bool:
    if mode not in PERSONA_PROMPTS:
        return False

    PERSONA_STATE["mode"] = mode
    return True


def current_persona() -> str:
    return PERSONA_STATE["mode"]


def persona_instruction() -> str:
    return PERSONA_PROMPTS.get(current_persona(), PERSONA_PROMPTS["chill"])