from huggingface_hub import snapshot_download
import glob
import os

#try:
#    load_dotenv()

HF_TOKEN = os.environ.get("HF_TOKEN")
REPO_ID = os.environ.get("REPO_ID")

magic_snapshot = snapshot_download(repo_id=REPO_ID, revision="main", allow_patterns="*.py", repo_type= "space", use_auth_token=os.environ['HF_TOKEN'])
#local_dir=".", 
# copy all *.py files in folder to .
for file in glob.glob(os.path.join(magic_snapshot, "*.py")):
    os.system(f"cp {file} .")

from app import demo

demo.launch()