"""Upload a finished run (metrics + final model) to a private HF model repo under runs/<run>/."""

import argparse
from pathlib import Path

from huggingface_hub import HfApi


def main():
    p = argparse.ArgumentParser()
    p.add_argument("--repo", default="Broyojo/world-pretraining-runs")
    p.add_argument("--run", required=True)
    p.add_argument("dirs", nargs="+", help="local dirs whose contents go to runs/<run>/")
    args = p.parse_args()
    api = HfApi()
    api.create_repo(args.repo, private=True, exist_ok=True)
    for d in args.dirs:
        if Path(d).is_dir():
            api.upload_folder(repo_id=args.repo, folder_path=d, path_in_repo=f"runs/{args.run}",
                              commit_message=f"{args.run}: {Path(d).name}")
            print(f"uploaded {d} -> {args.repo}/runs/{args.run}")


if __name__ == "__main__":
    main()