from __future__ import annotations

import os
from pathlib import Path

from huggingface_hub import HfApi

REPO_ID = "arun-gharami/lead-ai-customer-predictor"
PROJECT_DIR = Path(__file__).resolve().parent


def push() -> None:
    token = os.environ.get("HF_TOKEN")
    if not token:
        raise EnvironmentError("Set HF_TOKEN before running this script.")

    api = HfApi(token=token)
    api.create_repo(repo_id=REPO_ID, repo_type="model", private=False, exist_ok=True)
    api.upload_folder(
        repo_id=REPO_ID,
        repo_type="model",
        folder_path=str(PROJECT_DIR),
        ignore_patterns=["data/*", "__pycache__/*", "*.pyc", ".DS_Store"],
    )
    print(f"Uploaded project to https://huggingface.co/{REPO_ID}")


if __name__ == "__main__":
    push()