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