Prompt tuning

src/api.py

@@ -4,20 +4,21 @@ from fastapi.responses import HTMLResponse
 from fastapi.middleware.cors import CORSMiddleware
 from eventlog_handler import Event_handler 
 from data_handler import Data_handler
-import aiofiles
 import json
 import os
 from pydantic.v1 import BaseModel
+import shutil
+from pathlib import Path
+from tempfile import NamedTemporaryFile
 
 class Item(BaseModel):
     toDelete:list
 
-root_path:str = os.getenv("ROOT_PATH","/app/data")
-tmp_path:str = "/tmpfs"
-if not os.path.exists(tmp_path):
-    os.makedirs(tmp_path)
-for file in os.listdir(tmp_path):
-    os.remove(tmp_path+"/"+file)
+root_path:str = "/app/data"
+#if not os.path.exists(tmp_path):
+#    os.makedirs(tmp_path)
+#for file in os.listdir(tmp_path):
+#    os.remove(tmp_path+"/"+file)
 
 data_handler:Data_handler = Data_handler(root_path)
 
@@ -53,11 +54,15 @@ async def get_columns(name: str):
 
 @app.post("/api/upload_file/{name}")
 async def upload_file(name:str,cleanup: bool,file: UploadFile):
-    filepath="{}/{}-{}".format(tmp_path,name,file.filename)
-    async with aiofiles.open(filepath, 'wb') as out_file:
-        while content := await file.read(1024):
-            await out_file.write(content)
-    data_handler.upload_file(name,filepath,cleanup)
+    try:
+        suffix = Path(file.filename).suffix
+        with NamedTemporaryFile(delete=False, suffix=suffix) as tmp:
+            shutil.copyfileobj(file.file, tmp)
+            tmp_path = Path(tmp.name)
+    finally:
+        file.file.close()
+    print(tmp_path)
+    data_handler.upload_file(name,str(tmp_path),cleanup)
     return {"status":"OK"}
 
 @app.post("/api/select_columns/{name}")