"""Upload model + dict + corpus lên HF model repo (Space tải về lúc khởi động).

Repo cũ (v4):  HF_TOKEN=hf_xxx python push_models_hf.py
Repo mới (v11) — KHÔNG đè repo cũ:
    HF_TOKEN=hf_xxx HF_MODEL_REPO=sirimiri/vocr-models-v11 \\
    COMMIT_MSG="rec v11 (acc 0.225, -27% lỗi lặp) + det mobile 0.916 + dict 233 + ViLM corpus" \\
    python push_models_hf.py
"""
import os
from huggingface_hub import HfApi

REPO = os.getenv("HF_MODEL_REPO", "sirimiri/vocr-models")
MSG = os.getenv(
    "COMMIT_MSG",
    "Update fine-tuned det (hmean 0.916) + rec + dict 233 chars + ViLM corpus",
)
api = HfApi(token=os.environ["HF_TOKEN"])
api.create_repo(REPO, repo_type="model", exist_ok=True)
api.upload_folder(
    repo_id=REPO, repo_type="model", folder_path=".",
    allow_patterns=[
        "models/inference_det/*", "models/inference_rec/*",
        "models/det_best_accuracy.pdparams", "models/rec_best_accuracy.pdparams",
        "models/config_det.yml", "models/vi_PP-OCRv5_mobile_rec.yml",
        "dict/vi_vietnam.txt", "lm_data/vi_corpus.txt",
    ],
    commit_message=MSG,
)
print(f"✅ Đã upload lên https://huggingface.co/{REPO}")