import os
from datasets import load_dataset
from huggingface_hub import login

# 🔒 Obter o Token do ambiente
HF_API_KEY = os.getenv("HF_API_KEY")
if not HF_API_KEY:
    raise ValueError("❌ ERRO: Token Hugging Face não encontrado no ambiente.")

# ✅ Fazer login no Hugging Face
login(HF_API_KEY)

# ✅ Carregar o dataset
dataset = load_dataset("rwayz/tributario", token=HF_API_KEY)

# ✅ Listar os splits disponíveis
print("✅ Splits disponíveis no dataset:")
print(dataset)