from fastapi import FastAPI
from fastapi.responses import JSONResponse
from fetchYoutubeSubtitle import fetchYoutubeSubtitle

app = FastAPI()

@app.get("/")
def read_root():
    return {"Hello": "World!"}


@app.get("/json")
def read_json():
    return JSONResponse(content={"Hello": "World!"})


@app.get("/subtitle/")
async def get_subtitle(vid: str):
    subtitle = await fetchYoutubeSubtitle(vid)
    return JSONResponse(content=subtitle)