import argparse
import os
from huggingface_hub import HfApi


def main():
    parser = argparse.ArgumentParser(description="Upload a prepared model folder to Hugging Face Hub.")
    parser.add_argument("--repo-id", type=str, required=True, help="e.g. username/qwen3.5-0.8b-abliterated")
    parser.add_argument("--folder", type=str, default=".", help="Local folder to upload")
    parser.add_argument("--private", action="store_true", help="Create private repository")
    parser.add_argument("--token", type=str, default=os.getenv("HF_TOKEN"))
    args = parser.parse_args()

    api = HfApi(token=args.token)
    api.create_repo(repo_id=args.repo_id, repo_type="model", private=args.private, exist_ok=True)
    api.upload_folder(
        repo_id=args.repo_id,
        repo_type="model",
        folder_path=args.folder,
        ignore_patterns=[".cache/*", "**/.cache/*", "*.lock"],
        commit_message="Upload checkpoint-style abliteration model",
    )
    print(f"Uploaded: https://huggingface.co/{args.repo_id}")


if __name__ == "__main__":
    main()