Update app.py

app.py

@@ -1,21 +1,21 @@
 import gradio as gr
 from langchain_community.document_loaders import YoutubeLoader
-from langchain.chains.summarize import load_summarize_chain
 from langchain_core.prompts import ChatPromptTemplate
 from youtube_transcript_api import NoTranscriptFound
+from pytube.exceptions import PytubeError
 
 def load_youtube_video_info(url):
     # Initialize YouTube loader with preferred language as Portuguese
-    loader = YoutubeLoader.from_youtube_url(url, add_video_info=True, language="pt")
-
     try:
+        loader = YoutubeLoader.from_youtube_url(url, add_video_info=True, language="pt")
+
         # Load the video information and transcript
         result = loader.load()
 
         if result:
             # Retrieve metadata information
-            author = result[0].metadata['author']
-            length = result[0].metadata['length']
+            author = result[0].metadata.get('author', 'Unknown Author')
+            length = result[0].metadata.get('length', 'Unknown Length')
             # Check if transcript is available
             transcript = result[0].page_content if hasattr(result[0], 'page_content') else "No transcript available."
 
@@ -25,6 +25,10 @@ def load_youtube_video_info(url):
             info = "No result found."
     except NoTranscriptFound:
         info = "Transcript not found for this video."
+    except PytubeError as e:
+        info = f"Error loading video information: {str(e)}"
+    except Exception as e:
+        info = f"An unexpected error occurred: {str(e)}"
 
     return info