| | from fastapi import FastAPI, Query |
| | from fastapi.responses import JSONResponse |
| | import subprocess |
| | import json |
| |
|
| | app = FastAPI() |
| |
|
| | @app.get("/") |
| | def root(): |
| | return {"message": "Welcome to yt-dlp API (FastAPI)"} |
| |
|
| | @app.get("/download") |
| | def download_info(url: str = Query(..., description="Video URL")): |
| | try: |
| | result = subprocess.run( |
| | ["yt-dlp", "-J", url], |
| | capture_output=True, text=True, timeout=20 |
| | ) |
| | if result.returncode != 0: |
| | return JSONResponse(status_code=500, content={"error": result.stderr}) |
| |
|
| | data = json.loads(result.stdout) |
| | return { |
| | "title": data.get("title"), |
| | "uploader": data.get("uploader"), |
| | "duration": data.get("duration"), |
| | "formats": [ |
| | { |
| | "format_id": f["format_id"], |
| | "ext": f["ext"], |
| | "resolution": f.get("resolution") or f"{f.get('height')}p", |
| | "url": f["url"] |
| | } |
| | for f in data.get("formats", []) |
| | ] |
| | } |
| | except Exception as e: |
| | return JSONResponse(status_code=500, content={"error": str(e)}) |
