change downloading

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import requests
-from yt_dlp import YoutubeDL
+from pytube import YouTube  # Import pytube
 import os
 import tempfile
 from typing import Optional
@@ -14,29 +14,15 @@ def download_youtube_audio(url: str, output_dir: Optional[str] = None) -> str:
     if output_dir is None:
         output_dir = tempfile.gettempdir()
 
-    # Updated command to use a different method for downloading
-    yt_dlp_command = [
-        "yt-dlp",
-        "-f", "bestaudio/best",
-        "-x",  # Extract audio
-        "--audio-format", "mp3",
-        "-o", os.path.join(output_dir, "%(title)s.%(ext)s"),  # Changed output filename to use title
-        url
-    ]
-
     try:
-        print(' '.join(yt_dlp_command))
-        result = subprocess.run(yt_dlp_command, capture_output=True, text=True, check=True)
-        
-        # Get the output filepath from the last line of stdout
-        audio_file = os.path.join(output_dir, f"{result.stdout.splitlines()[-1].strip()}.mp3")  # Adjusted to match new output
+        yt = YouTube(url)  # Create a YouTube object
+        audio_stream = yt.streams.filter(only_audio=True).first()  # Get the first audio stream
+        audio_file = audio_stream.download(output_path=output_dir, filename=f"{yt.title}.mp3")  # Download audio
 
         print(f"Successfully downloaded: {audio_file}")
         return audio_file
-    except subprocess.CalledProcessError as e:
-        raise Exception(f"Error downloading YouTube audio: {e.stderr}")
     except Exception as e:
-        raise Exception(f"Unexpected error: {str(e)}")
+        raise Exception(f"Error downloading YouTube audio: {str(e)}")
 
 
 def run_asr(audio_file, youtube_url):

requirements.txt

@@ -1,2 +1 @@
-yt-dlp
-
+pytube