Add api check.

Dockerfile

@@ -1,4 +1,4 @@
-FROM python:3.9
+FROM python:3.10
 
 WORKDIR /code
 

main.py

@@ -1,11 +1,17 @@
-from fastapi import FastAPI
+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("APP_KEY")
 app = FastAPI()
 
 @app.get("/")
-def read_root():
+def read_root(request:Request):
+    print(request.headers)
+    print(request.client.host)
+    print(request.client.port)
     return {"Hello": "World!"}
 
 
@@ -15,12 +21,16 @@ def read_json():
 
 
 @app.get("/subtitle/")
-async def get_subtitle(url: str, subtype: str="srt"):
+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):
+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)

requirements.txt

@@ -1,7 +1,7 @@
-fastapi==0.74.*
-requests==2.27.*
-sentencepiece==0.1.*
-torch==1.11.*
-transformers==4.*
+fastapi==0.95.*
+# requests==2.27.*
+# sentencepiece==0.1.*
+# torch==1.11.*
+# transformers==4.*
 uvicorn[standard]==0.17.*
 yt-dlp