Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from pytube import YouTube
+import yt_dlp
 import os
 import re
 
@@ -13,29 +13,41 @@ def download_audio(url, output_dir="downloads"):
         if not os.path.exists(output_dir):
             os.makedirs(output_dir)
         
-        # Create YouTube object
-        yt = YouTube(url)
-        
-        # Get audio stream
-        audio_stream = yt.streams.filter(only_audio=True).first()
-        
-        # Sanitize the title for filename
-        safe_title = sanitize_filename(yt.title)
-        
-        # Download the audio
-        output_file = audio_stream.download(
-            output_path=output_dir,
-            filename=f"{safe_title}.mp3"
-        )
-        
-        return {
-            "status": "success",
-            "message": f"Successfully downloaded: {safe_title}",
-            "title": yt.title,
-            "author": yt.author,
-            "duration": f"{yt.length // 60}:{yt.length % 60:02d}",
-            "file_path": output_file
+        # Configure yt-dlp options
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'mp3',
+                'preferredquality': '192',
+            }],
+            'outtmpl': os.path.join(output_dir, '%(title)s.%(ext)s'),
+            'verbose': True
         }
+        
+        # Download with yt-dlp
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            # Get video info first
+            info = ydl.extract_info(url, download=False)
+            title = info.get('title', 'video')
+            duration = info.get('duration', 0)
+            author = info.get('uploader', 'Unknown')
+            
+            # Download the video
+            ydl.download([url])
+            
+            # Construct the output filepath
+            safe_title = sanitize_filename(title)
+            output_file = os.path.join(output_dir, f"{safe_title}.mp3")
+            
+            return {
+                "status": "success",
+                "message": f"Successfully downloaded: {safe_title}",
+                "title": title,
+                "author": author,
+                "duration": f"{duration // 60}:{duration % 60:02d}",
+                "file_path": output_file
+            }
     
     except Exception as e:
         return {