Update ydl options

app/services/youtube_service.py

@@ -1,14 +1,18 @@
-import asyncio
 from urllib.parse import urlparse, parse_qs
-from pathlib import Path
 import yt_dlp
 from ..core.config import settings
+import random
 
 
 class YouTubeService:
     def __init__(self):
         self.output_dir = settings.AUDIO_DIR
-        # Configure yt-dlp with robust but simple defaults
+        self.user_agents = [
+            "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+            "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.0 Safari/605.1.15",
+            "Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+            "Mozilla/5.0 (iPhone; CPU iPhone OS 16_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.0 Mobile/15E148 Safari/604.1",
+        ]
         self.ydl_opts = {
             "format": "bestaudio[ext=m4a]/bestaudio/best",
             "postprocessors": [
@@ -24,15 +28,15 @@ class YouTubeService:
             "ignoreerrors": False,
             "no_color": True,
             "geo_bypass": True,
-            "sleep_interval": 1,
-            "max_sleep_interval": 5,
-            "retries": 3,
-            "fragment_retries": 3,
+            "sleep_interval": 5,
+            "max_sleep_interval": 10,
+            "retries": 5,
+            "fragment_retries": 5,
             "skip_unavailable_fragments": True,
             "keepvideo": False,
             "overwrites": True,
             "http_headers": {
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
+                "User-Agent": random.choice(self.user_agents),
                 "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8",
                 "Accept-Language": "en-us,en;q=0.5",
                 "Sec-Fetch-Mode": "navigate",
@@ -67,41 +71,37 @@ class YouTubeService:
     async def convert_to_mp3(self, url: str) -> str:
         """Convert YouTube video to MP3 and return the filename"""
         try:
-            # Extract video ID and validate URL
             video_id = self._extract_video_id(url)
             filename = f"{video_id}.mp3"
             output_path = self.output_dir / filename
 
-            # Skip if file already exists
             if output_path.exists():
                 return filename
 
-            # Set output path for this specific download
             opts = dict(self.ydl_opts)
             opts["outtmpl"] = str(output_path.with_suffix(""))
+            opts["http_headers"]["User-Agent"] = random.choice(
+                self.user_agents
+            )  # Rotate user agents
 
-            # Try download with fallback
             try:
                 with yt_dlp.YoutubeDL(opts) as ydl:
                     ydl.download([url])
             except Exception as e:
-                # If first attempt fails, try with lower quality
+                print(f"First attempt failed: {str(e)}")
                 opts["format"] = "worstaudio/worst"
                 with yt_dlp.YoutubeDL(opts) as ydl:
                     ydl.download([url])
 
-            # Verify the file was created
             if not output_path.exists():
                 raise Exception("Failed to download and convert the video")
 
             return filename
 
         except ValueError as e:
-            # URL validation errors
             print(f"URL validation error: {str(e)}")
             raise
         except Exception as e:
-            # All other errors
             print(f"Error in convert_to_mp3: {str(e)}")
             raise Exception(f"Error converting video: {str(e)}")