Update app.py

app.py

@@ -15,9 +15,14 @@ def get_youtube_captions(youtube_url):
     """
     # 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]
     else:
+        # For regular YouTube videos, extract the video ID after "v="
         video_id = youtube_url.split("v=")[-1].split("&")[0]
+    
+    # Print extracted video ID for debugging
+    print(f"Extracted Video ID: {video_id}")
 
     try:
         # Fetch transcript for the video
@@ -39,10 +44,13 @@ youtube_url = st.text_input(
 )
 
 if youtube_url:
+    # Clean URL before passing to the function (remove extra parts)
+    youtube_url_cleaned = youtube_url.split("?")[0]  # Remove URL parameters
+    
     try:
         with st.spinner("Fetching and processing the YouTube video..."):
             # Get captions using youtube-transcript-api
-            captions = get_youtube_captions(youtube_url)
+            captions = get_youtube_captions(youtube_url_cleaned)
 
             # Check if captions are available
             if "Error" not in captions: