from pathlib import Path
from huggingface_hub import snapshot_download

def download_model(model_name, save_dir="models"):
    # 创建保存目录
    save_path = Path(save_dir) / model_name.split("/")[-1]

    if save_path.exists():
        print(f"✅ 模型已存在: {save_path}")
        return str(save_path)
    
    save_path.mkdir(parents=True, exist_ok=True)
    
    download_path = snapshot_download(
        repo_id=model_name,
        local_dir=save_path,
        local_dir_use_symlinks=False
    )
    print(f"✅ 下载完成: {download_path}")
    return download_path