upload_to_hf.py

"""Upload this project as a Hugging Face Space."""

from __future__ import annotations

import argparse
from pathlib import Path

from huggingface_hub import HfApi, create_repo, upload_folder


def main():
    parser = argparse.ArgumentParser(description="Upload coding-llm project to Hugging Face Space.")
    parser.add_argument("--repo-id", required=True, help="Example: username/coding-llm-space")
    parser.add_argument("--token", required=True, help="Hugging Face access token")
    parser.add_argument("--private", action="store_true", help="Create private repo/space")
    args = parser.parse_args()

    root = Path(__file__).resolve().parent
    api = HfApi(token=args.token)
    create_repo(
        repo_id=args.repo_id,
        token=args.token,
        repo_type="space",
        private=args.private,
        exist_ok=True,
        space_sdk="gradio",
    )
    upload_folder(
        folder_path=str(root),
        repo_id=args.repo_id,
        repo_type="space",
        token=args.token,
        ignore_patterns=[".git*", "__pycache__", "*.pyc"],
    )
    info = api.repo_info(repo_id=args.repo_id, repo_type="space")
    print(f"Uploaded successfully: {info.id}")


if __name__ == "__main__":
    main()