Update app.py

app.py

@@ -27,7 +27,7 @@ from infer_tools.infer_tool import *
 import io
 
 clip_completed = False
-def render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2):
+def render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2, Crepe, noise_step, use_mel_as_base):
     logging.getLogger('numba').setLevel(logging.WARNING)
     title = int(title)
     title2 = int(title2)
@@ -47,8 +47,8 @@ def render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title
     
     # Show the spinner and run the run_clip function inside the 'with' block
     with st.spinner("Rendering Audio..."):
-      f0_tst, f0_pred, audio = run_clip(svc_model,file_path=wav_fn, key=key, acc=title2, use_crepe=True, use_pe=True, thre=0.05,
-                                        use_gt_mel=False, add_noise_step=500,project_name=project_name,out_path=wav_gen)
+      f0_tst, f0_pred, audio = run_clip(svc_model,file_path=wav_fn, key=key, acc=title2, use_crepe=Crepe, use_pe=False, thre=0.05,
+                                        use_gt_mel=use_mel_as_base, add_noise_step=noise_step,project_name=project_name,out_path=wav_gen)
     clip_completed = True
     if clip_completed:
     
@@ -61,11 +61,10 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 ############
-st.title('DIFF-SVC Render')
-
-###CKPT LOADER
-ckpt = st.file_uploader("Choose your CKPT", type= 'ckpt')
-# Check if user uploaded a CKPT file
+col1, col2 = st.columns(2)
+col1.title('DIFF-SVC Render')
+col2.title('Settings')
+ckpt = col1.file_uploader("Choose your CKPT", type='ckpt')
 if ckpt is not None:
   #TEMP FUNCTION
   with tempfile.NamedTemporaryFile(mode="wb", suffix='.ckpt', delete=False) as temp:
@@ -80,8 +79,7 @@ if ckpt is not None:
 if "ckpt_temp_file" in locals():
     st.success("File saved to: {}".format(ckpt_temp_file))
 # File uploader
-config = st.file_uploader("Choose your config", type= 'yaml')
-# Check if user uploaded a config file
+config = col1.file_uploader("Choose your config", type='yaml')
 if config is not None:
   #TEMP FUNCTION
   with tempfile.NamedTemporaryFile(mode="wb", suffix='.yaml', delete=False) as temp:
@@ -95,8 +93,8 @@ if config is not None:
 # Display the file path
 if "config_temp_file" in locals():
     st.success("File saved to: {}".format(config_temp_file))
-audio = st.file_uploader("Choose your audio", type=["wav"])
-# Check if user uploaded an audio file
+
+audio = col1.file_uploader("Choose your audio", type=["wav"])
 if audio is not None:
 #EMP FUNCTION
   with tempfile.NamedTemporaryFile(mode="wb", suffix='.wav', delete=False) as temp:
@@ -110,16 +108,30 @@ if audio is not None:
 # Display the file path
 if "audio_temp_file" in locals():
     st.success("File saved to: {}".format(audio_temp_file))
-# Add a text input for the title with a default value of 0
-title = st.text_input("Key", value="0")
-title2 = st.text_input("Speedup", value="20")
-password = st.text_input("Enter password")
+
+title = col2.number_input("Key", value=0, step=1, min_value=-12, max_value=12)
+title2 = col2.number_input("Speedup", value=20, step=1, min_value=5, max_value=100)
+title3 = col2.number_input("Gender Flag", value=1.00, step=0.01, min_value=0.70, max_value=1.30, help='Default is 1.0, it works by decimals, setting it at 1.05 will make your render sound more female-ish, setting it to 0.95 will make it sound more masculine, for example.')
+choice = col2.selectbox('Resampler', ('Crepe', 'Harvest'))
+# Create checkbox for using Mel as Base
+use_mel_as_base = col2.checkbox('Use Mel as Base', value=False, help='gt mel: Enabling this will use the input audio as a base and will unlock a new parameter, do not use this if you dont know what it does.')
+noise_step = 600
+# Show "Noise Step" input parameter when checkbox is checked
+if use_mel_as_base:
+    noise_step = col2.number_input('Noise Step', value=600, min_value=1, max_value=1000, step=50)
+else:
+    noise_step = 600
+password = col2.text_input("Enter password", help='Password can be got by agreeing to TOS and getting allowed after validation, you can go to the TOS here:')
 correct_password = os.environ.get("gatepassword")
-# Add a button to start the rendering process
-# Add a button to start the rendering process
+if choice == Crepe
+    Crepe = True
+else:
+    Crepe = False
+
+###
 if st.button("Render audio"):
     if password == correct_password:
-        render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2)
+        render_audio(ckpt_temp_file, config_temp_file, audio_temp_file, title, title2, noise_step, Crepe, use_mel_as_base)
   
     else:
         st.error("Incorrect password")