Create app.py

app.py

@@ -0,0 +1,61 @@
+from youtube_transcript_api import YouTubeTranscriptApi
+import re
+import torch
+from transformers import pipeline
+import gradio as gr
+
+text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6",
+                torch_dtype=torch.bfloat16)
+
+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)
+    if match:
+        return match.group(1)
+    else:
+        raise ValueError("Invalid YouTube URL")
+
+
+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
+
+
+# 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)
+    except Exception as e:
+        print(f"Error: {e}")
+
+
+# Call the function with your desired URL command line based
+# video_url = input("Enter the YouTube video URL: ")
+# get_video_summary(video_url)
+
+
+# Web app Gradio based
+gr.close_all()
+
+demo = gr.Interface(
+    fn=fetch_transcript_and_summary,
+    inputs=[gr.Textbox(label="Input YouTube video url to summarize", lines=1)],
+    outputs=[gr.Textbox(label="Summarized YouTube video Script", lines=4)],
+    title="Project 02: YouTube Video Script Summarizer",
+    description="As understood from the title, if not already, this application will summarize your YouTube video"
+                " Script"
+)
+
+demo.launch()