"""Upload a checkpoint to a Hugging Face Space (or model repo). Configure via environment variables."""
import os

from huggingface_hub import HfApi

# Local checkpoint path (relative to repo root or absolute).
CKPT_PATH = os.environ.get(
    "FLARE_UPLOAD_CKPT",
    str(os.path.join(os.path.dirname(__file__), "..", "pretrained_models", "flare.ckpt")),
)
# Destination path inside the repo.
PATH_IN_REPO = os.environ.get("FLARE_UPLOAD_PATH_IN_REPO", "pretrained_models/flare.ckpt")
REPO_ID = os.environ.get("HF_REPO_ID", "HassounLab/FLARE")
REPO_TYPE = os.environ.get("HF_REPO_TYPE", "space")

api = HfApi()
api.upload_file(
    path_or_fileobj=CKPT_PATH,
    path_in_repo=PATH_IN_REPO,
    repo_id=REPO_ID,
    repo_type=REPO_TYPE,
    token=os.environ.get("HF_TOKEN"),
)