Update app.py

app.py

@@ -206,10 +206,10 @@ def _download_yt_audio(yt_url, filename):
         raise gr.Error(f"Requested format not available. Available formats: {available_formats}", duration=20)
         
 def _return_yt_video_id(yt_url):
-    if "https://www.youtube.com/watch?v=" in yt_url:
-        video_id = yt_url.split("?v=")[-1]
-    elif "https://youtu.be/" in yt_url:
-        video_id = yt_url.split("be/")[1]
+    if "youtube.com/watch?v=" in yt_url:
+        video_id = yt_url.split("?v=")[1].split("&")[0]
+    elif "youtu.be/" in yt_url:
+        video_id = yt_url.split("youtu.be/")[1]
     return video_id
     
 def _return_yt_html_embed(yt_url):