Update app.py

app.py

@@ -3,6 +3,19 @@ import streamlit as st
 import os
 import google.generativeai as genai
 
+import re
+
+def extract_video_id(url):
+    # Regular expression pattern to match the video ID
+    pattern = r'(?<=/|v=|vi/|youtu\.be/|embed/)[^#&?]*'
+    match = re.search(pattern, url)
+    if match:
+        return match.group(0)
+    else:
+        return None
+
+
+
 from youtube_transcript_api import YouTubeTranscriptApi
 
 secret_key = os.getenv("SECRET_KEY")
@@ -17,7 +30,8 @@ within 250 words. Please provide the summary of the text given here:  """
 ## getting the transcript data from yt videos
 def extract_transcript_details(youtube_video_url):
     try:
-        video_id=youtube_video_url.split("=")[1]
+        # video_id=youtube_video_url.split("=")[1]
+        video_id=extract_video_id(youtube_link)
         
         transcript_text=YouTubeTranscriptApi.get_transcript(video_id)
 
@@ -41,7 +55,8 @@ st.title("YouTube Transcript to Detailed Notes Converter")
 youtube_link = st.text_input("Enter YouTube Video Link:")
 
 if youtube_link:
-    video_id = youtube_link.split("=")[1]
+    # video_id = youtube_link.split("=")[1]
+    video_id=extract_video_id(youtube_link)
     print(video_id)
     st.image(f"http://img.youtube.com/vi/{video_id}/0.jpg", use_column_width=True)