"""Download Piper TTS Indonesian voice model."""
import urllib.request
import os
import sys

models = {
    'id_ID-ardianto-medium.onnx': 'https://huggingface.co/rhasspy/piper-voices/resolve/main/id/id_ID/ardianto/medium/id_ID-ardianto-medium.onnx',
    'id_ID-ardianto-medium.onnx.json': 'https://huggingface.co/rhasspy/piper-voices/resolve/main/id/id_ID/ardianto/medium/id_ID-ardianto-medium.onnx.json',
}

dest = os.path.expanduser('~/.piper/models')
os.makedirs(dest, exist_ok=True)

for name, url in models.items():
    path = os.path.join(dest, name)
    if os.path.exists(path):
        print(f'Already exists: {name}')
        continue
    print(f'Downloading {name}...')
    try:
        urllib.request.urlretrieve(url, path)
        print(f'  OK ({os.path.getsize(path)} bytes)')
    except Exception as e:
        print(f'  Failed: {e}')

print('Done!')