Update app.py

app.py

@@ -14,9 +14,9 @@ def get_youtube_captions(youtube_url):
     - The captions of the video in SRT format, or a message if captions are not available.
     """
     # Extract video ID from YouTube URL (for both regular videos and Shorts)
-    if "shorts" in youtube_url:
-        # For YouTube Shorts, extract the video ID after "shorts/"
-        video_id = youtube_url.split("shorts/")[-1].split("?")[0]
+    if "si=" in youtube_url:
+        # For YouTube Shorts or URLs with "si=", extract the video ID after "si="
+        video_id = youtube_url.split("si=")[-1].split("?")[0]
     else:
         # For regular YouTube videos, extract the video ID after "v="
         video_id = youtube_url.split("v=")[-1].split("&")[0]