Update app.py

app.py

@@ -20,13 +20,15 @@ def run_ultrasinger(opt_i, opt_o, mode, whisper_model, language, crepe_model, ex
     if whisper_model:
         cmd.extend(["--whisper", whisper_model])
     if language:
-        cmd.extend(["--language", language])
+        language_codes = {
+            "English": "en", "French": "fr", "German": "de", "Spanish": "es",
+            "Italian": "it", "Japanese": "ja", "Chinese": "zh", "Dutch": "nl",
+            "Ukrainian": "uk", "Portuguese": "pt"
+        }
+        cmd.extend(["--language", language_codes[language]])
 
     # Add pitcher options
-    if crepe_model:
-        cmd.extend(["--crepe", "full"])
-    else:
-        cmd.extend(["--crepe", "tiny"])
+    cmd.extend(["--crepe", crepe_model])
 
     # Add extra options
     if extra:
@@ -51,7 +53,7 @@ def load_text_file(file_path):
         return str(e)
 
 # Define Gradio inputs and outputs for UltraSinger
-opt_i = gr.File(label="Ultrastar.txt or audio file (.mp3, .wav, .txt)")
+opt_i = gr.File(label="Ultrastar.txt or audio file (.mp3, .wav, YouTube link)")
 opt_o = gr.Textbox(label="Output folder")
 mode = gr.Dropdown(
     label="Mode options", 
@@ -66,8 +68,19 @@ whisper_model = gr.Dropdown(
     ], 
     value="large-v2"
 )
-language = gr.Textbox(label="Language (e.g., en)")
-crepe_model = gr.Checkbox(label="Use Full Crepe Model", value=True)
+language = gr.Dropdown(
+    label="Language", 
+    choices=[
+        "English", "French", "German", "Spanish", "Italian", 
+        "Japanese", "Chinese", "Dutch", "Ukrainian", "Portuguese"
+    ], 
+    value="English"
+)
+crepe_model = gr.Radio(
+    label="Crepe Model", 
+    choices=["full", "tiny"], 
+    value="full"
+)
 extra = gr.Textbox(label="Extra options (e.g., --hyphenation True)")
 device = gr.Dropdown(
     label="Device options",