Update app.py

app.py

@@ -1,28 +1,44 @@
 import gradio as gr
 from langchain_community.document_loaders import YoutubeLoader
-
-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
-
-
+    """
+    Load YouTube video information including metadata and transcript.
+    
+    Args:
+        url (str): URL of the YouTube video.
+        
+    Returns:
+        str: Information about the video, including metadata and transcript or error messages.
+    """
     try:
+        # Initialize YouTube loader with preferred language as Portuguese
         loader = YoutubeLoader.from_youtube_url(url, add_video_info=True, language="pt")
-
+        
         # Load the video information and transcript
         result = loader.load()
 
-        if result:
+        if result and len(result) > 0:
             # Retrieve metadata information
             author = result[0].metadata.get('author', 'Unknown Author')
             length = result[0].metadata.get('length', 'Unknown Length')
+            title = result[0].metadata.get('title', 'Unknown Title')
+            
             # Check if transcript is available
             transcript = result[0].page_content if hasattr(result[0], 'page_content') else "No transcript available."
 
+            # Format information
+            info = (
+                f"Video Title: {title}\n"
+                f"Author: {author}\n"
+                f"Length: {length}\n\n"
+                f"Transcript:\n{transcript}"
+            )
+        else:
             info = "No result found."
+
     except NoTranscriptFound:
         info = "Transcript not found for this video."
     except PytubeError as e:
@@ -30,4 +46,4 @@ def load_youtube_video_info(url):
     except Exception as e:
         info = f"An unexpected error occurred: {str(e)}"
 
-    return info
+    return info