Update app.py

app.py

@@ -53,8 +53,12 @@ def get_youtube_captions(youtube_url):
     Returns:
     - The captions of the video in SRT format, or a message if captions are not available.
     """
-    video_id = youtube_url.split("v=")[-1].split("&")[0]
-    
+    # Extract video ID from YouTube URL (for both regular videos and Shorts)
+    if "shorts" in youtube_url:
+        video_id = youtube_url.split("shorts/")[-1].split("?")[0]
+    else:
+        video_id = youtube_url.split("v=")[-1].split("&")[0]
+
     try:
         # Fetch transcript for the video
         transcript = YouTubeTranscriptApi.get_transcript(video_id)