import subprocess, os
assets_folder = "./assets/"
if not os.path.exists(assets_folder):
    os.makedirs(assets_folder)
files = {
    "rmvpe/hpa-rmvpe.pt":"https://huggingface.co/NeoPy/Ultimate-Models/resolve/main/predictors/hpa-rmvpe.pt",
    "hubert/hubert_base.pt":"https://huggingface.co/NeoPy/Ultimate-Models/resolve/main/embedders/fairseq/hubert_base.pt",
    "assets/pretrained_v2/f0Ov2Super32kD.pth":"https://huggingface.co/ORVC/Ov2Super/resolve/main/f0Ov2Super32kD.pth",
    "assets/pretrained_v2/f0Ov2Super32kG.pth":" https://huggingface.co/ORVC/Ov2Super/resolve/main/f0Ov2Super32kG.pth",
}
for file, link in files.items():
    file_path = os.path.join(assets_folder, file)
    if not os.path.exists(file_path):
        try:
            subprocess.run(['wget', link, '-O', file_path], check=True)
        except subprocess.CalledProcessError as e:
            print(f"Error downloading {file}: {e}")