Update app.py

app.py

@@ -237,6 +237,50 @@ def mvsep_download_separated_audio(hash_textbox):
         return json.dumps(urls, indent=4)
     except requests.exceptions.JSONDecodeError:
         return gr.Info("Nothing to download yet. Check back later.")
+    
+def mvsep_list_available_models():
+    return [
+        "0 - spleeter (vocals, music)",
+        "1 - spleeter (vocals, drums, bass, other)",
+        "2 - spleeter (vocals, drums, bass, piano, other)",
+        "3 - unmix XL (vocals, drums, bass, other)",
+        "4 - unmix HQ (vocals, drums, bass, other)",
+        "5 - unmix SD (vocals, drums, bass, other)",
+        "6 - unmix SE (vocals, music)",
+        "7 - MDX A (vocals, drums, bass, other)",
+        "8 - MDX B (vocals, drums, bass, other)",
+        "9 - UVR HQ (vocals, music)",
+        "10 - Demucs3 Model A (vocals, drums, bass, other)",
+        "11 - Demucs3 Model B (vocals, drums, bass, other)",
+        "12 - MDX-B Karaoke (lead/back vocals)",
+        "13 - Demucs2 (vocals, drums, bass, other)",
+        "14 - Zero Shot (Query Based) (LQ)",
+        "15 - Danna sep (vocals, drums, bass, other)",
+        "16 - Byte Dance (vocals, drums, bass, other)",
+        "17 - UVRv5 Demucs (vocals, music)",
+        "18 - MVSep DNR (music, sfx, speech)",
+        "19 - MVSep Vocal Model (vocals, music)",
+        "20 - Demucs4 HT (vocals, drums, bass, other)",
+        "22 - FoxJoy Reverb Removal (other)",
+        "23 - MDX B (vocals, instrumental)",
+        "24 - MVSep Demucs4HT DNR (dialog, sfx, music)",
+        "25 - MDX23C (vocals, instrumental)",
+        "26 - Ensemble (vocals, instrumental) [PREMIUM ONLY]",
+        "27 - Demucs4 Vocals 2023 (vocals, instrumental)",
+        "28 - Ensemble (vocals, instrumental, bass, drums, other) [PREMIUM ONLY]",
+        "29 - MVSep Piano (piano, other)",
+        "30 - Ensemble All-In (vocals, bass, drums, piano, guitar, lead/back vocals, other) [PREMIUM ONLY]",
+        "31 - MVSep Guitar (guitar, other)",
+        "33 - Vit Large 23 (vocals, instrum)",
+        "34 - MVSep Crowd removal (crowd, other)",
+        "35 - MVSep MelBand Roformer (vocals, instrum)",
+        "36 - BandIt Plus (speech, music, effects)",
+        "37 - DrumSep (kick, snare, cymbals, toms)",
+        "38 - LarsNet (kick, snare, cymbals, toms, hihat)",
+        "39 - Whisper (extract text from audio)",
+        "40 - BS Roformer (vocals, instrumental)",
+        "41 - MVSep Bass (bass, other)"
+    ]
 
 
 with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") as app:
@@ -293,7 +337,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             mindur2 = gr.Number(label="Min duration", minimum=1, maximum=10, value=1)
                             maxdur2 = gr.Number(label="Max duration", minimum=1, maximum=10, value=5)
                             name_for_split_files2 = gr.Textbox(label="Name for split files")
-                            strict = gr.Checkbox(False, label="Enable strict duration?", info="Use this option if you want to minimize the '(audio_file) is less than 0.76 seconds' warning on Colab. Keep in mind that this only applies for min duration, max is ignored.")
+                            strict = gr.Checkbox(True, label="Enable strict duration?", info="Use this option if you want to minimize the '(audio_file) is less than 0.76 seconds' warning on Colab. Keep in mind that this only applies for min duration, max is ignored.")
                             audiofileuploadbtn = gr.Button("Split", variant='primary')
                         audiofileuploadbtn.click(
                             split_wav_or_mp3_file,
@@ -341,13 +385,19 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             with gr.Row():
                                 mvsep_key = gr.Textbox(placeholder="Enter your MVSEP API key.", label="API key")
                                 audio_file = gr.File(file_count='single', file_types=[".mp3"], label="Audio file")
-                                sep_int = gr.Number(11, label="Separation type (default is 11).", minimum=0, maximum=40, interactive=True)
+                                sep_int = gr.Number(11, label="Separation type (default is 11).", minimum=0, maximum=41, interactive=True)
                                 send_req = gr.Button("Send request", variant='primary')
+                                get_available_models = gr.Button("Get available models", variant='secondary')
                             send_req.click(
                                 mvsep_api_request,
                                 [mvsep_key, audio_file, sep_int],
                                 [gr.Text(label="Output")]
                             )
+                            get_available_models.click(
+                                mvsep_list_available_models,
+                                None,
+                                [gr.Text(label="Models")]
+                            )
                 with gr.Tab("Get status of request"):
                     with gr.Row():
                         with gr.Column():
@@ -367,6 +417,7 @@ with gr.Blocks(theme='sudeepshouche/minimalist', title="Global Dataset Maker") a
                             )
 
         with gr.TabItem("Changelog"):
+            gr.Markdown("v0.99.4 - Added a button to display the available models for MVSEP.")
             gr.Markdown("v0.99.3 - Added MVSEP in Misc Tools. This is VERY EXPERIMENTAL and there will be bugs present.")
             gr.Markdown("v0.99.2 - Added an mp4 file converter.")
             gr.Markdown("v0.99.1 - Removed very old tools (including the 'Upcoming Features' tab) that did not fit with the nature of the program.")