import subprocess
import os

base_url = "https://huggingface.co/nolanaatama/models/resolve/main/"
output_directory = "/content/imagination/extensions/microsoftexcel-controlnet/models/"

files_to_download = [
    "control_v11e_sd15_ip2p_fp16.safetensors",
    "control_v11e_sd15_shuffle_fp16.safetensors",
    "control_v11p_sd15_canny_fp16.safetensors",
    "control_v11p_sd15_depth_fp16.safetensors",
    "control_v11p_sd15_inpaint_fp16.safetensors",
    "control_v11p_sd15_lineart_fp16.safetensors",
    "control_v11p_sd15_mlsd_fp16.safetensors",
    "control_v11p_sd15_normalbae_fp16.safetensors",
    "control_v11p_sd15_openpose_fp16.safetensors",
    "control_v11p_sd15_scribble_fp16.safetensors",
    "control_v11p_sd15_seg_fp16.safetensors",
    "control_v11p_sd15_softedge_fp16.safetensors",
    "control_v11p_sd15s2_lineart_anime_fp16.safetensors",
    "control_v11f1e_sd15_tile_fp16.safetensors",
    "t2iadapter_canny_sd14v1.pth",
    "t2iadapter_color_sd14v1.pth",
    "t2iadapter_depth_sd14v1.pth",
    "t2iadapter_keypose_sd14v1.pth",
    "t2iadapter_openpose_sd14v1.pth",
    "t2iadapter_seg_sd14v1.pth",
    "t2iadapter_sketch_sd14v1.pth",
    "t2iadapter_style_sd14v1.pth"
]

os.makedirs(output_directory, exist_ok=True)

for file_name in files_to_download:
    file_url = base_url + file_name
    save_path = os.path.join(output_directory, file_name)
    
    subprocess.run(["wget", file_url, "-O", save_path])
    
    print(f"Downloaded {file_name}")

print("All files downloaded successfully!")