Update app.py

app.py

@@ -1,15 +1,16 @@
 import streamlit as st
-from yt_dlp import YoutubeDL
+from pytube import YouTube
 from phi.agent import Agent
 from phi.model.google import Gemini
 from phi.tools.duckduckgo import DuckDuckGo
 from google.generativeai import upload_file, get_file
 import google.generativeai as genai
+
 import time
-import os
 from pathlib import Path
-import tempfile
 from dotenv import load_dotenv
+import os
+import tempfile
 
 # Load environment variables
 load_dotenv()
@@ -39,35 +40,34 @@ def initialize_agent():
 # Initialize the agent
 multimodal_Agent = initialize_agent()
 
-import tempfile
-from yt_dlp import YoutubeDL
-
-import browser_cookie3
-from yt_dlp import YoutubeDL
-
+# Function to download YouTube video using pytube
 def download_youtube_video(youtube_url):
-    # Extract cookies from browser (Chrome, Firefox, etc.)
-    cookies = browser_cookie3.load()
-
-    # yt-dlp options
-    ydl_opts = {
-        'format': 'bestvideo+bestaudio/best',  # Download best video and audio
-        'outtmpl': 'downloaded_video.mp4',      # Output file path
-        'cookies': cookies,                     # Use cookies directly from browser
-        'quiet': True,                          # Suppress yt-dlp output
-    }
-
+    """
+    Downloads a YouTube video using pytube.
+    
+    Parameters:
+    - youtube_url: The URL of the YouTube video.
+    
+    Returns:
+    - The path to the downloaded video file.
+    """
+    # 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
-        with YoutubeDL(ydl_opts) as ydl:
-            ydl.download([youtube_url])
-
-        print("Download successful!")
+        # 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:
-        print(f"An error occurred: {str(e)}")
-
-
+        raise RuntimeError(f"An error occurred while downloading the video: {str(e)}")
 
 # YouTube video URL input
 youtube_url = st.text_input(
@@ -79,7 +79,7 @@ youtube_url = st.text_input(
 if youtube_url:
     try:
         with st.spinner("Downloading and processing the YouTube video..."):
-            # Download video using yt-dlp
+            # Download video using pytube
             video_path = download_youtube_video(youtube_url)
 
         # Display video