import subprocess, shlex, os


assets_folder = "./assets/pretrained_v2/"
if not os.path.exists(assets_folder):
    os.makedirs(assets_folder)

files = {
    # Ov2 Super
    "f0Ov2Super32kD.pth": "https://huggingface.co/ORVC/Ov2Super/resolve/main/f0Ov2Super32kD.pth",
    "f0Ov2Super40kG.pth": "https://huggingface.co/ORVC/Ov2Super/resolve/main/f0Ov2Super40kG.pth",
    
    # TITAN
    "D-f040k-TITAN-Medium.pth": "https://huggingface.co/blaise-tk/TITAN/blob/main/models/medium/40k/pretrained/D-f040k-TITAN-Medium.pth",
    "G-f032k-TITAN-Medium.pth": "https://huggingface.co/blaise-tk/TITAN/blob/main/models/medium/32k/pretrained/D-f032k-TITAN-Medium.pth",


    # Snowie V3
    "D_SnowieV3.1_40k.pth": "https://huggingface.co/MUSTAR/SnowieV3.1-40k/resolve/main/D_SnowieV3.1_40k.pth",
    "G_SnowieV3.1_48k.pth": "https://huggingface.co/MUSTAR/SnowieV3.1-48k/resolve/main/G_SnowieV3.1_48k.pth",

    # RIN E3
    "RIN_E3_G.pth": "https://huggingface.co/MUSTAR/RIN_E3/resolve/main/RIN_E3_G.pth",
    "RIN_E3_D.pth": "https://huggingface.co/MUSTAR/RIN_E3/resolve/main/RIN_E3_D.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}")