feat: mozhi tts support (closes #375)

you-apps · Dec 6, 2023 · c8c2a54 · c8c2a54
1 parent 17e9dbd
commit c8c2a54
Show file tree

Hide file tree

Showing 6 changed files with 30 additions and 20 deletions.
diff --git a/.idea/deploymentTargetDropDown.xml b/.idea/deploymentTargetDropDown.xml
diff --git a/.idea/gradle.xml b/.idea/gradle.xml
diff --git a/app/src/main/java/com/bnyro/translate/api/lv/LVEngine.kt b/app/src/main/java/com/bnyro/translate/api/lv/LVEngine.kt
@@ -75,10 +75,9 @@ class LVEngine : TranslationEngine(
     override suspend fun getAudioFile(lang: String, query: String): File? {
         val byteArray = api.getAudio(lang, query).toByteArray()
         if (byteArray.isEmpty()) return null
-        return withContext(Dispatchers.IO) {
-            File.createTempFile("audio", ".mp3").apply {
-                writeBytes(byteArray)
-            }
+
+        return File.createTempFile("audio", ".mp3").apply {
+            writeBytes(byteArray)
         }
     }
 }
diff --git a/app/src/main/java/com/bnyro/translate/api/mh/MhEngine.kt b/app/src/main/java/com/bnyro/translate/api/mh/MhEngine.kt
@@ -22,6 +22,7 @@ import com.bnyro.translate.db.obj.Language
 import com.bnyro.translate.obj.Translation
 import com.bnyro.translate.util.RetrofitHelper
 import com.bnyro.translate.util.TranslationEngine
+import java.io.File
 
 class MhEngine : TranslationEngine(
     name = "Mozhi",
@@ -38,7 +39,8 @@ class MhEngine : TranslationEngine(
         "mymemory",
         "watson",
         "yandex"
-    )
+    ),
+    supportsAudio = true
 ) {
     lateinit var api: Mozhi
 
@@ -64,4 +66,12 @@ class MhEngine : TranslationEngine(
             detectedLanguage = response.sourceLanguage
         )
     }
+
+    override suspend fun getAudioFile(lang: String, query: String): File? {
+        val audioBytes = api.getAudioFile(lang = lang, text = query).body()?.bytes() ?: return null
+
+        return File.createTempFile("audio", ".mp3").apply {
+            writeBytes(audioBytes)
+        }
+    }
 }
diff --git a/app/src/main/java/com/bnyro/translate/api/mh/Mozhi.kt b/app/src/main/java/com/bnyro/translate/api/mh/Mozhi.kt
@@ -19,6 +19,8 @@ package com.bnyro.translate.api.mh
 
 import com.bnyro.translate.api.mh.obj.MhLanguage
 import com.bnyro.translate.api.st.obj.STTranslationResponse
+import okhttp3.ResponseBody
+import retrofit2.Response
 import retrofit2.http.GET
 import retrofit2.http.Query
 
@@ -35,4 +37,11 @@ interface Mozhi {
     suspend fun getLanguages(
         @Query("engine") engine: String?
     ): List<MhLanguage>
+
+    @GET("api/tts")
+    suspend fun getAudioFile(
+        @Query("engine") engine: String? = "google",
+        @Query("lang") lang: String,
+        @Query("text") text: String
+    ): Response<ResponseBody>
 }
diff --git a/app/src/main/java/com/bnyro/translate/ui/models/TranslationModel.kt b/app/src/main/java/com/bnyro/translate/ui/models/TranslationModel.kt
@@ -264,6 +264,7 @@ class TranslationModel : ViewModel() {
 
     fun playAudio() {
         releaseMediaPlayer()
+
         viewModelScope.launch(Dispatchers.IO) {
             audioFile = runCatching {
                 engine.getAudioFile(targetLanguage.code, translation.translatedText)