Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import subprocess
 import os
 
-def run_ultrasinger(opt_i, opt_o, mode, transcription, pitcher, extra, device):
+def run_ultrasinger(opt_i, opt_o, mode, whisper_model, language, crepe_model, extra, device):
     # Construct the command based on inputs
     cmd = ["python", "UltraSinger.py"]
 
@@ -17,12 +17,14 @@ def run_ultrasinger(opt_i, opt_o, mode, transcription, pitcher, extra, device):
         cmd.extend(mode.split())
 
     # Add transcription options
-    if transcription:
-        cmd.extend(transcription.split())
+    if whisper_model:
+        cmd.extend(["--whisper", whisper_model])
+    if language:
+        cmd.extend(["--language", language])
 
     # Add pitcher options
-    if pitcher:
-        cmd.extend(pitcher.split())
+    if crepe_model:
+        cmd.extend(["--crepe", crepe_model])
 
     # Add extra options
     if extra:
@@ -40,11 +42,27 @@ def run_ultrasinger(opt_i, opt_o, mode, transcription, pitcher, extra, device):
         return str(e), "Error occurred during execution"
 
 # Define Gradio inputs and outputs
-opt_i = gr.File(label="Ultrastar.txt or audio file (.mp3, .wav, YouTube link)")
+opt_i = gr.File(label="Ultrastar.txt or audio file (.mp3, .wav, .txt)")
 opt_o = gr.Textbox(label="Output folder")
-mode = gr.Textbox(label="Mode options (e.g., default -u -m -s)")
-transcription = gr.Textbox(label="Transcription options (e.g., --whisper large-v2 --language en)")
-pitcher = gr.Textbox(label="Pitcher options (e.g., --crepe full)")
+mode = gr.Dropdown(
+    label="Mode options", 
+    choices=["default", "-u", "-m", "-s", "-r", "-p"], 
+    value="default"
+)
+whisper_model = gr.Dropdown(
+    label="Whisper Model", 
+    choices=[
+        "tiny", "base", "small", "medium", "large-v1", "large-v2", 
+        "tiny.en", "base.en", "small.en", "medium.en"
+    ], 
+    value="large-v2"
+)
+language = gr.Textbox(label="Language (e.g., en)")
+crepe_model = gr.Dropdown(
+    label="Crepe Model", 
+    choices=["tiny", "full"], 
+    value="full"
+)
 extra = gr.Textbox(label="Extra options (e.g., --hyphenation True)")
 device = gr.Textbox(label="Device options (e.g., --force_cpu False)")
 
@@ -54,9 +72,9 @@ error_text = gr.Textbox(label="Error Output")
 # Create Gradio interface
 interface = gr.Interface(
     fn=run_ultrasinger,
-    inputs=[opt_i, opt_o, mode, transcription, pitcher, extra, device],
+    inputs=[opt_i, opt_o, mode, whisper_model, language, crepe_model, extra, device],
     outputs=[output_text, error_text],
-    title="UltraSinger App",
+    title="UltraSinger UI",
     description="Upload an Ultrastar.txt or an audio file, set the options, and run UltraSinger."
 )