Support German TTS (#394)

k2-fsa · Oct 27, 2023 · fbf4c90 · fbf4c90
1 parent 93ef4ee
commit fbf4c90
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 12 deletions.
diff --git a/android/SherpaOnnxTts/app/src/main/java/com/k2fsa/sherpa/onnx/MainActivity.kt b/android/SherpaOnnxTts/app/src/main/java/com/k2fsa/sherpa/onnx/MainActivity.kt
@@ -77,20 +77,13 @@ class MainActivity : AppCompatActivity() {
             return
         }
 
-        Toast.makeText(applicationContext, "Generating...Please wait", Toast.LENGTH_LONG).show()
+        play.isEnabled = false;
         val audio = tts.generate(text = textStr, sid = sidInt, speed = speedFloat)
 
         val filename = application.filesDir.absolutePath + "/generated.wav"
         val ok = audio.samples.size > 0 && audio.save(filename)
         if (ok) {
             play.isEnabled = true
-            Toast.makeText(
-                applicationContext,
-                "Generated! Please click play to listen to it",
-                Toast.LENGTH_LONG
-            ).show()
-        } else {
-            play.isEnabled = false
         }
     }
 
@@ -110,4 +103,4 @@ class MainActivity : AppCompatActivity() {
         val config = getOfflineTtsConfig(type = type, debug = true)!!
         tts = OfflineTts(assetManager = application.assets, config = config)
     }
-}
+}
diff --git a/sherpa-onnx/csrc/lexicon.cc b/sherpa-onnx/csrc/lexicon.cc
@@ -129,10 +129,12 @@ std::vector<int64_t> Lexicon::ConvertTextToTokenIds(
   switch (language_) {
     case Language::kEnglish:
       return ConvertTextToTokenIdsEnglish(text);
+    case Language::kGerman:
+      return ConvertTextToTokenIdsGerman(text);
     case Language::kChinese:
       return ConvertTextToTokenIdsChinese(text);
     default:
-      SHERPA_ONNX_LOGE("Unknonw language: %d", static_cast<int32_t>(language_));
+      SHERPA_ONNX_LOGE("Unknown language: %d", static_cast<int32_t>(language_));
       exit(-1);
   }
 
@@ -246,6 +248,8 @@ void Lexicon::InitLanguage(const std::string &_lang) {
   ToLowerCase(&lang);
   if (lang == "english") {
     language_ = Language::kEnglish;
+  } else if (lang == "german") {
+    language_ = Language::kGerman;
   } else if (lang == "chinese") {
     language_ = Language::kChinese;
   } else {
@@ -269,8 +273,8 @@ void Lexicon::InitLexicon(std::istream &is) {
     ToLowerCase(&word);
 
     if (word2ids_.count(word)) {
-      SHERPA_ONNX_LOGE("Duplicated word: %s", word.c_str());
-      return;
+      SHERPA_ONNX_LOGE("Duplicated word: %s. Ignore it.", word.c_str());
+      continue;
     }
 
     while (iss >> phone) {

diff --git a/sherpa-onnx/csrc/lexicon.h b/sherpa-onnx/csrc/lexicon.h
@@ -36,6 +36,11 @@ class Lexicon {
   std::vector<int64_t> ConvertTextToTokenIds(const std::string &text) const;
 
  private:
+  std::vector<int64_t> ConvertTextToTokenIdsGerman(
+      const std::string &text) const {
+    return ConvertTextToTokenIdsEnglish(text);
+  }
+
   std::vector<int64_t> ConvertTextToTokenIdsEnglish(
       const std::string &text) const;
 
@@ -50,6 +55,7 @@ class Lexicon {
  private:
   enum class Language {
     kEnglish,
+    kGerman,
     kChinese,
     kUnknown,
   };