Create app.py

app.py

@@ -0,0 +1,59 @@
+from youtube_transcript_api import YouTubeTranscriptApi
+import re
+import torch
+import gradio as gr
+
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+
+text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6")
+# model_path = ("../Models/models--sshleifer--distilbart-cnn-6-6/snapshots/d2fde4ca965ba893255479612e4b801aa6500029")
+
+# text_summary = pipeline("summarization", model=model_path,
+#                 torch_dtype=torch.bfloat16)
+
+
+def split_text_to_chunks(text, chunk_size=1024):
+    return [text[i:i+chunk_size] for i in range(0, len(text), chunk_size)]
+
+def summary(input):
+    output = ''
+    text_array = split_text_to_chunks(input)
+    for i, chunk in enumerate(text_array):
+        output += text_summary(chunk[:1024])[0]['summary_text']
+    return output
+
+def get_video_id(url):
+    """Extracts the video ID from a YouTube URL."""
+    pattern = r"(?:v=|\/)([0-9A-Za-z_-]{11}).*"
+    match = re.search(pattern, url)
+    return match.group(1) if match else None
+
+def get_transcript(video_url):
+    """Fetches the transcript of a YouTube video."""
+    video_id = get_video_id(video_url)
+    if not video_id:
+        return "Invalid YouTube URL!"
+
+    try:
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
+        transcript_text = "\n".join([entry["text"] for entry in transcript])
+        output = summary(transcript_text)
+        return output
+    except Exception as e:
+        return f"Error fetching transcript: {e}"
+
+# youtube_url = input("Enter YouTube URL: ")
+# transcript = get_transcript(youtube_url)
+# output = summary(transcript)
+# print("\n--- Video Transcript ---\n")
+# print(output)
+
+gr.close_all()
+# demo = gr.Interface(fn=summary,inputs="text", outputs="text")
+demo = gr.Interface(fn=get_transcript,
+                    inputs=[gr.Textbox(label="Input text to summarize", lines=6)],
+                    outputs=[gr.Textbox(label="Summarized text", lines=4)],
+                    title="Text Summarizer",
+                    description="This application will be used to summarise the text")
+demo.launch()