Update app.py

app.py

@@ -902,6 +902,19 @@ def download_model_files(model):
     else:
         return None, f'Could not find {model}.pth or corresponding Index file.'
 
+def update_visibility(visible):
+    if visible:
+        return {"visible":True,"__type__":"update"},{"visible":True,"__type__":"update"}
+    else:
+        return {"visible":False,"__type__":"update"},{"visible":False,"__type__":"update"}
+
+def get_pretrains(string):
+    pretrains = []
+    for file in os.listdir('assets/pretrained_v2'):
+        if string in file:
+            pretrains.append(os.path.join('assets/pretrained_v2',file))
+    return pretrains
+
 with gr.Blocks(title="🔊",theme=gr.themes.Base(primary_hue="rose",neutral_hue="zinc")) as app:
     with gr.Row():
         gr.HTML("<img  src='file/a.png' alt='image'>")
@@ -1347,18 +1360,25 @@ with gr.Blocks(title="🔊",theme=gr.themes.Base(primary_hue="rose",neutral_hue=
                             value=i18n("是"),
                             interactive=True,
                         )
-                        pretrained_G14 = gr.Textbox(
+                        display_advanced_settings = gr.Checkbox(
+                            value=False,
+                            label="Advanced Settings"
+                        )
+                        pretrained_G14 = gr.Dropdown(
                             label=i18n("加载预训练底模G路径"),
                             value="assets/pretrained_v2/f0G40k.pth",
+                            choices=get_pretrains('40k'),
                             interactive=True,
-                            visible=True
+                            visible=False
                         )
-                        pretrained_D15 = gr.Textbox(
+                        pretrained_D15 = gr.Dropdown(
                             label=i18n("加载预训练底模D路径"),
                             value="assets/pretrained_v2/f0D40k.pth",
                             interactive=True,
-                            visible=True
+                            choices=get_pretrains('40k'),
+                            visible=False
                         )
+                        display_advanced_settings.change(fn=update_visibility,inputs=[display_advanced_settings],outputs=[pretrained_G14,pretrained_D15])
                     with gr.Row():
                         download_model = gr.Button('5.Download Model')
                     with gr.Row():
@@ -1438,4 +1458,4 @@ with gr.Blocks(title="🔊",theme=gr.themes.Base(primary_hue="rose",neutral_hue=
             inbrowser=not config.noautoopen,
             server_port=config.listen_port,
             quiet=True,
-        )
+        )