Update app.py

app.py

@@ -1,11 +1,11 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Query
 from fastapi.responses import RedirectResponse
 import subprocess
 
 app = FastAPI(
     title="YouTube Audio Streamer",
     description="Stream best audio of a YouTube video using yt-dlp.",
-    version="1.0.0",
+    version="1.0.1",
     docs_url="/docs",
     redoc_url="/redoc"
 )
@@ -14,16 +14,19 @@ app = FastAPI(
 async def root():
     return {"status": "FastAPI working", "message": "YT-DLP backend ready"}
 
-@app.get("/stream/{vidid}", tags=["Stream"])
-async def stream_audio(vidid: str):
+@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:
+        raise HTTPException(status_code=400, detail="Invalid YouTube URL.")
+
     try:
         result = subprocess.run(
-            ["yt-dlp", "-f", "bestaudio", "--get-url", f"https://www.youtube.com/watch?v={vidid}"],
+            ["yt-dlp", "-f", "bestaudio", "--get-url", url],
             capture_output=True, text=True
         )
-        url = result.stdout.strip()
-        if not url:
+        audio_url = result.stdout.strip()
+        if not audio_url:
             raise HTTPException(status_code=404, detail="No audio URL found.")
-        return RedirectResponse(url)
+        return RedirectResponse(audio_url)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Error: {str(e)}")