File size: 1,091 Bytes
5d322d8
 
093a866
5d322d8
c2eed51
093a866
4bfa0b3
093a866
 
 
5d322d8
 
 
 
093a866
 
 
 
 
 
 
 
 
5d322d8
 
 
d030b89
ba9fae4
 
 
 
5d322d8
 
 
ba9fae4
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import os
from fastapi import FastAPI, Header, Request
from fastapi.responses import JSONResponse
from typing import Annotated
from fetchYoutubeSubtitle import fetchSubtitle, fetchSubtitleUrls

token = os.getenv("HF_X_TOKEN")
app = FastAPI()

@app.get("/")
def read_root(request:Request):
    print(request.headers)
    print(request.client.host)
    print(request.client.port)
    return {"Hello": "World!"}


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


@app.get("/subtitle/")
async def get_subtitle(url: str, subtype: str="srt", x_token: Annotated[str | None, Header()] = None):
    if token != x_token:
        return JSONResponse({"error": "Invalid token"})
    subtitle = await fetchSubtitle(url,subType=subtype)
    return JSONResponse(content=subtitle)


@app.get("/subtitle-urls/")
async def get_subtitleUrls(url: str, x_token: Annotated[str | None, Header()] = None):
    if token != x_token:
        return JSONResponse({"error": "Invalid token"})
    subtitles = await fetchSubtitleUrls(url)
    return JSONResponse(content=subtitles)