Update app.py

app.py

@@ -12,9 +12,17 @@ def summary(text_youtube):
     output = text_summary(text_youtube)
     return output[0]['summary_text']
 
+# def extract_video_id(url):
+#     # Regex to extract the video ID from various YouTube URL formats
+#     regex = r"(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})"
+#     match = re.search(regex, url)
+#     if match:
+#         return match.group(1)
+#     return None
+
 def extract_video_id(url):
-    # Regex to extract the video ID from various YouTube URL formats
-    regex = r"(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})"
+    # Updated regex to extract the video ID from various YouTube URL formats, including /shorts/ and others with parameters
+    regex = r"(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=|shorts\/)|youtu\.be\/)([a-zA-Z0-9_-]{11})"
     match = re.search(regex, url)
     if match:
         return match.group(1)