Update app.py

app.py

@@ -15,6 +15,7 @@ from langchain_core.documents import Document
 from huggingface_hub import InferenceClient
 import inspect
 import logging
+import shutil
 
 
 # Set up basic configuration for logging
@@ -116,11 +117,7 @@ def update_vectors(files, parser):
         if os.path.exists("faiss_database"):
             logging.info("Updating existing FAISS database")
             database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
-            if len(database.docstore._dict) == 0:
-                logging.info("Existing database is empty. Creating new FAISS database")
-                database = FAISS.from_documents(all_data, embed)
-            else:
-                database.add_documents(all_data)
+            database.add_documents(all_data)
         else:
             logging.info("Creating new FAISS database")
             database = FAISS.from_documents(all_data, embed)
@@ -159,14 +156,14 @@ def delete_documents(selected_docs):
     logging.info(f"Documents to delete: {len(deleted_docs)}")
     
     if not docs_to_keep:
-        # If all documents are deleted, remove the FAISS database file
+        # If all documents are deleted, remove the FAISS database directory
         if os.path.exists("faiss_database"):
-            os.remove("faiss_database")
-        logging.info("All documents deleted. Removed FAISS database file.")
+            shutil.rmtree("faiss_database")
+        logging.info("All documents deleted. Removed FAISS database directory.")
     else:
         # Create new FAISS index with remaining documents
-        database = FAISS.from_documents(docs_to_keep, embed)
-        database.save_local("faiss_database")
+        new_database = FAISS.from_documents(docs_to_keep, embed)
+        new_database.save_local("faiss_database")
         logging.info(f"Created new FAISS index with {len(docs_to_keep)} documents.")
     
     # Update uploaded_documents list