Update neorvc/main_cli.py

neorvc/main_cli.py

@@ -60,39 +60,46 @@ def get_youtube_video_id(url: str, ignore_playlist: bool = True) -> str | None:
             return parsed.path.split("/")[-1]
     return None
 
-async def yt_download(link: str, cookies_path: str = os.path.join(BASE_DIR, "neorvc", "config.txt"), progress: Optional[Union[tqdm, 'GradioProgress']] = None) -> Path:
+async def yt_download(
+    url: str,
+    cookies_path: str = os.path.join(BASE_DIR, "neorvc", "config.txt"),
+    progress: Optional[Union['tqdm', 'GradioProgress']] = None
+) -> Path:
     if not os.path.exists(cookies_path):
         raise FileNotFoundError(f"Cookies file not found: {cookies_path}")
 
-    video_id = get_youtube_video_id(link)
+    video_id = get_youtube_video_id(url)
     if not video_id:
         raise ValueError("Invalid YouTube URL: could not extract video ID.")
 
-    output_file = os.path.join(OUTPUT_DIR, f"{video_id}.mp3")
+    output_file = os.path.join(OUTPUT_DIR, f"{video_id}.wav")
     if os.path.exists(output_file):
         return Path(output_file)
 
     handle_progress(progress, description="Downloading YouTube audio", value=10)
 
+    os.makedirs(OUTPUT_DIR, exist_ok=True)
+
     cmd = [
         "yt-dlp",
         "--format", "bestaudio/best",
         "--extract-audio",
-        "--audio-format", "mp3",
-        "--audio-quality", "192K",
+        "--audio-format", "wav",
+        "--audio-quality", "0",  # highest quality for wav
+        "--postprocessor-args", "-acodec pcm_f32le",
         "--cookies", str(cookies_path),
-        "--output", str(output_file),
+        "--output", output_file,
         "--no-check-certificate",
-        link
+        url
     ]
 
     process = await asyncio.create_subprocess_exec(
         *cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE
     )
-    await process.communicate()
+    stdout, stderr = await process.communicate()
 
     if process.returncode != 0:
-        raise RuntimeError("yt-dlp failed")
+        raise RuntimeError(f"yt-dlp failed: {stderr.decode().strip()}")
 
     if not os.path.exists(output_file):
         raise RuntimeError(f"Downloaded file not found: {output_file}")