"""
Dataset de exemplo embutido, usado como fallback quando não há
acesso à internet ou o usuário quer testar rápido sem baixar nada.

Em produção (Hugging Face Space), troque por um dataset real do Hub, ex:

    from datasets import load_dataset
    ds = load_dataset("ag_news", split="train[:2000]")
    documents = [{"id": str(i), "text": row["text"], "category": str(row["label"])}
                 for i, row in enumerate(ds)]

Veja load_documents() em app.py — é só trocar essa função.
"""

SAMPLE_DOCUMENTS = [
    # Tecnologia
    {"id": "1", "text": "Novo modelo de linguagem bate recorde em benchmarks de raciocínio matemático.", "category": "tecnologia"},
    {"id": "2", "text": "Empresa lança chip capaz de rodar inferência de IA com metade do consumo de energia.", "category": "tecnologia"},
    {"id": "3", "text": "Framework open source para agentes de IA ultrapassa 50 mil estrelas no GitHub.", "category": "tecnologia"},
    {"id": "4", "text": "Pesquisadores apresentam técnica de compressão de embeddings sem perda de precisão.", "category": "tecnologia"},
    {"id": "5", "text": "Startup brasileira de IA generativa recebe aporte para expandir para o exterior.", "category": "tecnologia"},
    {"id": "6", "text": "Vazamento de dados expõe falha de segurança em API amplamente usada por desenvolvedores.", "category": "tecnologia"},
    {"id": "7", "text": "Nova versão de banco de dados vetorial promete buscas 3x mais rápidas em larga escala.", "category": "tecnologia"},
    {"id": "8", "text": "Big tech anuncia investimento bilionário em data centers voltados para treinamento de IA.", "category": "tecnologia"},

    # Esportes
    {"id": "9", "text": "Time surpreende ao vencer final de campeonato após virada nos últimos minutos.", "category": "esportes"},
    {"id": "10", "text": "Atleta bate recorde mundial dos 100 metros rasos em competição na Europa.", "category": "esportes"},
    {"id": "11", "text": "Clube anuncia contratação de novo técnico após sequência de derrotas.", "category": "esportes"},
    {"id": "12", "text": "Seleção nacional se classifica para a próxima fase do torneio com goleada.", "category": "esportes"},
    {"id": "13", "text": "Jogador sofre lesão grave e deve ficar fora dos gramados por meses.", "category": "esportes"},
    {"id": "14", "text": "Torcida lota estádio para acompanhar decisão do campeonato regional.", "category": "esportes"},
    {"id": "15", "text": "Federação anuncia mudanças nas regras do esporte para a próxima temporada.", "category": "esportes"},

    # Economia
    {"id": "16", "text": "Banco central mantém taxa de juros e sinaliza cautela com a inflação.", "category": "economia"},
    {"id": "17", "text": "Bolsa de valores fecha em alta impulsionada por resultados do setor financeiro.", "category": "economia"},
    {"id": "18", "text": "Governo anuncia pacote de medidas para estimular a indústria nacional.", "category": "economia"},
    {"id": "19", "text": "Dólar recua após dados de emprego melhores que o esperado nos EUA.", "category": "economia"},
    {"id": "20", "text": "Inflação desacelera pelo terceiro mês consecutivo, aponta relatório oficial.", "category": "economia"},
    {"id": "21", "text": "Empresa de varejo reporta queda nas vendas em meio à retração do consumo.", "category": "economia"},
    {"id": "22", "text": "Especialistas projetam crescimento moderado do PIB para o próximo ano.", "category": "economia"},

    # Saúde
    {"id": "23", "text": "Estudo aponta relação entre sono de qualidade e redução de doenças cardíacas.", "category": "saude"},
    {"id": "24", "text": "Nova vacina mostra eficácia acima de 90% em testes clínicos de fase 3.", "category": "saude"},
    {"id": "25", "text": "Ministério da Saúde amplia campanha de vacinação para grupos prioritários.", "category": "saude"},
    {"id": "26", "text": "Pesquisa relaciona uso excessivo de telas com aumento de ansiedade em jovens.", "category": "saude"},
    {"id": "27", "text": "Hospital adota inteligência artificial para agilizar diagnóstico de exames de imagem.", "category": "saude"},
    {"id": "28", "text": "Organização mundial alerta para aumento de casos de doença respiratória no inverno.", "category": "saude"},

    # Entretenimento
    {"id": "29", "text": "Filme independente surpreende e lidera bilheteria no fim de semana de estreia.", "category": "entretenimento"},
    {"id": "30", "text": "Cantora anuncia turnê mundial com passagem confirmada pelo Brasil.", "category": "entretenimento"},
    {"id": "31", "text": "Série de streaming é renovada para nova temporada após sucesso de audiência.", "category": "entretenimento"},
    {"id": "32", "text": "Festival de cinema revela lista de filmes selecionados para a mostra deste ano.", "category": "entretenimento"},
    {"id": "33", "text": "Ator vence prêmio internacional por atuação em produção aclamada pela crítica.", "category": "entretenimento"},
]