Yt-api / app.py
dragxd's picture
Update app.py
7451bec verified
raw
history blame
971 Bytes
from fastapi import FastAPI, HTTPException
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",
docs_url="/docs",
redoc_url="/redoc"
)
@app.get("/", tags=["Health"])
async def root():
return {"status": "FastAPI working", "message": "YT-DLP backend ready"}
@app.get("/stream/{vidid}", tags=["Stream"])
async def stream_audio(vidid: str):
try:
result = subprocess.run(
["yt-dlp", "-f", "bestaudio", "--get-url", f"https://www.youtube.com/watch?v={vidid}"],
capture_output=True, text=True
)
url = result.stdout.strip()
if not url:
raise HTTPException(status_code=404, detail="No audio URL found.")
return RedirectResponse(url)
except Exception as e:
raise HTTPException(status_code=500, detail=f"Error: {str(e)}")