Add XTTS primary voice with Piper fallback

2026-08-23 12:22:59 +02:00
parent 16a6288a4d
commit f90fc93f9c
18 changed files with 715 additions and 28 deletions
@@ -0,0 +1,59 @@
+import importlib.util
+import pathlib
+import unittest
+
+
+MODULE_PATH = pathlib.Path(__file__).with_name("tts_gateway.py")
+SPEC = importlib.util.spec_from_file_location("tts_gateway", MODULE_PATH)
+gateway = importlib.util.module_from_spec(SPEC)
+SPEC.loader.exec_module(gateway)
+
+
+class LanguageSegmentationTests(unittest.TestCase):
+    def test_german_only(self):
+        self.assertEqual(
+            gateway.segment_languages("Guten Abend, wie warm ist es heute?"),
+            [("de", "Guten Abend, wie warm ist es heute?")],
+        )
+
+    def test_english_only(self):
+        text = "This is a short test and it is running on the local server."
+        self.assertEqual(gateway.segment_languages(text), [("en", text)])
+
+    def test_mixed_compounds(self):
+        text = "Ich öffne das Unraid-Dashboard und prüfe die Docker-Container."
+        self.assertEqual(
+            gateway.segment_languages(text),
+            [
+                ("de", "Ich öffne das "),
+                ("en", "Unraid-Dashboard"),
+                ("de", " und prüfe die "),
+                ("en", "Docker-Container"),
+                ("de", "."),
+            ],
+        )
+
+
+class FallbackTests(unittest.TestCase):
+    def setUp(self):
+        self.original_xtts = gateway.synthesize_xtts
+        self.original_piper = gateway.synthesize_piper
+
+    def tearDown(self):
+        gateway.synthesize_xtts = self.original_xtts
+        gateway.synthesize_piper = self.original_piper
+
+    def test_piper_is_used_when_xtts_fails(self):
+        def fail(*_args):
+            raise RuntimeError("synthetic XTTS failure")
+
+        gateway.synthesize_xtts = fail
+        gateway.synthesize_piper = lambda *_args: (b"piper", "audio/wav")
+        self.assertEqual(
+            gateway.synthesize("synthetic test", "wav", 1.0),
+            (b"piper", "audio/wav"),
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()