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):
+def get_audio_from_yt_video(yt_link: str, tag: str):
     try:
         yt = pt.YouTube(yt_link)
         t = yt.streams.filter(only_audio=True)
-        filename = os.path.join(yt_video_dir, binascii.hexlify(os.urandom(8)).decode() + ".wav")
+        filename = os.path.join(yt_video_dir, tag + ".wav")
         t[0].download(filename=filename)
     except VideoUnavailable as e:
         warnings.warn(f"Video Not Found at {yt_link} ({e})")
@@ -89,7 +89,7 @@ with gr.Blocks() as demo:
                     )
                     yt_btn_in.click(
                         get_audio_from_yt_video,
-                        inputs=[yt_link_in],
+                        inputs=[yt_link_in, 'input'],
                         outputs=[yt_audio_path_in, file_uploaded_in],
                     )
             with gr.Blocks():
@@ -106,7 +106,7 @@ with gr.Blocks() as demo:
                     )
                     yt_btn_ref.click(
                         get_audio_from_yt_video,
-                        inputs=[yt_link_ref],
+                        inputs=[yt_link_ref, 'reference'],
                         outputs=[yt_audio_path_ref, file_uploaded_ref],
                     )
             with gr.Column():