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,
- )
+ )