Update app.py

app.py

@@ -1,64 +1,59 @@
+import re
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api.formatters import TextFormatter
-import re
 import torch
 import gradio as gr
 from transformers import pipeline
 
-# model_path = ("../Model/models--sshleifer--distilbart-cnn-12-6/snapshots"
-#               "/a4f8f3ea906ed274767e9906dbaede7531d660ff")
-# text_summary = pipeline("summarization", model=model_path
-#                 ,torch_dtype=torch.bfloat16)
-
 text_summary = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
 
-def summary(input):
+# model_path = ("../Models/models--sshleifer--distilbart-cnn-12-6/snapshots"
+#               "/a4f8f3ea906ed274767e9906dbaede7531d660ff")
+# text_summary = pipeline("summarization", model=model_path,
+#                 torch_dtype=torch.bfloat16)
+
+def summary (input):
     output = text_summary(input)
     return output[0]['summary_text']
 
-# Function to extract video ID from the YouTube URL
 def extract_video_id(url):
-    video_id = None
-    # Updated regex to handle extra parameters in the YouTube URL
-    regex = r"(?:https?:\/\/)?(?:www\.)?(?:youtube\.com\/(?:[^\/\n\s]+\/\S+\/|\S*[\?&]v=|\S*[\?&]v%2F)?([\w-]+))|(?:youtu\.be\/([\w-]+))"
+    # 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:
-        video_id = match.group(1) if match.group(1) else match.group(2)
-    return video_id
+        return match.group(1)
+    return None
+
 
-# Function to get transcript for a YouTube video
-def get_transcript(video_url):
+def get_youtube_transcript(video_url):
     video_id = extract_video_id(video_url)
-    if video_id:
-        try:
-            transcript = YouTubeTranscriptApi.get_transcript(video_id)
+    if not video_id:
+        return "Video ID could not be extracted."
 
-            # Formatting transcript into plain text
-            formatter = TextFormatter()
-            formatted_transcript = formatter.format_transcript(transcript)
-            summary_text = summary(formatted_transcript)
-            return summary_text
+    try:
+        # Fetch the transcript
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
 
-        except Exception as e:
-            return f"Error retrieving transcript: {e}"
-    else:
-        return "Invalid YouTube URL."
+        # Format the transcript into plain text
+        formatter = TextFormatter()
+        text_transcript = formatter.format_transcript(transcript)
+        summary_text = summary(text_transcript)
 
-# Main function to take URL input and fetch the transcript
-# if __name__ == "__main__":
-#     youtube_url = input("Enter the YouTube video URL: ")
-#     transcript = get_transcript(youtube_url)
-#     print("\nTranscript:\n")
-#     print(transcript)
+        return summary_text
+    except Exception as e:
+        return f"An error occurred: {e}"
 
 
-gr.close_all()
+# Example URL (Replace this with the actual URL when using the script)
+# video_url = "https://youtu.be/5PibknhIsTc"
+# print(get_youtube_transcript(video_url))
 
-# demo = gr.Interface(fn=summary,inputs=["text"],outputs=["text"])
+gr.close_all()
 
-demo = gr.Interface(fn = get_transcript,
-                    inputs=[gr.Textbox(label="Input youtube url to summarize",lines=1)],
-                    outputs=[gr.Textbox(label="Summarized Text ",lines=4)],
-                    title="@SherryAhuja Project : Youtube Video Summarizer",
-                    description="This AI application will be used to summarize the Youtube Video Script")
+# demo = gr.Interface(fn=summary, inputs="text",outputs="text")
+demo = gr.Interface(fn=get_youtube_transcript,
+                    inputs=[gr.Textbox(label="Input YouTube Url to summarize",lines=1)],
+                    outputs=[gr.Textbox(label="Summarized text",lines=4)],
+                    title="@GenAILearniverse Project 2: YouTube Script Summarizer",
+                    description="THIS APPLICATION WILL BE USED TO SUMMARIZE THE YOUTUBE VIDEO SCRIPT.")
 demo.launch()