Update app.py

app.py

@@ -35,7 +35,7 @@ app.add_middleware(
 UPLOAD_DIR = "uploads"
 os.makedirs(UPLOAD_DIR, exist_ok=True)
 
-persist_directory = "./chroma_db"
+persist_directory = "/home/user/.cache/chroma_db"
 
 load_dotenv()
 os.environ["HF_TOKEN"] = os.getenv("HF_TOKEN")
@@ -64,7 +64,8 @@ def generate_file_id(file_path):
 
 def delete_existing_embedding(file_id):
     if os.path.exists(persist_directory):
-        shutil.rmtree(persist_directory)
+        shutil.rmtree(persist_directory, ignore_errors = True)
+        os.makedirs(persist_directory, exist_ok=True)
 
 def tempUploadFile(filePath,file):
     with open(filePath,'wb') as buffer:
@@ -111,7 +112,7 @@ def loadAndSplitPdfFile(filePath):
 
 def prepare_retriever(filePath = "", load_from_chromadb = False):
     if load_from_chromadb:
-        vector_store = Chroma(persist_directory=persist_directory, embedding_function = embeddings)
+        vector_store = Chroma(persist_directory=persist_directory, embedding_function = embeddings, client_settings={"allow_reset": True})
         return vector_store.as_retriever(search_kwargs={"k": 5})    
     elif filePath:   
         doc_chunks = loadAndSplitPdfFile(filePath)        
@@ -125,7 +126,7 @@ def prepare_retriever(filePath = "", load_from_chromadb = False):
                     if isinstance(value, (str, int, float, bool, Path))
                 }
     
-        vector_store = Chroma.from_documents(documents= doc_chunks, persist_directory=persist_directory, embedding= embeddings)
+        vector_store = Chroma.from_documents(documents= doc_chunks, persist_directory=persist_directory, embedding= embeddings, read)
         vector_store.persist()
 
 def get_retriever_chain(retriever):