Update modules/app.py

modules/app.py

@@ -8,10 +8,14 @@
 from fastapi import FastAPI, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse, StreamingResponse
+import uvicorn
+
 import ctranslate2
 import sentencepiece as spm
 import fasttext
-import uvicorn
+
 import pytz
 from datetime import datetime
 import os
@@ -31,7 +35,7 @@ app.add_middleware(
 fasttext.FastText.eprint = lambda x: None
 
 # User interface
-templates_folder = os.path.join(os.path.dirname(__file__), "templates")
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
 
 # Get time of request
@@ -125,10 +129,11 @@ def translate_enter(userinput: str, source_lang: str, target_lang: str):
   return translations_desubword[0]
 
     
-@app.get("/", response_class=HTMLResponse)
-async def read_root(request: Request):
-    return HTMLResponse(content=open(os.path.join(templates_folder, "translator.html"), "r").read(), status_code=200)
-
+@app.head("/")
+@app.get("/")
+def index() -> FileResponse:
+    return FileResponse(path="static/index.html", media_type="text/html")
+    
 @app.post("/translate_detect/")
 async def translate_detect_endpoint(request: Request):
     datad = await request.json()