Update app.py

app.py

@@ -99,6 +99,9 @@ def save_documents(documents):
 uploaded_documents = load_documents()
 
 # Modify the update_vectors function
+from langchain.vectorstores import FAISS
+import faiss
+
 def update_vectors(files, parser):
     global uploaded_documents
     logging.info(f"Entering update_vectors with {len(files)} files and parser: {parser}")
@@ -139,14 +142,14 @@ 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)
-            initial_size = len(database.index)
+            initial_size = database.index.ntotal
             database.add_documents(all_data)
-            final_size = len(database.index)
+            final_size = database.index.ntotal
             logging.info(f"FAISS database updated. Initial size: {initial_size}, Final size: {final_size}")
         else:
             logging.info("Creating new FAISS database")
             database = FAISS.from_documents(all_data, embed)
-            logging.info(f"New FAISS database created with {len(database.index)} vectors")
+            logging.info(f"New FAISS database created with {database.index.ntotal} vectors")
         
         database.save_local("faiss_database")
         logging.info("FAISS database saved")