import os
import subprocess

from huggingface_hub import login, snapshot_download

HF_TOKEN = os.getenv("HF_TOKEN")
if HF_TOKEN is None:
    raise RuntimeError("HF_TOKEN is not set")

CODE_REPO_ID = "elsamueldev/confia-backend-code"

login(HF_TOKEN)

snapshot_download(
    repo_id=CODE_REPO_ID,
    repo_type="dataset",
    local_dir=".",
    local_dir_use_symlinks=False
)

local_dir = subprocess.run(["ls"], stdout=subprocess.PIPE).stdout.decode("utf-8")
print(local_dir)
if "main.py" not in local_dir:
    raise RuntimeError("main.py not found in the downloaded dataset")

subprocess.run(["python", "main.py"])