scripts/publish.py

#!/usr/bin/env python3
import argparse
import os

from huggingface_hub import HfApi


def parse_args() -> argparse.Namespace:
    parser = argparse.ArgumentParser(
        description="Upload a local folder to a Hugging Face Space, respecting .gitignore and showing progress."
    )
    parser.add_argument(
        "repo_id",
        help="Space repository ID in the form 'username/space-name'"
    )
    parser.add_argument(
        "--folder-path", "-f",
        default=".",
        help="Local folder path to upload (default: current directory)"
    )
    parser.add_argument(
        "--path-in-repo", "-p",
        default="",
        help="Target path in the repository (default: root)"
    )
    parser.add_argument(
        "--commit-message", "-m",
        default="Upload folder via script",
        help="Commit message for the upload"
    )
    parser.add_argument(
        "--token", "-t",
        default=None,
        help=(
            "Hugging Face access token (will use HF_TOKEN or HUGGINGFACE_HUB_TOKEN "
            "from environment if not provided)"
        )
    )
    parser.add_argument(
        "--large",
        action="store_true",
        help="Use upload_large_folder for large uploads"
    )
    return parser.parse_args()


def main() -> None:
    args = parse_args()
    # Determine token from argument or environment
    token = args.token or os.getenv("HF_TOKEN") or os.getenv("HUGGINGFACE_HUB_TOKEN")
    if not token:
        print(
            "Warning: No token provided via --token or environment variable. "
            "Please set HF_TOKEN/HUGGINGFACE_HUB_TOKEN or run `huggingface-cli login`."
        )

    api = HfApi()

    if args.large:
        api.upload_large_folder(
            folder_path=args.folder_path,
            repo_id=args.repo_id,
            repo_type="space",
            print_report=True
        )
    else:
        api.upload_folder(
            folder_path=args.folder_path,
            path_in_repo=args.path_in_repo,
            repo_id=args.repo_id,
            repo_type="space",
            token=token,
            commit_message=args.commit_message
        )

    print(f"Successfully uploaded '{args.folder_path}' to '{args.repo_id}'.")


if __name__ == "__main__":
    main()