import os
from huggingface_hub import snapshot_download

def maybe_download_model(repo_id, local_dir):
    if not os.path.exists(local_dir):
        print(f"Downloading {repo_id} into {local_dir}")
        snapshot_download(repo_id=repo_id, local_dir=local_dir, local_dir_use_symlinks=False)
    else:
        print(f"✅ {repo_id} already downloaded to {local_dir}")

if __name__ == "__main__":
    maybe_download_model("wedyanessam/fantasy-talking-model2", "./models/fantasytalking_model")
    maybe_download_model("guoyww/wan-video", "./models/Wan2.1-I2V-14B-720P")