Add application file

app.py

@@ -5,14 +5,14 @@ import threading
 import tempfile
 import ctypes
 import gc
-from typing import Optional
-from fastapi import FastAPI, Depends, HTTPException
+from fastapi import FastAPI, Depends, HTTPException, Security
 from fastapi.responses import FileResponse, JSONResponse
 from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
+from fastapi.security.api_key import APIKeyHeader
+from fastapi.openapi.docs import get_swagger_ui_html, get_redoc_html
 from pydantic import BaseModel
 from huggingface_hub import hf_hub_download
 from torch import no_grad, package
-from pydub import AudioSegment
 import uvicorn
 from accentor import accentification, stress_replace_and_shift
 import argparse
@@ -133,6 +133,28 @@ class ArgParser(argparse.ArgumentParser):
         )
         self.args = self.parse_args(sys.argv[1:])
 
+
+api_key_header = APIKeyHeader(name="access_token", auto_error=False)
+
+def getAPIKey(api_key_header: str = Security(api_key_header)):
+    if api_key_header == api_key:
+        return api_key_header
+    raise HTTPException(
+        status_code=403,
+        detail="Could not validate credentials",
+    )
+
+
+
+@app.get("/docs", include_in_schema=False)
+async def get_docs(api_key: str = Depends(getAPIKey)):
+    return get_swagger_ui_html(openapi_url=app.openapi_url, title="docs")
+
+@app.get("/redoc", include_in_schema=False)
+async def get_redoc(api_key: str = Depends(getAPIKey)):
+    return get_redoc_html(openapi_url=app.openapi_url, title="docs")
+
+
 if __name__ == "__main__":
     args = ArgParser().args
     uvicorn.run(app, host=args.server, port=args.port, reload=False)