fix

Dockerfile

@@ -22,8 +22,6 @@ RUN curl -fsSL https://ollama.com/install.sh | sh
 
 USER docker
 
-RUN mkdir ~/wtp_be_files/
-
 RUN nohup ollama serve & sleep 5
 
 #

app/main.py

@@ -39,6 +39,7 @@ def upload(files: list[UploadFile]):
     session_messages = []
     
     try:
+        os.mkdir("~/wtp_be_files/")
         for file in files:
             try:
                 file.file.seek(0)
@@ -48,7 +49,7 @@ def upload(files: list[UploadFile]):
                 file.file.close()
     finally:
         session_assistant.ingest("~/wtp_be_files/")
-        os.remove("~/wtp_be_files/")
+        shutil.rmtree("~/wtp_be_files/")
 
     return "Files inserted!"