From 87add19d0b84c2fb44742685c5f51979bfbc2e00 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9E=97=E5=8C=971941783147?= <11103332+linbei-1941783147@user.noreply.gitee.com> Date: Fri, 1 Mar 2024 20:05:53 +0800 Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E5=86=B2=E7=AA=81=EF=BC=8C?= =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E9=9F=B3=E9=A2=91=E5=90=88=E5=B9=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- GPT_SoVITS/inference_webui.py | 91 +++++++++++++++++++++++++++++++++++ 1 file changed, 91 insertions(+) diff --git a/GPT_SoVITS/inference_webui.py b/GPT_SoVITS/inference_webui.py index 2693647..a4e49a7 100644 --- a/GPT_SoVITS/inference_webui.py +++ b/GPT_SoVITS/inference_webui.py @@ -541,6 +541,73 @@ def get_weights_names(): if name.endswith(".ckpt"): GPT_names.append("%s/%s" % (GPT_weight_root, name)) return SoVITS_names, GPT_names +#region 输出音频历史记录相关 +output_history =[] +history_max_num = 20 + +def sync_output_history_to_checkbox_audio(): + checkbox_result = [] + audio_result = [] + for item in output_history: + label = item['label'] + if len(label)>15: + label=label[:15]+'...' + checkbox_result.append(gr.update(label=label,value=False)) + audio_result.append(gr.Audio.update(value=item['value'])) + for _ in range(len(audio_result),history_max_num): + checkbox_result.append(gr.Checkbox.update(label="",value=False)) + audio_result.append(gr.Audio.update(value = None)) + return [*checkbox_result,*audio_result] + +def add_to_history(audio,input_text): + if(audio is None or audio[1] is not None): + if len(output_history) == history_max_num: + output_history.pop() + output_history.insert(0,{'value':audio,'label':input_text}) + + return [*sync_output_history_to_checkbox_audio()] + +def clear_history(): + output_history = [] + checkbox_result = [] + audio_result = [] + for _ in range(history_max_num): + checkbox_result.append(gr.Checkbox.update(label="",value=False)) + audio_result.append(gr.Audio.update(value = None)) + return [*checkbox_result,*audio_result] + +def shown_audio_num_change(audio_num): + audio_num = int(audio_num) + audio_result = [] + checkbox_result = [] + for _ in range(audio_num): + audio_result.append(gr.Audio.update(visible=True)) + checkbox_result.append(gr.update(visible=True)) + for _ in range(audio_num,history_max_num): + audio_result.append(gr.Audio.update(visible=False)) + checkbox_result.append(gr.update(visible=False)) + return [*checkbox_result,*audio_result] + +def delete_selected_history(*selected_list): + for i in reversed(range(len(output_history))): + if(selected_list[i]): + output_history.pop(i) + return [*sync_output_history_to_checkbox_audio()] + +def merge_selected_history(*selected_list): + m_list = [] + labels = "" + for i in reversed(range(len(output_history))): + if(selected_list[i]): + labels+=" "+output_history[i]["label"] + m_list.append(output_history[i]["value"][1]) + if(m_list): + combined = np.hstack(m_list) + delete_selected_history(*selected_list) + return add_to_history((32000, combined),labels) + return [*sync_output_history_to_checkbox_audio()] + +#endregion SoVITS_names, GPT_names = get_weights_names() @@ -609,6 +676,30 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app: button5.click(cut5, [text_inp], [text_opt]) gr.Markdown(value=i18n("后续将支持转音素、手工修改音素、语音合成分步执行。")) + history_audio = [] + history_checkbox = [] + with gr.Accordion("生成历史"): + with gr.Row(): + shown_audio_num = gr.Slider(1,20,history_max_num,step=1,interactive=True,label="记录显示数量") + add_history_button = gr.Button("添加当前音频记录",variant="primary") + merge_audio_button = gr.Button("合并选中音频",variant="primary") + delete_select_history_button = gr.Button("删除选择的记录") + clear_history_button = gr.Button("清空记录") + index=0 + while(index