From 55f82e9ad17f98f3b25803f426e5d423e0245d46 Mon Sep 17 00:00:00 2001
From: KamioRinn <snowsdream@live.com>
Date: Mon, 19 Feb 2024 02:11:02 +0800
Subject: [PATCH] Fix text formatting

---
 GPT_SoVITS/inference_webui.py | 9 ++++++---
 GPT_SoVITS/text/chinese.py    | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/GPT_SoVITS/inference_webui.py b/GPT_SoVITS/inference_webui.py
index 39ae7e4..407437f 100644
--- a/GPT_SoVITS/inference_webui.py
+++ b/GPT_SoVITS/inference_webui.py
@@ -562,10 +562,13 @@ def cut5(inp):
     # if not re.search(r'[^\w\s]', inp[-1]):
     # inp += '。'
     inp = inp.strip("\n")
-    punds = r'[,.;?!、，。？！;：]'
+    punds = r'[,.;?!、，。？！;：…]'
     items = re.split(f'({punds})', inp)
-    items = ["".join(group) for group in zip(items[::2], items[1::2])]
-    opt = "\n".join(items)
+    mergeitems = ["".join(group) for group in zip(items[::2], items[1::2])]
+    # 在句子不存在符号或句尾无符号的时候保证文本完整
+    if len(items)%2 == 1:
+        mergeitems.append(items[-1])
+    opt = "\n".join(mergeitems)
     return opt
 
 
diff --git a/GPT_SoVITS/text/chinese.py b/GPT_SoVITS/text/chinese.py
index ea41db1..5334326 100644
--- a/GPT_SoVITS/text/chinese.py
+++ b/GPT_SoVITS/text/chinese.py
@@ -30,7 +30,7 @@ rep_map = {
     "\n": ".",
     "·": ",",
     "、": ",",
-    "...": "…",
+    # "...": "…",
     "$": ".",
     "/": ",",
     "—": "-",