Update app.py

app.py

@@ -1,20 +1,30 @@
 import gradio as gr
 import subprocess
-import os
 
-def run_ultrasinger(opt_i, opt_o, mode, whisper_model, language, crepe_model, extra, device):
+def run_ultrasinger(opt_i, youtube_link, opt_o, mode, whisper_model, language, crepe_model, extra, device):
     # Construct the command based on inputs
-    cmd = ["python", "src/UltraSinger.py"]
+    cmd = ["python", "UltraSinger.py"]
 
-    # Add options
+    # Add input option
     if opt_i:
-        cmd.extend(["-i", opt_i.name])
+        cmd.extend(["-i", f'"{opt_i.name}"'])
+    elif youtube_link:
+        cmd.extend(["-i", f'"{youtube_link}"'])
+    else:
+        return "Error: No input file or YouTube link provided", ""
+
+    # Add output folder option
     if opt_o:
-        cmd.extend(["-o", opt_o])
+        cmd.extend(["-o", f'"{opt_o}"'])
 
     # Add mode
-    if mode:
-        cmd.extend(mode.split())
+    if mode != "default":
+        mode_flags = {
+            "Create Ultrastar txt file": "-u",
+            "Create MIDI file": "-m",
+            "Create sheet file": "-s"
+        }
+        cmd.append(mode_flags[mode])
 
     # Add transcription options
     if whisper_model:
@@ -38,6 +48,9 @@ def run_ultrasinger(opt_i, opt_o, mode, whisper_model, language, crepe_model, ex
     if device:
         cmd.extend(device.split())
 
+    # Debug: Print the command to check if it's constructed correctly
+    print("Running command:", ' '.join(cmd))
+
     # Execute the command
     try:
         result = subprocess.run(cmd, capture_output=True, text=True)
@@ -53,11 +66,15 @@ 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, YouTube link)")
+opt_i = gr.File(label="Ultrastar.txt or audio file (.mp3, .wav)")
+youtube_link = gr.Textbox(label="YouTube Link", placeholder="Enter YouTube URL here")
 opt_o = gr.Textbox(label="Output folder")
-mode = gr.Dropdown(
+mode = gr.Radio(
     label="Mode options", 
-    choices=["default", "-u", "-m", "-s", "-r", "-p"], 
+    choices=[
+        "default", "Create Ultrastar txt file", "Create MIDI file", 
+        "Create sheet file"
+    ], 
     value="default"
 )
 whisper_model = gr.Dropdown(
@@ -98,7 +115,7 @@ error_text = gr.Textbox(label="Error Output")
 # Define Gradio interface for UltraSinger
 ultrasinger_tab = gr.Interface(
     fn=run_ultrasinger,
-    inputs=[opt_i, opt_o, mode, whisper_model, language, crepe_model, extra, device],
+    inputs=[opt_i, youtube_link, opt_o, mode, whisper_model, language, crepe_model, extra, device],
     outputs=[output_text, error_text],
     title="UltraSinger UI",
     description="Upload an Ultrastar.txt or an audio file, set the options, and run UltraSinger."