add self hosted docs

Author: MrLYC

app/main.py

@@ -4,9 +4,12 @@
 from typing import List
 from dataclasses import dataclass
 import time
+from starlette.staticfiles import StaticFiles
+from fastapi.openapi.docs import get_swagger_ui_html
 
+app = FastAPI(docs_url=None, redoc_url=None)
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
-app = FastAPI()
 new_model = DdddOcr(show_ad=False)
 old_model = DdddOcr(old=True, show_ad=False)
 
@@ -47,5 +50,16 @@ async def ocr_v1(files: List[UploadFile] = File(...)) -> List[ClassificationResu
     return await helper.classification()
 
 
+@app.get("/docs", include_in_schema=False)
+async def custom_swagger_ui_html():
+    return get_swagger_ui_html(
+        openapi_url=app.openapi_url,
+        title="OCR API",
+        oauth2_redirect_url=app.swagger_ui_oauth2_redirect_url,
+        swagger_js_url="/static/swagger-ui-bundle.js",
+        swagger_css_url="/static/swagger-ui.css",
+    )
+
+
 if __name__ == "__main__":
     uvicorn.run("main:app", host="0.0.0.0", port=8080, reload=False, debug=False)

poetry.lock

@@ -10,6 +10,8 @@ fastapi = "^0.45.0"
 uvicorn = "^0.11.1"
 ddddocr = "^1.3.1"
 python-multipart = "^0.0.5"
+starlette = "0.12.9"
+aiofiles = "^0.8.0"
 
 [tool.poetry.dev-dependencies]
 black = {version = "^19.10b0", allow-prereleases = true}