Ensure input for speaker ID is a valid number. (#552)

Fix #547
2024-01-26 20:42:10 +08:00
parent 3f2a17ef47
commit 4fbad6a368
1 changed files with 7 additions and 1 deletions
--- a/android/SherpaOnnxTtsEngine/app/src/main/java/com/k2fsa/sherpa/onnx/tts/engine/MainActivity.kt
+++ b/android/SherpaOnnxTtsEngine/app/src/main/java/com/k2fsa/sherpa/onnx/tts/engine/MainActivity.kt
@@ -38,6 +38,7 @@ import androidx.compose.ui.tooling.preview.Preview
 import androidx.compose.ui.unit.dp
 import com.k2fsa.sherpa.onnx.tts.engine.ui.theme.SherpaOnnxTtsEngineTheme
 import java.io.File
+import java.lang.NumberFormatException

 const val TAG = "sherpa-onnx-tts-engine"

@@ -77,7 +78,12 @@ class MainActivity : ComponentActivity() {
                                            if (it.isEmpty() || it.isBlank()) {
                                                TtsEngine.speakerId = 0
                                            } else {
-                                                TtsEngine.speakerId = it.toString().toInt()
+                                                try {
+                                                    TtsEngine.speakerId = it.toString().toInt()
+                                                } catch (ex: NumberFormatException) {
+                                                    Log.i(TAG, "Invalid input: ${it}")
+                                                    TtsEngine.speakerId = 0
+                                                }
                                            }
                                        },
                                        label = {