modify youtube download

app.py

@@ -47,6 +47,12 @@ def process_youtube_url(url):
     except Exception as e:
         return None, f"Error processing YouTube URL: {str(e)}"
 
+def download_youtube_audios(input_youtube_url, reference_youtube_url):
+    input_audio, input_error = process_youtube_url(input_youtube_url) if input_youtube_url else (None, None)
+    reference_audio, reference_error = process_youtube_url(reference_youtube_url) if reference_youtube_url else (None, None)
+    
+    return input_audio, reference_audio, input_error, reference_error
+
 def process_audio_with_youtube(input_audio, input_youtube_url, reference_audio, reference_youtube_url):
     if input_youtube_url:
         input_audio, error = process_youtube_url(input_youtube_url)
@@ -191,6 +197,9 @@ with gr.Blocks() as demo:
         with gr.Row():
             input_youtube_url = gr.Textbox(label="Input YouTube URL")
             reference_youtube_url = gr.Textbox(label="Reference YouTube URL")
+        
+        download_button = gr.Button("Download YouTube Audios")
+    
         with gr.Row():
             input_audio_yt = gr.Audio(label="Source Audio (Do not put when using YouTube URL)")
             reference_audio_yt = gr.Audio(label="Reference Style Audio (Do not put when using YouTube URL)")
@@ -200,24 +209,42 @@ with gr.Blocks() as demo:
         
         with gr.Row():
             with gr.Column():
-                output_audio = gr.Audio(label="Output Audio y'", type='numpy')
-                normalized_input = gr.Audio(label="Normalized Source Audio", type='numpy')
-            param_output = gr.Textbox(label="Predicted Parameters", lines=5)
+                output_audio_yt = gr.Audio(label="Output Audio y'", type='numpy')
+                normalized_input_yt = gr.Audio(label="Normalized Source Audio", type='numpy')
+            param_output_yt = gr.Textbox(label="Predicted Parameters", lines=5)
         
         error_message_yt = gr.Textbox(label="Error Message", visible=False)
 
-        def process_and_handle_errors(input_audio, input_youtube_url, reference_audio, reference_youtube_url):
-            result = process_audio_with_youtube(input_audio, input_youtube_url, reference_audio, reference_youtube_url)
-            if len(result) == 3 and isinstance(result[2], str):  # Error occurred check
-                return None, None, None, gr.update(visible=True, value=result[2])
-            return result[0], result[1], result[2], gr.update(visible=False, value="")
+        def handle_download_youtube_audios(input_youtube_url, reference_youtube_url):
+            input_audio, reference_audio, input_error, reference_error = download_youtube_audios(input_youtube_url, reference_youtube_url)
+            if input_error or reference_error:
+                return None, None, gr.update(visible=True, value=input_error or reference_error)
+            return input_audio, reference_audio, gr.update(visible=False, value="")
+
+        download_button.click(
+            handle_download_youtube_audios,
+            inputs=[input_youtube_url, reference_youtube_url],
+            outputs=[input_audio_yt, reference_audio_yt, error_message_yt]
+        )
 
         process_button_yt.click(
-            process_and_handle_errors, 
-            inputs=[input_audio_yt, input_youtube_url, reference_audio_yt, reference_youtube_url], 
-            outputs=[output_audio_yt, param_output_yt, normalized_input, error_message_yt]
+            process_audio, 
+            inputs=[input_audio_yt, reference_audio_yt], 
+            outputs=[output_audio_yt, param_output_yt, normalized_input_yt]
         )
 
+        # def process_and_handle_errors(input_audio, input_youtube_url, reference_audio, reference_youtube_url):
+        #     result = process_audio_with_youtube(input_audio, input_youtube_url, reference_audio, reference_youtube_url)
+        #     if len(result) == 3 and isinstance(result[2], str):  # Error occurred check
+        #         return None, None, None, gr.update(visible=True, value=result[2])
+        #     return result[0], result[1], result[2], gr.update(visible=False, value="")
+
+        # process_button_yt.click(
+        #     process_and_handle_errors, 
+        #     inputs=[input_audio_yt, input_youtube_url, reference_audio_yt, reference_youtube_url], 
+        #     outputs=[output_audio_yt, param_output_yt, normalized_input_yt, error_message_yt]
+        # )
+
     gr.Markdown("## Step 2: Inference Time Optimization (ITO)")
     
     with gr.Row():