#!/usr/bin/env python3
"""Download ECAPA-TDNN model via ModelScope to /share/models/ecapa-tdnn/"""
import os, sys, time

MODEL_ID = 'iic/speech_ecapa-tdnn_sv_en_voxceleb_16k'
TARGET = '/share/models/ecapa-tdnn'

os.makedirs(TARGET, exist_ok=True)

# Try ModelScope first
for attempt in range(3):
    try:
        from modelscope import snapshot_download
        print(f"Attempt {attempt+1}: ModelScope download...")
        snapshot_download(MODEL_ID, local_dir=TARGET)
        print("SUCCESS via ModelScope")
        sys.exit(0)
    except Exception as e:
        print(f"ModelScope failed: {e}")
        time.sleep(10)

# Fallback: try HF mirror
print("Trying HF mirror...")
os.system(f'export HF_ENDPOINT=https://hf-mirror.com && python3 -c "from huggingface_hub import snapshot_download; snapshot_download(\'speechbrain/spkrec-ecapa-voxceleb\', local_dir=\'{TARGET}\')"')