Update split_sentence

gradio_app.py

@@ -235,8 +235,7 @@ def inference(input_text, language, speaker_id=None, gpt_cond_latent=None, speak
             sentence = normalize_vietnamese_text(sentence)
         text_tokens = torch.IntTensor(xtts_model.tokenizer.encode(sentence, lang=lang)).unsqueeze(0).to(xtts_model.device)
         num_of_tokens += text_tokens.shape[-1]
-        txts = [sentence]
-        if len(sentence) >= max_text_length: txts = split_sentence(sentence)
+        txts = split_sentence(sentence, max_text_length=max_text_length)
         for txt in txts:
             logger.info(f"[{lang}] {txt}")
             try:
@@ -250,11 +249,11 @@ def inference(input_text, language, speaker_id=None, gpt_cond_latent=None, speak
                     top_k=top_k,
                     repetition_penalty=repetition_penalty,
                     length_penalty=dynamic_length_penalty(len(sentence)),
-                    enable_text_splitting=True,
+                    enable_text_splitting=False,
                 )
                 out_wavs.append(out["wav"])
             except Exception as e:
-                logger.error(f"Error processing text: {txt} - {e}")            
+                logger.error(f"Error processing text: {e}")
     return np.concatenate(out_wavs), num_of_tokens
 
 

utils/sentence.py

@@ -1,5 +1,5 @@
 
-def split_sentence(sentence, delimiters=",;-!?"):
+def split_sentence(sentence, max_text_length=180, delimiters=",;-!?"):
     """
     Splits a sentence into two halves, prioritizing the delimiter closest to the middle.
     If no delimiter is found, it ensures words are not split in the middle.
@@ -11,6 +11,9 @@ def split_sentence(sentence, delimiters=",;-!?"):
     Returns:
         tuple: A tuple containing the two halves of the sentence.
     """
+    if len(sentence) < max_text_length:
+        return [sentence]
+
     # Find all delimiter indices in the sentence
     delimiter_indices = [i for i, char in enumerate(sentence) if char in delimiters]
 
@@ -50,7 +53,8 @@ def split_sentence(sentence, delimiters=",;-!?"):
         first_half = sentence[:split_index].strip()
         second_half = sentence[split_index:].strip()
 
-    return first_half, second_half
+    return split_sentence(first_half, max_text_length=max_text_length) \
+        + split_sentence(second_half, max_text_length=max_text_length)
 
 
 def merge_sentences(sentences):