Update app.py

app.py

@@ -1,11 +1,12 @@
 from fastapi import FastAPI, HTTPException, Query
 from fastapi.responses import RedirectResponse
 import subprocess
+import urllib.parse
 
 app = FastAPI(
     title="YouTube Audio Streamer",
     description="Stream best audio of a YouTube video using yt-dlp.",
-    version="1.0.1",
+    version="1.0.2",
     docs_url="/docs",
     redoc_url="/redoc"
 )
@@ -16,14 +17,22 @@ async def root():
 
 @app.get("/stream", tags=["Stream"])
 async def stream_audio(url: str = Query(..., description="Full YouTube video URL")):
-    if "youtube.com/watch?v=" not in url and "youtu.be/" not in url:
+    decoded_url = urllib.parse.unquote(url)
+
+    if "youtube.com/watch?v=" not in decoded_url and "youtu.be/" not in decoded_url:
         raise HTTPException(status_code=400, detail="Invalid YouTube URL.")
 
+    # Remove URL query params (yt-dlp can fail with things like ?si=xxx)
+    base_url = decoded_url.split("&")[0].split("?")[0]
+
     try:
         result = subprocess.run(
-            ["yt-dlp", "-f", "bestaudio", "--get-url", url],
+            ["yt-dlp", "-f", "bestaudio", "--get-url", base_url],
             capture_output=True, text=True
         )
+        if result.returncode != 0:
+            raise HTTPException(status_code=500, detail=f"yt-dlp error: {result.stderr.strip()}")
+
         audio_url = result.stdout.strip()
         if not audio_url:
             raise HTTPException(status_code=404, detail="No audio URL found.")