blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-10-07 15:19:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

支持按照任意标点符号分句

Browse Source
This commit is contained in:
Erythrocyte3803 2024-01-29 12:30:55 +09:00
parent 1371898b5a
commit 2437d0d080
1 changed files with 31 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
GPT_SoVITS/inference_webui.py
View File

@ -310,6 +310,8 @@ def get_tts_wav(ref_wav_path, prompt_text, prompt_language, text, text_language,
elif(how_to_cut==i18n("凑50字一切")):text=cut2(text) elif(how_to_cut==i18n("凑50字一切")):text=cut2(text)
elif(how_to_cut==i18n("按中文句号。切")):text=cut3(text) elif(how_to_cut==i18n("按中文句号。切")):text=cut3(text)
elif(how_to_cut==i18n("按英文句号.切")):text=cut4(text) elif(how_to_cut==i18n("按英文句号.切")):text=cut4(text)
elif(how_to_cut==i18n("按中文标点分句切")):text=cut5(text)
elif(how_to_cut==i18n("按英文标点分句切")):text=cut6(text)
text = text.replace("\n\n","\n").replace("\n\n","\n").replace("\n\n","\n") text = text.replace("\n\n","\n").replace("\n\n","\n").replace("\n\n","\n")
if(text[-1]not in splits):text+="。"if text_language!="en"else "." if(text[-1]not in splits):text+="。"if text_language!="en"else "."
texts=text.split("\n") texts=text.split("\n")
@ -438,10 +440,27 @@ def cut2(inp):
def cut3(inp): def cut3(inp):
inp = inp.strip("\n") inp = inp.strip("\n")
return "\n".join(["%s。" % item for item in inp.strip("。").split("。")]) return "\n".join(["%s。" % item for item in inp.strip("。").split("。")])
def cut4(inp): def cut4(inp):
inp = inp.strip("\n") inp = inp.strip("\n")
return "\n".join(["%s." % item for item in inp.strip(".").split(".")]) return "\n".join(["%s." % item for item in inp.strip(".").split(".")])
def cut5(inp):
inp = inp.strip("\n")
punds = r'[、，。？！;：]'
items = re.split(f'({punds})', inp)
items = ["".join(group) for group in zip(items[::2], items[1::2])]
opt = "\n".join(items)
return opt
def cut6(inp):
inp = inp.strip("\n")
punds = r'[,.;?!]'
items = re.split(f'({punds})', inp)
items = ["".join(group) for group in zip(items[::2], items[1::2])]
opt = "\n".join(items)
return opt
def custom_sort_key(s): def custom_sort_key(s):
# 使用正则表达式提取字符串中的数字部分和非数字部分 # 使用正则表达式提取字符串中的数字部分和非数字部分
parts = re.split('(\d+)', s) parts = re.split('(\d+)', s)
@ -497,7 +516,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
) )
how_to_cut = gr.Radio( how_to_cut = gr.Radio(
label=i18n("怎么切"), label=i18n("怎么切"),
choices=[i18n("不切"),i18n("凑四句一切"),i18n("凑50字一切"),i18n("按中文句号。切"),i18n("按英文句号.切"),], choices=[i18n("不切"),i18n("凑四句一切"),i18n("凑50字一切"),i18n("按中文句号。切"),i18n("按英文句号.切"),i18n("按中文标点分句切"),i18n("按英文标点分句切"),],
value=i18n("凑50字一切"), value=i18n("凑50字一切"),
interactive=True, interactive=True,
) )
@ -513,15 +532,22 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
gr.Markdown(value=i18n("文本切分工具。太长的文本合成出来效果不一定好，所以太长建议先切。合成会根据文本的换行分开合成再拼起来。")) gr.Markdown(value=i18n("文本切分工具。太长的文本合成出来效果不一定好，所以太长建议先切。合成会根据文本的换行分开合成再拼起来。"))
with gr.Row(): with gr.Row():
text_inp = gr.Textbox(label=i18n("需要合成的切分前文本"),value="") text_inp = gr.Textbox(label=i18n("需要合成的切分前文本"),value="")
with gr.Row():
button1 = gr.Button(i18n("凑四句一切"), variant="primary") button1 = gr.Button(i18n("凑四句一切"), variant="primary")
button2 = gr.Button(i18n("凑50字一切"), variant="primary") button2 = gr.Button(i18n("凑50字一切"), variant="primary")
with gr.Row():
button3 = gr.Button(i18n("按中文句号。切"), variant="primary") button3 = gr.Button(i18n("按中文句号。切"), variant="primary")
button4 = gr.Button(i18n("按英文句号.切"), variant="primary") button4 = gr.Button(i18n("按英文句号.切"), variant="primary")
with gr.Row():
button5 = gr.Button(i18n("按中文标点分句切"), variant="primary")
button6 = gr.Button(i18n("按英文标点分句切"), variant="primary")
text_opt = gr.Textbox(label=i18n("切分后文本"), value="") text_opt = gr.Textbox(label=i18n("切分后文本"), value="")
button1.click(cut1, [text_inp], [text_opt]) button1.click(cut1, [text_inp], [text_opt])
button2.click(cut2, [text_inp], [text_opt]) button2.click(cut2, [text_inp], [text_opt])
button3.click(cut3, [text_inp], [text_opt]) button3.click(cut3, [text_inp], [text_opt])
button4.click(cut4, [text_inp], [text_opt]) button4.click(cut4, [text_inp], [text_opt])
button5.click(cut5, [text_inp], [text_opt])
button6.click(cut6, [text_inp], [text_opt])
gr.Markdown(value=i18n("后续将支持混合语种编码文本输入。")) gr.Markdown(value=i18n("后续将支持混合语种编码文本输入。"))
app.queue(concurrency_count=511, max_size=1022).launch( app.queue(concurrency_count=511, max_size=1022).launch(