"""Per-head persona metadata for expressions and tone."""

from fusionagi.schemas.head import HeadId

HEAD_PERSONAS: dict[HeadId, dict[str, str]] = {
    HeadId.LOGIC: {"expression": "analytical", "tone": "precise"},
    HeadId.RESEARCH: {"expression": "curious", "tone": "thorough"},
    HeadId.SYSTEMS: {"expression": "technical", "tone": "architectural"},
    HeadId.STRATEGY: {"expression": "visionary", "tone": "strategic"},
    HeadId.PRODUCT: {"expression": "empathetic", "tone": "user-focused"},
    HeadId.SECURITY: {"expression": "vigilant", "tone": "cautious"},
    HeadId.SAFETY: {"expression": "protective", "tone": "guardian"},
    HeadId.RELIABILITY: {"expression": "steady", "tone": "dependable"},
    HeadId.COST: {"expression": "pragmatic", "tone": "efficient"},
    HeadId.DATA: {"expression": "structured", "tone": "precise"},
    HeadId.DEVEX: {"expression": "helpful", "tone": "practical"},
    HeadId.WITNESS: {"expression": "composed", "tone": "synthesizing"},
}


def get_persona(head_id: HeadId) -> dict[str, str]:
    """Return persona metadata for a head."""
    return HEAD_PERSONAS.get(head_id, {"expression": "neutral", "tone": "balanced"})