Update app.py

app.py

@@ -7,6 +7,24 @@ import gradio
 
 pipe = pipeline("summarization", model="Falconsai/text_summarization")
 
+def extract_youtube_id(url):
+    """
+    Extracts the YouTube video ID from a given URL.
+
+    Args:
+        url (str): The YouTube video URL.
+
+    Returns:
+        str: The extracted video ID, or None if no match is found.
+    """
+    # Regular expression to match YouTube video IDs
+    pattern = r"(?:https?:\/\/)?(?:www\.)?(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|(?:v|e(?:mbed)?)\/|\S*?[?&]v=)|youtu\.be\/)([a-zA-Z0-9_-]{11})"
+    match = re.search(pattern, url)
+
+    if match:
+        return match.group(1)
+    return None
+
 def summary(text_transcript):
     output = pipe(text_transcript)
     return output[0]['summary_text']