Update app.py

app.py

@@ -140,10 +140,10 @@ def delete_documents(selected_docs):
     deleted_docs = []
     docs_to_keep = []
     for doc in database.docstore._dict.values():
-        if doc.metadata["source"] not in selected_docs:
+        if doc.metadata.get("source") not in selected_docs:
             docs_to_keep.append(doc)
         else:
-            deleted_docs.append(doc.metadata["source"])
+            deleted_docs.append(doc.metadata.get("source", "Unknown"))
     
     # Print debugging information
     print(f"Total documents before deletion: {len(database.docstore._dict)}")
@@ -153,9 +153,11 @@ def delete_documents(selected_docs):
     if not docs_to_keep:
         # If all documents are deleted, create an empty FAISS index
         database = FAISS.from_texts([""], embed)
+        logging.info("All documents deleted. Created an empty FAISS index.")
     else:
         # Create new FAISS index with remaining documents
         database = FAISS.from_documents(docs_to_keep, embed)
+        logging.info(f"Created new FAISS index with {len(docs_to_keep)} documents.")
     
     # Save updated FAISS database
     database.save_local("faiss_database")