#!/usr/bin/env bash
# scripts/deploy_huggingface.sh — publica o LLM Fiscal no Hugging Face Hub
set -euo pipefail

: "${HF_TOKEN:?Defina HF_TOKEN antes de executar}"
: "${HF_REPO_ID:?Defina HF_REPO_ID (ex: LesterCerioli/Speed)}"

MODEL_DIR=${MODEL_DIR:-output}
PRIVATE=${PRIVATE:-false}
COMMIT_MSG=${COMMIT_MSG:-"Upload LLM Fiscal Brasileiro checkpoint"}

echo "==> Publicando LLM Fiscal no Hugging Face Hub"
echo "    repo   : $HF_REPO_ID"
echo "    modelo : $MODEL_DIR"
echo "    privado: $PRIVATE"
echo ""

python3 - <<EOF
import os
from huggingface_hub import HfApi

api = HfApi(token=os.environ["HF_TOKEN"])

api.create_repo(
    repo_id=os.environ["HF_REPO_ID"],
    repo_type="model",
    private=os.environ.get("PRIVATE", "false") == "true",
    exist_ok=True,
)

api.upload_folder(
    folder_path=".",
    repo_id=os.environ["HF_REPO_ID"],
    repo_type="model",
    ignore_patterns=[
        "*.pyc",
        "**/__pycache__/**",
        ".git/**",
        "output/**",
        "scripts/**",
    ],
    commit_message=os.environ.get("COMMIT_MSG", "Upload checkpoint"),
)

print(f"✅ Modelo publicado em: https://huggingface.co/{os.environ['HF_REPO_ID']}")
EOF