import os
from huggingface_hub import Repository


def download_repo():
    hf_token = os.getenv("HF_TOKEN")
    if not hf_token:
        raise ValueError("HF_TOKEN not found in environment variables.")

    repo_id = "Pujan-Dev/test"
    local_dir = "../Ai-Text-Detector/"

    repo = Repository(local_dir, clone_from=repo_id, token=hf_token)
    print(f"Repository downloaded to: {local_dir}")


if __name__ == "__main__":
    download_repo()