From 5081168918d5d2af2a4b219c29987049da6a7fef Mon Sep 17 00:00:00 2001
From: Downupanddownup <pingdengjia0liu@163.com>
Date: Mon, 29 Apr 2024 13:04:33 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=88=87=E6=8D=A2=E9=87=8D?=
 =?UTF-8?q?=E7=BD=AE=E6=8C=89=E9=92=AE=E4=BA=8B=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../ref_audio_selector_webui.py               | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/Ref_Audio_Selector/ref_audio_selector_webui.py b/Ref_Audio_Selector/ref_audio_selector_webui.py
index bfbb6b7..0e22ace 100644
--- a/Ref_Audio_Selector/ref_audio_selector_webui.py
+++ b/Ref_Audio_Selector/ref_audio_selector_webui.py
@@ -597,6 +597,38 @@ def chang_refer_type_param(selected_value):
                                                                                                     "__type__": "update"}
 
 
+def switch_role_and_refresh():
+    global hide_voice_similarity_dir, hide_config_file
+    init.init_all()
+    reset_list = []
+    # 基础
+    reset_list.extend([
+        init.text_refer_audio_file_dir_default,
+        init.text_inference_audio_file_dir_default
+    ])
+    # 第一步
+    reset_list.extend([
+        init.text_sample_dir_default,
+        '',  # text_list_input
+        '',  # text_base_voice_path
+    ])
+    # 第二步
+    # 第三步
+    reset_list.extend([
+        init.text_asr_audio_dir_default,
+        init.text_text_similarity_analysis_path_default
+    ])
+    # 第四步
+    reset_list.extend([
+        '',  # text_base_audio_path
+        '',  # text_compare_audio_dir
+    ])
+    hide_voice_similarity_dir = ''
+    hide_config_file = ''
+    # 第五步
+    return reset_list
+
+
 def init_ui():
     init.init_all()
 
@@ -985,6 +1017,34 @@ def init_ui():
                                      [text_model_inference_info, text_asr_audio_dir, text_inference_audio_file_dir])
         button_model_inference_result_open.click(open_file, [text_inference_audio_file_dir], [])
 
+        # 设置重置刷新事件
+        refresh_list = []
+        # 基础
+        refresh_list.extend([
+            text_refer_audio_file_dir,
+            text_inference_audio_file_dir
+        ])
+        # 第一步
+        refresh_list.extend([
+            text_sample_dir,
+            text_list_input,
+            text_base_voice_path
+        ])
+        # 第二步
+        # 第三步
+        refresh_list.extend([
+            text_asr_audio_dir,
+            text_text_similarity_analysis_path
+        ])
+        # 第四步
+        refresh_list.extend([
+            text_base_audio_path,
+            text_compare_audio_dir
+        ])
+        # 第五步
+
+        button_switch_role_and_refresh.click(switch_role_and_refresh, [], refresh_list)
+
     app.launch(
         server_port=params.server_port,
         inbrowser=True,