Update app.py

app.py

@@ -6,22 +6,30 @@ from transformers import pipeline
 summarizer = pipeline("summarization")
 
 def generate_summary(video_url):
-    # Get the video title and description from YouTube
-    yt = YouTube(video_url)
-    title = yt.title
-    description = yt.description
+    try:
+        # Get the video title and description from YouTube
+        yt = YouTube(video_url)
+        title = yt.title
+        description = yt.description
 
-    # Generate a summary of the video using the AI model
-    summary = summarizer(description, max_length=130, min_length=30, do_sample=False)
+        # Check if the description is empty
+        if not description.strip():
+            return "Description is empty. Unable to generate a summary."
 
-    # Return the summary
-    return summary[0]["summary_text"]
+        # Generate a summary of the video using the AI model
+        summary = summarizer(description, max_length=130, min_length=30, do_sample=False)
+
+        # Return the summary
+        return summary[0]["summary_text"]
+    
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
 
 # Create the Gradio interface
 demo = gr.Interface(
-    generate_summary,
-    gr.Textbox(label="YouTube Video URL"),
-    gr.Textbox(label="Video Summary"),
+    fn=generate_summary,
+    inputs=gr.Textbox(label="YouTube Video URL", placeholder="Enter the URL of the YouTube video"),
+    outputs=gr.Textbox(label="Video Summary", lines=5),
     title="AI Video Summarizer",
     description="Enter a YouTube video URL to generate a summary of the video",
 )