Update app.py

app.py

@@ -133,11 +133,17 @@ def delete_files(files_to_delete):
         embed = get_embeddings()
         database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
         
-        # Remove documents from the database
-        database.delete(lambda doc: doc.metadata["source"] in deleted_files)
+        # Create a new list of documents, excluding the deleted ones
+        new_docs = []
+        for i, doc_id in enumerate(database.index_to_docstore_id.values()):
+            if database.docstore._dict[doc_id].metadata["source"] not in deleted_files:
+                new_docs.append(database.docstore._dict[doc_id])
+
+        # Create a new FAISS index with the remaining documents
+        new_db = FAISS.from_documents(new_docs, embed)
         
         # Save the updated database
-        database.save_local("faiss_database")
+        new_db.save_local("faiss_database")
 
     # Update the document selector
     updated_selector = gr.CheckboxGroup(