| 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)}) |
