from __future__ import annotations

from pathlib import Path

from huggingface_hub import HfApi


USERNAME = "Sami94"
ROOT = Path(__file__).resolve().parents[1]


MODEL_REPOS = {
    f"{USERNAME}/ppo-LunarLander-v2": ROOT / "artifacts/unit1/ppo-LunarLander-v2",
    f"{USERNAME}/q-FrozenLake-v1-4x4-noSlippery": ROOT / "artifacts/unit2/q-FrozenLake-v1-4x4-noSlippery",
    f"{USERNAME}/q-Taxi-v3": ROOT / "artifacts/unit2/q-Taxi-v3",
}

COURSE_REPO = f"{USERNAME}/hf-deep-rl-course-certification"


def upload_model_repo(api: HfApi, repo_id: str, folder: Path) -> None:
    if not folder.exists():
        print(f"SKIP {repo_id}: missing folder {folder}")
        return

    api.create_repo(repo_id=repo_id, repo_type="model", exist_ok=True)
    api.upload_folder(repo_id=repo_id, repo_type="model", folder_path=folder, path_in_repo=".")
    print(f"Uploaded model repo: https://huggingface.co/{repo_id}")


def upload_course_folder(api: HfApi) -> None:
    api.create_repo(repo_id=COURSE_REPO, repo_type="model", exist_ok=True)
    api.upload_folder(
        repo_id=COURSE_REPO,
        repo_type="model",
        folder_path=ROOT,
        path_in_repo=".",
        ignore_patterns=[
            ".git/*",
            "__pycache__/*",
            "*.pyc",
        ],
    )
    print(f"Uploaded course folder: https://huggingface.co/{COURSE_REPO}")


def main() -> None:
    api = HfApi()
    whoami = api.whoami()
    print(f"Authenticated as: {whoami.get('name')} ({whoami.get('fullname')})")

    for repo_id, folder in MODEL_REPOS.items():
        upload_model_repo(api, repo_id, folder)

    upload_course_folder(api)


if __name__ == "__main__":
    main()