Update app.py

app.py

@@ -40,34 +40,25 @@ def initialize_agent():
 # Initialize the agent
 multimodal_Agent = initialize_agent()
 
-# Function to download YouTube video using pytube
-def download_youtube_video(youtube_url):
+# Function to get YouTube video captions using pytube
+def get_youtube_captions(youtube_url):
     """
-    Downloads a YouTube video using pytube.
+    Retrieves YouTube video captions using pytube.
     
     Parameters:
     - youtube_url: The URL of the YouTube video.
     
     Returns:
-    - The path to the downloaded video file.
+    - The captions of the video in SRT format, or a message if captions are not available.
     """
-    # Initialize YouTube object using pytube
     yt = YouTube(youtube_url)
 
-    # Get the highest resolution stream available
-    video_stream = yt.streams.get_highest_resolution()
-
-    # Temporary file for the video
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_video:
-        temp_video_path = temp_video.name
-        
-    try:
-        # Download the video to the temporary file
-        video_stream.download(output_path=temp_video_path)
-
-        return temp_video_path  # Return the video file path
-    except Exception as e:
-        raise RuntimeError(f"An error occurred while downloading the video: {str(e)}")
+    # Check if captions are available in English (you can modify this language)
+    if 'en' in yt.captions:
+        caption = yt.captions['en']
+        return caption.generate_srt_captions()  # Return captions in SRT format
+    else:
+        return "No captions available for this video."
 
 # YouTube video URL input
 youtube_url = st.text_input(
@@ -78,12 +69,13 @@ youtube_url = st.text_input(
 
 if youtube_url:
     try:
-        with st.spinner("Downloading and processing the YouTube video..."):
-            # Download video using pytube
-            video_path = download_youtube_video(youtube_url)
+        with st.spinner("Fetching captions from the YouTube video..."):
+            # Get captions for the video
+            captions = get_youtube_captions(youtube_url)
 
-        # Display video
-        st.video(video_path, format="video/mp4", start_time=0)
+        # Display captions
+        st.subheader("Video Captions (SRT format)")
+        st.text(captions)
 
         # User query input
         user_query = st.text_area(
@@ -97,25 +89,19 @@ if youtube_url:
                 st.warning("Please enter a question or insight to analyze the video.")
             else:
                 try:
-                    with st.spinner("Analyzing video and gathering insights..."):
-                        # Upload and process video file
-                        processed_video = upload_file(video_path)
-                        while processed_video.state.name == "PROCESSING":
-                            time.sleep(1)
-                            processed_video = get_file(processed_video.name)
-
-                        # Prompt generation for analysis
+                    with st.spinner("Analyzing captions and gathering insights..."):
+                        # Generate the analysis prompt using captions
                         analysis_prompt = (
                             f"""
-                            Analyze the content of the uploaded YouTube video.
-                            Respond to the following query using video insights and supplementary web research:
+                            Analyze the content of the following YouTube video captions.
+                            Respond to the following query using the video captions and supplementary web research:
                             {user_query}
                             Provide a detailed, user-friendly, and actionable response.
                             """
                         )
 
                         # AI agent processing
-                        response = multimodal_Agent.run(analysis_prompt, videos=[processed_video])
+                        response = multimodal_Agent.run(analysis_prompt, video_captions=captions)
 
                     # Display the result
                     st.subheader("Analysis Result")
@@ -123,10 +109,8 @@ if youtube_url:
 
                 except Exception as error:
                     st.error(f"An error occurred during analysis: {error}")
-                finally:
-                    # Clean up temporary video file
-                    Path(video_path).unlink(missing_ok=True)
+
     except Exception as e:
-        st.error(f"An error occurred while downloading the video: {e}")
+        st.error(f"An error occurred while fetching captions: {e}")
 else:
     st.info("Enter a YouTube video link to begin analysis.")