Update main.py

main.py

@@ -33,6 +33,8 @@ from random import randint, sample
 from enum import Enum
 from re import sub
 
+import redis.asyncio as redis
+r = redis.Redis()
 
 import logging
 
@@ -51,7 +53,7 @@ class QType(Enum):
 
 routes = ...
 
-temp_files = {}
+
 
 middleware = [
     Middleware(
@@ -67,28 +69,46 @@ sys_random = SystemRandom()
 
 #pytesseract.pytesseract.tesseract_cmd = r"C:\Users\Admin\AppData\Local\Programs\Tesseract-OCR\tesseract.exe"
 
+async def __internal_tmp_w(id, content:any):
+    async with r.pipeline(transaction=True) as pipeline:
+        ok = await pipeline.set(id, json.dumps(content).encode("utf-8")).execute()
+    return ok
+
+async def __internal_tmp_r(id):
+    async with r.pipeline(transaction=True) as pipeline:
+        res = await (pipeline.get(id).execute())
+        if res[-1] == None:
+            return [None, None, None]
+        res = res[-1].decode("utf-8")
+    return json.loads(res)
+
+async def __internal_tmp_d(id):
+    async with r.pipeline(transaction=True) as pipeline:
+        res = await (pipeline.del(id).execute())
 
 async def __mltest(request):
     pass
 
 async def __save_temp(request):
-    file_id = sys_random.randbytes(16).hex()
+    file_id = sys_random.randbytes(20).hex()
     content = ""
     # async with request.form(max_fields=3) as form:
     form = await request.json()
     content = form['content']
     title = form['title']
     keywords = form['keywords']
-    temp_files[file_id] = [title, content, keywords]
+
+    await __internal_tmp_w(file_id, [title, content, keywords])
+    
     print(file_id)
     return PlainTextResponse(file_id, 200)
     
 async def __get_temp(request, entry = 1):
-    return JSONResponse(temp_files.get(request.path_params['id'], [None, None, None]))
+    return JSONResponse(await __internal_tmp_r(request.path_params['id']))
 
 async def __remove_temp(request):
     try:
-        del temp_files[request.path_params['id']]
+        __internal_tmp_d(request.path_params['id'])
     except:
         return PlainTextResponse("", 500)
     
@@ -362,7 +382,8 @@ async def generate_questions(request):
 
     # parse paragraphs from document file
     try:
-        __ps = await __parse_paragraphs(temp_files[request.path_params['id']][1],batching=True)
+        __cont = await __internal_tmp_r(requests.path_params['id'])
+        __ps = await __parse_paragraphs(__cont[1], batching=True)
     except:
         return JSONResponse({}, 500)
 
@@ -495,9 +516,12 @@ async def convert2html(request):
 
 async def get_flashcards(request):
     # [title, content, keywords]
-    __content = temp_files[request.path_params['id']][1]
+
+    __file = await __internal_tmp_r(requests.path_params['id'])
+    __content = __file[1]
+    
     __lang = request.path_params['lang']
-    __keywords = [r.strip() for r in temp_files[request.path_params['id']][2] if len(r) > 0]
+    __keywords = [r.strip() for r in __file[2] if len(r) > 0]
 
     __tldr = await tldr(__content, __lang)
     print(__tldr)