Update app.py

app.py

@@ -1,53 +1,54 @@
-from youtube_transcript_api import YouTubeTranscriptApi
 import re
+from youtube_transcript_api import YouTubeTranscriptApi
+from youtube_transcript_api.formatters import TextFormatter
 import torch
-from transformers import pipeline
 import gradio as gr
+from transformers import pipeline
+
+text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
 
-text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
-                torch_dtype=torch.bfloat16)
 
 
-def summary(input):
+def summary (input):
     output = text_summary(input)
     return output[0]['summary_text']
 
 
-def get_video_id(url):
-    """Extract the video ID from a YouTube URL."""
-    pattern = r"(?:https?:\/\/)?(?:www\.)?youtube\.com\/watch\?v=([a-zA-Z0-9_-]{11})"
-    match = re.match(pattern, url)
+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)
-    else:
-        raise ValueError("Invalid YouTube URL")
+    return None
 
 
-def fetch_transcript_and_summary(url):
-    """Fetch the full transcript of a YouTube video and its summary."""
-    video_id = get_video_id(url)
-    transcript = YouTubeTranscriptApi.get_transcript(video_id)
-    full_transcript = " ".join([item["text"] for item in transcript])
-    summarized_transcript = summary(full_transcript)
-    return summarized_transcript
+def get_youtube_transcript(video_url):
+    video_id = extract_video_id(video_url)
+    if not video_id:
+        return "Video ID could not be extracted."
 
-
-# Example usage
-def get_video_summary(url):
-    """Main function to fetch and display transcript and summary."""
     try:
-        summarized_transcript = fetch_transcript_and_summary(url)
-        print(summarized_transcript)
+        # Fetch the transcript
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
+
+        # Format the transcript into plain text
+        formatter = TextFormatter()
+        text_transcript = formatter.format_transcript(transcript)
+        summary_text = summary(text_transcript)
+
+        return summary_text
     except Exception as e:
-        print(f"Error: {e}")
+        return f"An error occurred: {e}"
 
 
-# Call the function with your desired URL command line based
-# video_url = input("Enter the YouTube video URL: ")
-# get_video_summary(video_url)
+# Example URL (Replace this with the actual URL when using the script)
+# video_url = "https://youtu.be/5PibknhIsTc"
+# print(get_youtube_transcript(video_url))
 
+gr.close_all()
 
-# Web app Gradio based
+# demo = gr.Interface(fn=summary, inputs="text",outputs="text")
 gr.close_all()
 
 demo = gr.Interface(