Add SSML removal to async client as well

playht · Jul 29, 2024 · 2935d38 · 2935d38
1 parent 78c6188
commit 2935d38
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/pyht/async_client.py b/pyht/async_client.py
@@ -223,9 +223,10 @@ async def _tts(
             lease_data = self._lease.data
 
         if isinstance(text, str):
-            text = split_text(normalize(text))
-        else:
-            text = [normalize(x) for x in text]
+            text = split_text(text)
+        if self._advanced.remove_ssml_tags:
+            text = [re.sub(r'<[^>]*>', '', x) for x in text]
+        text = [normalize(x) for x in text]
         text = ensure_sentence_end(text)
 
         request = api_pb2.TtsRequest(params=options.tts_params(text, voice_engine), lease=lease_data)

diff --git a/pyht/client.py b/pyht/client.py
@@ -281,12 +281,11 @@ def _tts(
             lease_data = self._lease.data
 
         if isinstance(text, str):
-            text = split_text(normalize(text))
-        else:
-            text = [normalize(x) for x in text]
-        text = ensure_sentence_end(text)
+            text = split_text(text)
         if self._advanced.remove_ssml_tags:
-            text = [normalize(re.sub(r'<[^>]*>', '', x)) for x in text]
+            text = [re.sub(r'<[^>]*>', '', x) for x in text]
+        text = [normalize(x) for x in text]
+        text = ensure_sentence_end(text)
 
         request = api_pb2.TtsRequest(params=options.tts_params(text, voice_engine), lease=lease_data)