Update app.py

app.py

@@ -138,20 +138,30 @@ def delete_documents(selected_docs):
     database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
     
     deleted_docs = []
-    for doc_name in selected_docs:
-        # Remove from uploaded_documents list
-        uploaded_documents = [doc for doc in uploaded_documents if doc["name"] != doc_name]
-        
-        # Remove from FAISS database
-        docs_to_keep = [doc for doc in database.docstore._dict.values() if doc.metadata["source"] != doc_name]
+    docs_to_keep = []
+    for doc in database.docstore._dict.values():
+        if doc.metadata["source"] not in selected_docs:
+            docs_to_keep.append(doc)
+        else:
+            deleted_docs.append(doc.metadata["source"])
+    
+    # Print debugging information
+    print(f"Total documents before deletion: {len(database.docstore._dict)}")
+    print(f"Documents to keep: {len(docs_to_keep)}")
+    print(f"Documents to delete: {len(deleted_docs)}")
+    
+    if not docs_to_keep:
+        # If all documents are deleted, create an empty FAISS index
+        database = FAISS.from_texts([""], embed)
+    else:
+        # Create new FAISS index with remaining documents
         database = FAISS.from_documents(docs_to_keep, embed)
-        
-        deleted_docs.append(doc_name)
     
     # Save updated FAISS database
     database.save_local("faiss_database")
     
-    # Save updated uploaded_documents list
+    # Update uploaded_documents list
+    uploaded_documents = [doc for doc in uploaded_documents if doc["name"] not in deleted_docs]
     save_documents(uploaded_documents)
     
     return f"Deleted documents: {', '.join(deleted_docs)}", display_documents()