Update tabs/infer/infer.py

tabs/infer/infer.py

@@ -15,6 +15,8 @@ def get_current_models(models_dir):
         logger.error(f"Error accessing models directory: {e}\n{format_exc()}")
         raise gr.Error(f"Failed to list models. Check directory permissions: {e}")
 
+def swap_visibility():
+    return gr.update(visible=True), gr.update(visible=False), gr.update(value=''), gr.update(value=None)
 
 
 def inference_tab(status_message, rvc_model):
@@ -44,13 +46,19 @@ def inference_tab(status_message, rvc_model):
         
         with gr.Group():
             with gr.Row(equal_height=True):
-                with gr.Column(visible=True, variant="panel", scale=1):
-                    song_input = gr.Textbox(
-                        label="Song Input",
-                        placeholder="Enter a YouTube link or local file path",
-                        lines=1
-                    )
-                
+                with gr.Column() as yt_link_col:
+                    song_input = gr.Text(label='Song input', info='Link to a song on YouTube or full path to a local file. For file upload, click the button below.')
+                    show_file_upload_button = gr.Button('Upload file instead?')
+
+                with gr.Column(visible=False) as file_upload_col:
+                    local_file = gr.Audio(label='Audio file')
+                    
+                    show_yt_link_button = gr.Button('Paste YouTube link instead?')
+                    song_input_file.upload(process_file_upload, inputs=[song_input_file], outputs=[local_file, song_input])
+
+            show_file_upload_button.click(swap_visibility, outputs=[file_upload_col, yt_link_col, song_input, local_file])
+            show_yt_link_button.click(swap_visibility, outputs=[yt_link_col, file_upload_col, song_input, local_file])
+
         with gr.Accordion("RVC Settings", open=False):
             with gr.Row(equal_height=True):
                 with gr.Column(variant="panel", scale=1):
@@ -87,7 +95,7 @@ def inference_tab(status_message, rvc_model):
                 inst_gain = gr.Slider(-20, 20, value=0, step=1, label="Instrumental Gain (dB)")
 
             with gr.Row(equal_height=True):
-                keep_files = gr.Checkbox(label="Keep Intermediate Files", value=True)
+                keep_files = gr.Checkbox(label="Keep Intermediate Files", value=False, visible=False)
                 vocal_only = gr.Checkbox(label="Vocal-Only Conversion", value=False)
 
         with gr.Row(equal_height=True):