"""
upload_to_hub.py
-----------------
Uploads this folder as a model repo to Hugging Face Hub.

Setup (once):
    pip install huggingface_hub
    huggingface-cli login          # paste a WRITE token from
                                    # https://huggingface.co/settings/tokens

Then:
    python upload_to_hub.py --repo-id your-username/asl-fingerspelling-transformer
"""

import argparse

from huggingface_hub import HfApi, create_repo

if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--repo-id", required=True, help="e.g. your-username/asl-fingerspelling-transformer")
    parser.add_argument("--private", action="store_true", help="Create/upload as a private repo")
    parser.add_argument("--folder", default=".", help="Local folder to upload (default: current dir)")
    args = parser.parse_args()

    create_repo(args.repo_id, repo_type="model", private=args.private, exist_ok=True)

    api = HfApi()
    api.upload_folder(
        folder_path=args.folder,
        repo_id=args.repo_id,
        repo_type="model",
        ignore_patterns=["*.ipynb_checkpoints*", "__pycache__/*"],
    )

    print(f"Done. View it at: https://huggingface.co/{args.repo_id}")