"""Push hv-reader to the Hugging Face Hub."""
import os

from huggingface_hub import HfApi, create_repo

from hv_reader import HVReader

REPO_ID = "zeechimp/hv-reader"


def main() -> None:
    api = HfApi()
    create_repo(REPO_ID, repo_type="model", exist_ok=True)

    model = HVReader()
    model.save_pretrained("./hv_reader_model")

    api.upload_folder(
        folder_path="./hv_reader_model",
        repo_id=REPO_ID,
        path_in_repo=".",
        commit_message="Upload hv-reader config",
    )

    for fname in [
        "hv_reader.py", "app.py", "example.py", "push_to_hub.py",
        "README.md", "requirements.txt", "LICENSE", ".gitattributes",
    ]:
        if not os.path.exists(fname):
            print(f"  skip (missing): {fname}")
            continue
        api.upload_file(
            path_or_fileobj=fname,
            path_in_repo=fname,
            repo_id=REPO_ID,
            commit_message=f"Upload {fname}",
        )
        print(f"  uploaded: {fname}")

    print()
    print(f"pushed to https://huggingface.co/{REPO_ID}")


if __name__ == "__main__":
    main()