Update main.py

app/main.py

@@ -1,10 +1,10 @@
 import os
 import logging
-import shutil
+import pathlib
 import time
 import re
 from typing import List
-from fastapi import FastAPI, UploadFile
+from fastapi import FastAPI, Request
 from fastapi.middleware import Middleware
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import StreamingResponse
@@ -22,6 +22,7 @@ middleware = [
 app = FastAPI(middleware=middleware)
 
 files_dir = os.path.expanduser("~/wtp_be_files/")
+os.makedirs(files_dir)
 session_assistant = ChatPDF()
 
 logging.basicConfig(level=logging.INFO)
@@ -52,22 +53,20 @@ async def process_input(text: str):
         generator = re.split(r'(\s)', message)
     return StreamingResponse(astreamer(generator), media_type='text/event-stream')
 
+async def parse_body(request: Request):
+    data: bytes = await request.body()
+    return data
 
 @app.post("/upload")
-def upload(files: list[UploadFile]):  
+def upload(data: bytes = Depends(parse_body)):
     try:
-        os.makedirs(files_dir)
-        for file in files:
-            try:
-                path = f"{files_dir}/{file.filename}"
-                file.file.seek(0)
-                with open(path, 'wb') as destination:
-                    shutil.copyfileobj(file.file, destination)
-            finally:
-                file.file.close()
-    finally:
+        path = f"{files_dir}/file"
+        with open(path, "wb") as f:
+            f.write(data)
         session_assistant.ingest(files_dir)
-        shutil.rmtree(files_dir)
+        pathlib.Path(path).unlink()
+    except Exception as e:
+        logging.error(traceback.format_exc())
 
     message = "Files inserted successfully."
     generator = re.split(r'(\s)', message)