Update app.py

app.py

@@ -5,14 +5,21 @@ 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
+# 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
+
+def parseYoutubeURL(url):
+    url=str(url)
+   data = re.findall(r"(?:v=|\/)([0-9A-Za-z_-]{11}).*", url)
+   if data:
+       return data[0]
+   return ""
 
 
 
@@ -31,7 +38,8 @@ within 250 words. Please provide the summary of the text given here:  """
 def extract_transcript_details(youtube_video_url):
     try:
         # video_id=youtube_video_url.split("=")[1]
-        video_id=extract_video_id(youtube_link)
+        # video_id=extract_video_id(youtube_link)
+        video_id=parseYoutubeUR(youtube_link)
         
         transcript_text=YouTubeTranscriptApi.get_transcript(video_id)
 
@@ -56,7 +64,8 @@ youtube_link = st.text_input("Enter YouTube Video Link:")
 
 if youtube_link:
     # video_id = youtube_link.split("=")[1]
-    video_id=extract_video_id(youtube_link)
+    # video_id=extract_video_id(youtube_link)
+    video_id=parseYoutubeUR(youtube_link)
     print(video_id)
     st.image(f"http://img.youtube.com/vi/{video_id}/0.jpg", use_column_width=True)