You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
100 lines
4.0 KiB
100 lines
4.0 KiB
#!/usr/bin/env bash |
|
# One-shot host prep for editor stack sidecars: |
|
# 1) LibreTranslate bind mount — dirs + ownership UID 1032 (official image user). |
|
# 2) Piper ONNX voices — trinity + read-aloud extras (rhasspy/piper-voices) into: |
|
# - ./.local-piper-data (dev compose bind mount), and |
|
# - Docker volume <project>_piper-stack-data when it exists (docker-compose.prod.yml Wyoming /data). |
|
# |
|
# Piper download logic lives in this file so you can copy **only** this script to a server and run it |
|
# from the repo root (still need curl, docker; full clone is easier: bash scripts/ensure-libretranslate-dirs.sh). |
|
# |
|
# Internal entry: bash ensure-libretranslate-dirs.sh --download-piper-only [DEST] |
|
# (used by scripts/download-piper-extra-voices.sh — keep voice list in sync with trinity-languages.ts) |
|
# |
|
# Optional env: |
|
# COMPOSE_PROJECT_NAME — Docker Compose project name (default: basename of repo dir), for volume *_piper-stack-data. |
|
# SKIP_PIPER_VOICES=1 — only fix LibreTranslate permissions, do not download Piper. |
|
# HF_BASE — Hugging Face resolve base for Piper ONNX (default rhasspy/piper-voices/main). |
|
set -euo pipefail |
|
|
|
_resolve_root() { |
|
_SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]:-$0}")" && pwd)" |
|
if [[ "$(basename "$_SCRIPT_DIR")" == "scripts" ]]; then |
|
ROOT="$(cd "$_SCRIPT_DIR/.." && pwd)" |
|
else |
|
ROOT="$_SCRIPT_DIR" |
|
fi |
|
} |
|
|
|
# Keep in sync with src/lib/trinity-languages.ts (TRINITY_PIPER_VOICE + EXTRA_READ_ALOUD_PIPER_VOICE) and server voiceMap. |
|
download_piper_voices_to() { |
|
local dest="${1:?destination directory}" |
|
local hf="${HF_BASE:-https://huggingface.co/rhasspy/piper-voices/resolve/main}" |
|
mkdir -p "$dest" |
|
local relpath base_name onnx json |
|
for relpath in \ |
|
"en/en_US/lessac/medium/en_US-lessac-medium" \ |
|
"de/de_DE/thorsten/medium/de_DE-thorsten-medium" \ |
|
"fr/fr_FR/siwis/medium/fr_FR-siwis-medium" \ |
|
"es/es_ES/davefx/medium/es_ES-davefx-medium" \ |
|
"ru/ru_RU/ruslan/medium/ru_RU-ruslan-medium" \ |
|
"zh/zh_CN/huayan/medium/zh_CN-huayan-medium" \ |
|
"pl/pl_PL/darkman/medium/pl_PL-darkman-medium" \ |
|
"nl/nl_NL/mls/medium/nl_NL-mls-medium" \ |
|
"cs/cs_CZ/jirka/medium/cs_CZ-jirka-medium" \ |
|
"tr/tr_TR/dfki/medium/tr_TR-dfki-medium" \ |
|
"ar/ar_JO/kareem/medium/ar_JO-kareem-medium" \ |
|
"it/it_IT/paola/medium/it_IT-paola-medium" \ |
|
"pt/pt_BR/cadu/medium/pt_BR-cadu-medium" |
|
do |
|
base_name="$(basename "$relpath")" |
|
onnx="${dest}/${base_name}.onnx" |
|
json="${dest}/${base_name}.onnx.json" |
|
if [[ -f "$onnx" && -f "$json" ]]; then |
|
echo "Skip (exists): ${base_name}" |
|
continue |
|
fi |
|
echo "Fetching ${base_name} …" |
|
curl -fsSL -o "$onnx" "${hf}/${relpath}.onnx" |
|
curl -fsSL -o "$json" "${hf}/${relpath}.onnx.json" |
|
done |
|
echo "Piper ONNX done → ${dest}" |
|
} |
|
|
|
if [[ "${1:-}" == "--download-piper-only" ]]; then |
|
shift |
|
_resolve_root |
|
download_piper_voices_to "${1:-${PIPER_DOWNLOAD_DIR:-$ROOT/.local-piper-data}}" |
|
exit 0 |
|
fi |
|
|
|
_resolve_root |
|
PROJECT="${COMPOSE_PROJECT_NAME:-$(basename "$ROOT")}" |
|
PIPER_VOL="${PROJECT}_piper-stack-data" |
|
|
|
echo "[ensure] LibreTranslate data dir (UID 1032) …" |
|
mkdir -p "$ROOT/.local-libretranslate/share" "$ROOT/.local-libretranslate/cache" |
|
docker run --rm \ |
|
-v "$ROOT/.local-libretranslate:/d" \ |
|
alpine:3.20 chown -R 1032:1032 /d |
|
|
|
if [[ "${SKIP_PIPER_VOICES:-}" == "1" ]]; then |
|
echo "[ensure] SKIP_PIPER_VOICES=1 — skipping Piper voice download." |
|
exit 0 |
|
fi |
|
|
|
echo "[ensure] Piper voices (bind mount .local-piper-data) …" |
|
download_piper_voices_to "$ROOT/.local-piper-data" |
|
|
|
if docker volume inspect "$PIPER_VOL" &>/dev/null; then |
|
echo "[ensure] Copying Piper voices into Docker volume ${PIPER_VOL} …" |
|
docker run --rm \ |
|
-v "$PIPER_VOL:/data" \ |
|
-v "$ROOT/.local-piper-data:/src:ro" \ |
|
alpine:3.20 \ |
|
sh -c 'set -e; for f in /src/*.onnx /src/*.onnx.json; do [ -f "$f" ] || continue; bn=$(basename "$f"); cp -a "$f" "/data/$bn"; done; ls -la /data | head -20' |
|
else |
|
echo "[ensure] No Docker volume ${PIPER_VOL} (prod Wyoming uses it). Skipping volume copy — dev-only .local-piper-data is ready." |
|
fi |
|
|
|
echo "[ensure] Done."
|
|
|