Update main.py

main.py

@@ -1,4 +1,4 @@
-from starlette.responses import PlainTextResponse, JSONResponse
+from starlette.responses import PlainTextResponse, JSONResponse, FileResponse
 from starlette.applications import Starlette
 from starlette.routing import Route
 from starlette.middleware import Middleware
@@ -697,6 +697,14 @@ async def validate_similarity(request):
     sim = 1 - cosine(vect1, vect2) >= 0.8
     return JSONResponse({"isSimilar": str(sim)})
 
+async def get_cached_img_from_disk(request):
+    _fn = request.path_params['fn']
+    # /images/img_-3711971785602203114.webp HTTP/1.1"
+    if _fn.startswith('img_') and _fn.endswith('.webp'):
+        return FileResponse(_fn)
+    else:
+        raise HTTPException(404)
+
 async def root(requests):
     return PlainTextResponse("Success")
 
@@ -722,6 +730,8 @@ app = Starlette(debug=True,routes=[
     Route('/temp/{id}', __remove_temp, methods=['DELETE']),
     # Route('/generateQuiz/{id}', generate_questions, methods=['GET']),
     Route('/generateQuiz/{id}/{lang}', generate_questions, methods=['GET']),
+    # /images/img_-3711971785602203114.webp HTTP/1.1"
+    Route('/images/{fn}', get_cached_img_from_disk, methods=['GET']),
     Route('/convert2md', convert2md, methods=['POST']),
     Route('/mltest', __mltest, methods=['GET']),
     Route('/validateSimilarity', validate_similarity, methods=['POST']),