Update app.py

app.py

@@ -1,75 +1,45 @@
-import re
 import gradio as gr
+import re
 from youtube_utils import get_youtube_captions, summarize_large_text_with_bart
 
+# Helper function to extract video ID from YouTube URL
 def extract_video_id(url):
-    """
-    Extract video ID from YouTube URLs using regular expressions.
-    """
     patterns = [
         r'(?:v=|\/)([0-9A-Za-z_-]{11}).*',  # Standard and shortened URLs
         r'(?:embed\/)([0-9A-Za-z_-]{11})',   # Embed URLs
         r'(?:youtu\.be\/)([0-9A-Za-z_-]{11})'  # Shortened URLs
     ]
+    
     for pattern in patterns:
         match = re.search(pattern, url)
         if match:
             return match.group(1)
     return None
 
-def summarize_youtube_video(video_url):
-    """
-    Summarize YouTube video by extracting captions and applying summarization.
-    """
+# Gradio function to process summarization
+def summarize_video(video_url):
     try:
-        # Extract video ID
         video_id = extract_video_id(video_url)
         if not video_id:
-            return "Invalid YouTube URL. Please provide a valid link.", None
-        
-        # Fetch captions
+            return "Invalid YouTube URL"
+
         captions = get_youtube_captions(video_id)
         if not captions:
-            return "Unable to fetch captions for this video. Ensure it has subtitles.", None
-        
-        # Generate summary
+            return "Unable to fetch video captions"
+
         summary = summarize_large_text_with_bart(captions)
-        return summary, f"https://www.youtube.com/watch?v={video_id}"
-    
+        return summary
     except Exception as e:
-        return f"An error occurred: {str(e)}", None
-
-# Gradio Interface
-def create_gradio_interface():
-    """
-    Create and return the Gradio Blocks interface.
-    """
-    with gr.Blocks(title="YouTube Video Summarizer") as interface:
-        gr.Markdown("# 🎥 YouTube Video Summarizer")
-        gr.Markdown(
-            "Provide a YouTube video URL, and this tool will extract captions and summarize the content using a BART model."
-        )
-        
-        with gr.Row():
-            video_url_input = gr.Textbox(
-                label="YouTube Video URL", 
-                placeholder="Enter the URL of the YouTube video"
-            )
-            summarize_button = gr.Button("Summarize")
-        
-        with gr.Row():
-            summary_output = gr.Textbox(label="Summary", lines=10, interactive=False)
-            video_link_output = gr.Textbox(label="YouTube Video Link", interactive=False)
-
-        summarize_button.click(
-            fn=summarize_youtube_video,
-            inputs=[video_url_input],
-            outputs=[summary_output, video_link_output]
-        )
+        return f"An error occurred: {e}"
 
-    return interface
+# Gradio interface
+iface = gr.Interface(
+    fn=summarize_video,
+    inputs=gr.Textbox(label="YouTube Video URL", placeholder="Enter the YouTube video URL here"),
+    outputs=gr.Textbox(label="Summary", lines=5),
+    title="YouTube Video Summarizer",
+    description="Enter a YouTube video URL to extract captions and summarize them using AI."
+)
 
-# Launch the interface
 if __name__ == "__main__":
-    interface = create_gradio_interface()
-    interface.launch()
+    iface.launch()