Update app.py

app.py

@@ -24,7 +24,9 @@ def run_ultrasinger(opt_i, opt_o, mode, whisper_model, language, crepe_model, ex
 
     # Add pitcher options
     if crepe_model:
-        cmd.extend(["--crepe", crepe_model])
+        cmd.extend(["--crepe", "full"])
+    else:
+        cmd.extend(["--crepe", "tiny"])
 
     # Add extra options
     if extra:
@@ -41,8 +43,15 @@ def run_ultrasinger(opt_i, opt_o, mode, whisper_model, language, crepe_model, ex
     except Exception as e:
         return str(e), "Error occurred during execution"
 
-# Define Gradio inputs and outputs
-opt_i = gr.File(label="Ultrastar.txt or audio file (.mp3, .wav, .txt)")
+def load_text_file(file_path):
+    try:
+        with open(file_path, 'r') as file:
+            return file.read()
+    except Exception as e:
+        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_o = gr.Textbox(label="Output folder")
 mode = gr.Dropdown(
     label="Mode options", 
@@ -58,26 +67,55 @@ whisper_model = gr.Dropdown(
     value="large-v2"
 )
 language = gr.Textbox(label="Language (e.g., en)")
-crepe_model = gr.Dropdown(
-    label="Crepe Model", 
-    choices=["tiny", "full"], 
-    value="full"
-)
+crepe_model = gr.Checkbox(label="Use Full Crepe Model", value=True)
 extra = gr.Textbox(label="Extra options (e.g., --hyphenation True)")
-device = gr.Textbox(label="Device options (e.g., --force_cpu False)")
+device = gr.Dropdown(
+    label="Device options",
+    choices=[
+        "", "--force_cpu True", "--force_cpu False", 
+        "--force_whisper_cpu True", "--force_whisper_cpu False", 
+        "--force_crepe_cpu True", "--force_crepe_cpu False"
+    ],
+    value=""
+)
 
 output_text = gr.Textbox(label="Standard Output")
 error_text = gr.Textbox(label="Error Output")
 
-# Create Gradio interface
-interface = gr.Interface(
+# 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],
     outputs=[output_text, error_text],
-    title="UltraSinger UI",
+    title="UltraSinger App",
     description="Upload an Ultrastar.txt or an audio file, set the options, and run UltraSinger."
 )
 
+# Define Gradio interfaces for the other two tabs
+tab1_content = gr.Interface(
+    fn=lambda: load_text_file("info.txt"),
+    inputs=[],
+    outputs=gr.Textbox(label="Tab 1 Content"),
+    title="Info"
+)
+
+tab2_content = gr.Interface(
+    fn=lambda: load_text_file("usdb.txt"),
+    inputs=[],
+    outputs=gr.Textbox(label="Tab 2 Content"),
+    title="FOR USDB USERS
+)
+
+# Create Gradio tabs
+with gr.Blocks(theme="soft") as demo:
+    with gr.Tabs():
+        with gr.TabItem("UltraSinger"):
+            ultrasinger_tab.render()
+        with gr.TabItem("Info"):
+            tab1_content.render()
+        with gr.TabItem("FOR USDB USERS"):
+            tab2_content.render()
+
 # Launch the app
 if __name__ == "__main__":
-    interface.launch()
+    demo.launch()