From ec7ec370ef3dd08454c6e41364a3dbe7be4ed797 Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Mon, 21 Apr 2025 21:30:36 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=AD=E5=AE=8C=E6=A8=A1=E8=87=AA=E5=8A=A8?=
 =?UTF-8?q?=E5=88=B7=E6=96=B0=E6=A8=A1=E5=9E=8B=E5=88=97=E8=A1=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Automatically refresh the model list after training the model.
---
 webui.py | 86 ++++++++++++++++++++++++++++----------------------------
 1 file changed, 43 insertions(+), 43 deletions(-)

diff --git a/webui.py b/webui.py
index e229b21..ef4800c 100644
--- a/webui.py
+++ b/webui.py
@@ -197,7 +197,7 @@ def set_default():
     else:
         default_sovits_epoch = 2
         default_sovits_save_every_epoch = 1
-        max_sovits_epoch = 50  # 40 # 3
+        max_sovits_epoch = 20  # 40 # 3
         max_sovits_save_every_epoch = 10  # 10 # 3
 
     default_batch_size = max(1, default_batch_size)
@@ -589,7 +589,6 @@ def close_denoise():
 p_train_SoVITS = None
 process_name_sovits = i18n("SoVITS训练")
 
-
 def open1Ba(
     batch_size,
     total_epoch,
@@ -642,22 +641,23 @@ def open1Ba(
         yield (
             process_info(process_name_sovits, "opened"),
             {"__type__": "update", "visible": False},
-            {"__type__": "update", "visible": True},
+            {"__type__": "update", "visible": True},{"__type__": "update"},{"__type__": "update"}
         )
         print(cmd)
         p_train_SoVITS = Popen(cmd, shell=True)
         p_train_SoVITS.wait()
         p_train_SoVITS = None
+        SoVITS_dropdown_update, GPT_dropdown_update = change_choices()
         yield (
             process_info(process_name_sovits, "finish"),
             {"__type__": "update", "visible": True},
-            {"__type__": "update", "visible": False},
+            {"__type__": "update", "visible": False},SoVITS_dropdown_update,GPT_dropdown_update
         )
     else:
         yield (
             process_info(process_name_sovits, "occupy"),
             {"__type__": "update", "visible": False},
-            {"__type__": "update", "visible": True},
+            {"__type__": "update", "visible": True},{"__type__": "update"},{"__type__": "update"}
         )
 
 
@@ -726,22 +726,23 @@ def open1Bb(
         yield (
             process_info(process_name_gpt, "opened"),
             {"__type__": "update", "visible": False},
-            {"__type__": "update", "visible": True},
+            {"__type__": "update", "visible": True},{"__type__": "update"},{"__type__": "update"}
         )
         print(cmd)
         p_train_GPT = Popen(cmd, shell=True)
         p_train_GPT.wait()
         p_train_GPT = None
+        SoVITS_dropdown_update, GPT_dropdown_update = change_choices()
         yield (
             process_info(process_name_gpt, "finish"),
             {"__type__": "update", "visible": True},
-            {"__type__": "update", "visible": False},
+            {"__type__": "update", "visible": False},SoVITS_dropdown_update,GPT_dropdown_update
         )
     else:
         yield (
             process_info(process_name_gpt, "occupy"),
             {"__type__": "update", "visible": False},
-            {"__type__": "update", "visible": True},
+            {"__type__": "update", "visible": True},{"__type__": "update"},{"__type__": "update"}
         )
 
 
@@ -1719,7 +1720,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
                                 label=i18n("文本模块学习率权重"),
                                 value=0.4,
                                 visible=True if version not in v3v4set else False,
-                            )  # v3 not need
+                            )  # v3v4 not need
                             lora_rank = gr.Radio(
                                 label=i18n("LoRA秩"),
                                 value="32",
@@ -1833,40 +1834,7 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
                     with gr.Row():
                         info1Bb = gr.Textbox(label=process_info(process_name_gpt, "info"))
 
-            button1Ba_open.click(
-                open1Ba,
-                [
-                    batch_size,
-                    total_epoch,
-                    exp_name,
-                    text_low_lr_rate,
-                    if_save_latest,
-                    if_save_every_weights,
-                    save_every_epoch,
-                    gpu_numbers1Ba,
-                    pretrained_s2G,
-                    pretrained_s2D,
-                    if_grad_ckpt,
-                    lora_rank,
-                ],
-                [info1Ba, button1Ba_open, button1Ba_close],
-            )
             button1Ba_close.click(close1Ba, [], [info1Ba, button1Ba_open, button1Ba_close])
-            button1Bb_open.click(
-                open1Bb,
-                [
-                    batch_size1Bb,
-                    total_epoch1Bb,
-                    exp_name,
-                    if_dpo,
-                    if_save_latest1Bb,
-                    if_save_every_weights1Bb,
-                    save_every_epoch1Bb,
-                    gpu_numbers1Bb,
-                    pretrained_s1,
-                ],
-                [info1Bb, button1Bb_open, button1Bb_close],
-            )
             button1Bb_close.click(close1Bb, [], [info1Bb, button1Bb_open, button1Bb_close])
 
             with gr.TabItem("1C-" + i18n("推理")):
@@ -1931,7 +1899,39 @@ with gr.Blocks(title="GPT-SoVITS WebUI") as app:
                         ],
                         [tts_info, open_tts, close_tts],
                     )
-
+            button1Ba_open.click(
+                open1Ba,
+                [
+                    batch_size,
+                    total_epoch,
+                    exp_name,
+                    text_low_lr_rate,
+                    if_save_latest,
+                    if_save_every_weights,
+                    save_every_epoch,
+                    gpu_numbers1Ba,
+                    pretrained_s2G,
+                    pretrained_s2D,
+                    if_grad_ckpt,
+                    lora_rank,
+                ],
+                [info1Ba, button1Ba_open, button1Ba_close,SoVITS_dropdown,GPT_dropdown],
+            )
+            button1Bb_open.click(
+                open1Bb,
+                [
+                    batch_size1Bb,
+                    total_epoch1Bb,
+                    exp_name,
+                    if_dpo,
+                    if_save_latest1Bb,
+                    if_save_every_weights1Bb,
+                    save_every_epoch1Bb,
+                    gpu_numbers1Bb,
+                    pretrained_s1,
+                ],
+                [info1Bb, button1Bb_open, button1Bb_close,SoVITS_dropdown,GPT_dropdown],
+            )
             version_checkbox.change(
                 switch_version,
                 [version_checkbox],