from pathlib import Path
import argparse
from huggingface_hub import HfApi, create_repo, upload_folder


def main():
    ap = argparse.ArgumentParser()
    ap.add_argument('--repo-id', required=True, help='e.g. yourname/gemma-4-26B-A4B-it-GGUF')
    ap.add_argument('--private', action='store_true')
    ap.add_argument('--folder', default=str(Path(__file__).resolve().parent))
    args = ap.parse_args()

    api = HfApi()
    create_repo(repo_id=args.repo_id, repo_type='model', private=args.private, exist_ok=True)
    upload_folder(
        repo_id=args.repo_id,
        repo_type='model',
        folder_path=args.folder,
        commit_message='Upload GGUF model',
    )
    print(f'Uploaded to https://huggingface.co/{args.repo_id}')


if __name__ == '__main__':
    main()