Build text-to-speech engine APKs (#509)

scripts/apk/build-apk-tts-engine.sh.in

@@ -0,0 +1,98 @@
+#!/usr/bin/env bash
+#
+# Auto generated! Please DO NOT EDIT!
+
+# Please set the environment variable ANDROID_NDK
+# before running this script
+
+# Inside the $ANDROID_NDK directory, you can find a binary ndk-build
+# and some other files like the file "build/cmake/android.toolchain.cmake"
+
+set -ex
+
+log() {
+  # This function is from espnet
+  local fname=${BASH_SOURCE[1]##*/}
+  echo -e "$(date '+%Y-%m-%d %H:%M:%S') (${fname}:${BASH_LINENO[0]}:${FUNCNAME[1]}) $*"
+}
+
+SHERPA_ONNX_VERSION=$(grep "SHERPA_ONNX_VERSION" ./CMakeLists.txt  | cut -d " " -f 2  | cut -d '"' -f 2)
+
+log "Building TTS engine APK for sherpa-onnx v${SHERPA_ONNX_VERSION}"
+
+log "====================arm64-v8a================="
+./build-android-arm64-v8a.sh
+log "====================armv7-eabi================"
+./build-android-armv7-eabi.sh
+log "====================x86-64===================="
+./build-android-x86-64.sh
+log "====================x86===================="
+./build-android-x86.sh
+
+mkdir -p apks
+
+{% for tts_model in tts_model_list %}
+pushd ./android/SherpaOnnxTtsEngine/app/src/main/assets/
+model_dir={{ tts_model.model_dir }}
+model_name={{ tts_model.model_name }}
+lang={{ tts_model.lang }}
+lang_iso_639_3={{ tts_model.lang_iso_639_3 }}
+
+wget -qq https://github.com/k2-fsa/sherpa-onnx/releases/download/tts-models/$model_dir.tar.bz2
+tar xf $model_dir.tar.bz2
+rm $model_dir.tar.bz2
+
+popd
+# Now we are at the project root directory
+
+git checkout .
+pushd android/SherpaOnnxTtsEngine/app/src/main/java/com/k2fsa/sherpa/onnx/tts/engine
+sed -i.bak s/"modelDir = null"/"modelDir = \"$model_dir\""/ ./TtsEngine.kt
+sed -i.bak s/"modelName = null"/"modelName = \"$model_name\""/ ./TtsEngine.kt
+sed -i.bak s/"lang = null"/"lang = \"$lang_iso_639_3\""/ ./TtsEngine.kt
+
+{% if tts_model.rule_fsts %}
+  rule_fsts={{ tts_model.rule_fsts }}
+  sed -i.bak s%"ruleFsts = null"%"ruleFsts = \"$rule_fsts\""% ./TtsEngine.kt
+{% endif %}
+
+{% if tts_model.data_dir %}
+  data_dir={{ tts_model.data_dir }}
+  sed -i.bak s%"dataDir = null"%"dataDir = \"$data_dir\""% ./TtsEngine.kt
+{% else %}
+  sed -i.bak s/"lexicon = null"/"lexicon = \"lexicon.txt\""/ ./TtsEngine.kt
+{% endif %}
+
+git diff
+popd
+
+for arch in arm64-v8a armeabi-v7a x86_64 x86; do
+  log "------------------------------------------------------------"
+  log "build tts apk for $arch"
+  log "------------------------------------------------------------"
+  src_arch=$arch
+  if [ $arch == "armeabi-v7a" ]; then
+    src_arch=armv7-eabi
+  elif [ $arch == "x86_64" ]; then
+    src_arch=x86-64
+  fi
+
+  ls -lh ./build-android-$src_arch/install/lib/*.so
+
+  cp -v ./build-android-$src_arch/install/lib/*.so ./android/SherpaOnnxTtsEngine/app/src/main/jniLibs/$arch/
+
+  pushd ./android/SherpaOnnxTtsEngine
+  ./gradlew build
+  popd
+
+  mv android/SherpaOnnxTtsEngine/app/build/outputs/apk/debug/app-debug.apk ./apks/sherpa-onnx-${SHERPA_ONNX_VERSION}-$arch-$lang-tts-engine-$model_dir.apk
+  ls -lh apks
+  rm -v ./android/SherpaOnnxTtsEngine/app/src/main/jniLibs/$arch/*.so
+done
+
+rm -rf ./android/SherpaOnnxTtsEngine/app/src/main/assets/$model_dir
+{% endfor %}
+
+git checkout .
+
+ls -lh apks/

scripts/apk/build-apk-tts.sh.in

@@ -18,7 +18,7 @@ log() {
 
 SHERPA_ONNX_VERSION=$(grep "SHERPA_ONNX_VERSION" ./CMakeLists.txt  | cut -d " " -f 2  | cut -d '"' -f 2)
 
-log "Building APK for sherpa-onnx v${SHERPA_ONNX_VERSION}"
+log "Building TTS APK for sherpa-onnx v${SHERPA_ONNX_VERSION}"
 
 log "====================arm64-v8a================="
 ./build-android-arm64-v8a.sh

scripts/apk/generate-tts-apk-script.py

@@ -6,6 +6,9 @@ from typing import List, Optional
 
 import jinja2
 
+# pip install iso639-lang
+from iso639 import Lang
+
 
 def get_args():
     parser = argparse.ArgumentParser()
@@ -31,6 +34,12 @@ class TtsModel:
     lang: str = ""  # en, zh, fr, de, etc.
     rule_fsts: Optional[List[str]] = None
     data_dir: Optional[str] = None
+    lang_iso_639_3: str = ""
+
+
+def convert_lang_to_iso_639_3(models: List[TtsModel]):
+    for m in models:
+        m.lang_iso_639_3 = Lang(m.lang).pt3
 
 
 def get_coqui_models() -> List[TtsModel]:
@@ -234,15 +243,12 @@ def main():
     index = args.index
     total = args.total
     assert 0 <= index < total, (index, total)
-    environment = jinja2.Environment()
-    with open("./build-apk-tts.sh.in") as f:
-        s = f.read()
-    template = environment.from_string(s)
     d = dict()
 
     all_model_list = get_vits_models()
     all_model_list += get_piper_models()
     all_model_list += get_coqui_models()
+    convert_lang_to_iso_639_3(all_model_list)
 
     num_models = len(all_model_list)
 
@@ -262,9 +268,16 @@ def main():
         d["tts_model_list"].append(all_model_list[s])
         print(f"{s}/{num_models}")
 
-    s = template.render(**d)
-    with open("./build-apk-tts.sh", "w") as f:
-        print(s, file=f)
+    filename_list = ["./build-apk-tts.sh", "./build-apk-tts-engine.sh"]
+    for filename in filename_list:
+        environment = jinja2.Environment()
+        with open(f"{filename}.in") as f:
+            s = f.read()
+        template = environment.from_string(s)
+
+        s = template.render(**d)
+        with open(filename, "w") as f:
+            print(s, file=f)
 
 
 if __name__ == "__main__":