Update app.py

app.py

@@ -112,16 +112,24 @@ def update_vectors(files, parser):
     
     logging.info(f"Total chunks processed: {total_chunks}")
     
-    if os.path.exists("faiss_database"):
-        logging.info("Updating existing FAISS database")
-        database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
-        database.add_documents(all_data)
-    else:
-        logging.info("Creating new FAISS database")
-        database = FAISS.from_documents(all_data, embed)
-    
-    database.save_local("faiss_database")
-    logging.info("FAISS database saved")
+    try:
+        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)
+        else:
+            logging.info("Creating new FAISS database")
+            database = FAISS.from_documents(all_data, embed)
+        
+        database.save_local("faiss_database")
+        logging.info("FAISS database saved")
+    except Exception as e:
+        logging.error(f"Error updating FAISS database: {str(e)}")
+        return f"Error updating vector store: {str(e)}", display_documents()
 
     # Save the updated list of documents
     save_documents(uploaded_documents)
@@ -146,22 +154,21 @@ def delete_documents(selected_docs):
             deleted_docs.append(doc.metadata.get("source", "Unknown"))
     
     # 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)}")
+    logging.info(f"Total documents before deletion: {len(database.docstore._dict)}")
+    logging.info(f"Documents to keep: {len(docs_to_keep)}")
+    logging.info(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)
-        logging.info("All documents deleted. Created an empty FAISS index.")
+        # If all documents are deleted, remove the FAISS database file
+        if os.path.exists("faiss_database"):
+            os.remove("faiss_database")
+        logging.info("All documents deleted. Removed FAISS database file.")
     else:
         # Create new FAISS index with remaining documents
         database = FAISS.from_documents(docs_to_keep, embed)
+        database.save_local("faiss_database")
         logging.info(f"Created new FAISS index with {len(docs_to_keep)} documents.")
     
-    # Save updated FAISS database
-    database.save_local("faiss_database")
-    
     # Update uploaded_documents list
     uploaded_documents = [doc for doc in uploaded_documents if doc["name"] not in deleted_docs]
     save_documents(uploaded_documents)