Update app.py

app.py

@@ -25,11 +25,11 @@ yt_video_dir = "./yt_dir"
 os.makedirs(yt_video_dir, exist_ok=True)
 
 
-def get_audio_from_yt_video(yt_link: str, tag: str):
+def get_audio_from_yt_video_input(yt_link: str)
     try:
         yt = pt.YouTube(yt_link)
         t = yt.streams.filter(only_audio=True)
-        filename = os.path.join(yt_video_dir, tag + ".wav")
+        filename = os.path.join(yt_video_dir, "input.wav")
         t[0].download(filename=filename)
     except VideoUnavailable as e:
         warnings.warn(f"Video Not Found at {yt_link} ({e})")
@@ -37,6 +37,17 @@ def get_audio_from_yt_video(yt_link: str, tag: str):
 
     return filename, filename
 
+def get_audio_from_yt_video_ref(yt_link: str)
+    try:
+        yt = pt.YouTube(yt_link)
+        t = yt.streams.filter(only_audio=True)
+        filename = os.path.join(yt_video_dir, "reference.wav")
+        t[0].download(filename=filename)
+    except VideoUnavailable as e:
+        warnings.warn(f"Video Not Found at {yt_link} ({e})")
+        filename = None
+
+    return filename, filename
 
 def inference(file_uploaded_in, file_uploaded_ref):
 
@@ -88,8 +99,8 @@ with gr.Blocks() as demo:
                         label="Input Audio Extracted from the YouTube Video", interactive=False
                     )
                     yt_btn_in.click(
-                        get_audio_from_yt_video,
-                        inputs=[yt_link_in, 'input'],
+                        get_audio_from_yt_video_input,
+                        inputs=[yt_link_in],
                         outputs=[yt_audio_path_in, file_uploaded_in],
                     )
             with gr.Blocks():
@@ -105,8 +116,8 @@ with gr.Blocks() as demo:
                         label="Reference Audio Extracted from the YouTube Video", interactive=False
                     )
                     yt_btn_ref.click(
-                        get_audio_from_yt_video,
-                        inputs=[yt_link_ref, 'reference'],
+                        get_audio_from_yt_video_ref,
+                        inputs=[yt_link_ref],
                         outputs=[yt_audio_path_ref, file_uploaded_ref],
                     )
             with gr.Column():