"""Tool: classifica em qual nível escolar um conteúdo se encaixa.

Heurística simples baseada em palavras-chave do currículo BNCC.
Útil quando o aluno descreve um problema sem dizer a série.
"""
from typing import Literal

_KEYWORDS_FUNDAMENTAL_1: list[str] = [
    "soma", "subtração", "adição", "tabuada", "número natural",
    "alfabeto", "sílaba", "leitura", "primeiro ano", "segundo ano",
    "terceiro ano", "quarto ano", "quinto ano",
]
_KEYWORDS_FUNDAMENTAL_2: list[str] = [
    "fração", "decimal", "porcentagem", "regra de três", "mmc", "mdc",
    "área", "perímetro", "ângulo", "polígono", "sexto ano", "sétimo ano",
    "oitavo ano", "nono ano",
]
_KEYWORDS_MEDIO: list[str] = [
    "equação", "função", "logaritmo", "trigonometria", "matriz",
    "vetor", "bhaskara", "primeiro grau", "segundo grau", "geometria analítica",
    "ensino médio", "vestibular", "enem",
]


def run_grade_level_classifier(text: str) -> str:
    """Classifica em qual etapa da educação básica o tema se encaixa."""
    t = text.lower()

    def _count(keywords: list[str]) -> int:
        return sum(1 for kw in keywords if kw in t)

    scores = {
        "fundamental_anos_iniciais": _count(_KEYWORDS_FUNDAMENTAL_1),
        "fundamental_anos_finais": _count(_KEYWORDS_FUNDAMENTAL_2),
        "ensino_medio": _count(_KEYWORDS_MEDIO),
    }
    best: str = max(scores, key=lambda k: scores[k])
    if scores[best] == 0:
        return (
            "Não foi possível classificar com confiança. "
            "Sugestão: pedir ao aluno em qual série/ano ele está."
        )

    labels: dict[str, str] = {
        "fundamental_anos_iniciais": "Ensino Fundamental · Anos Iniciais (1º-5º)",
        "fundamental_anos_finais": "Ensino Fundamental · Anos Finais (6º-9º)",
        "ensino_medio": "Ensino Médio (1º-3º ano)",
    }
    return (
        f"Nível classificado: {labels[best]} "
        f"(score: {scores[best]} palavras-chave encontradas). "
        f"Adapte sua resposta para este nível."
    )


grade_level_classifier_tool: dict = {
    "type": "function",
    "function": {
        "name": "grade_level_classifier",
        "description": (
            "Classifica em qual etapa da educação básica brasileira (Fund. Iniciais, "
            "Fund. Finais, Ensino Médio) o conteúdo se encaixa. Use no INÍCIO da "
            "interação quando o aluno não especifica a série."
        ),
        "parameters": {
            "type": "object",
            "properties": {
                "text": {
                    "type": "string",
                    "description": "Texto a classificar (a pergunta do aluno ou tópico).",
                }
            },
            "required": ["text"],
            "additionalProperties": False,
        },
    },
}