Update app.py

app.py

@@ -45,16 +45,24 @@ embedding = HuggingFaceEmbeddings()
 
 # Load or create FAISS index
 index_path = "faiss_index.pkl"
+document_texts_path = "document_texts.pkl"
+
 if os.path.exists(index_path):
     with open(index_path, "rb") as f:
-        index, document_texts = pickle.load(f)
-        print("Loaded FAISS index and documents from faiss_index.pkl")
+        index = pickle.load(f)
+        print("Loaded FAISS index from faiss_index.pkl")
+    if os.path.exists(document_texts_path):
+        with open(document_texts_path, "rb") as f:
+            document_texts = pickle.load(f)
+            print("Loaded document texts from document_texts.pkl")
+    else:
+        document_texts = []
 else:
     # Create a new FAISS index if it doesn't exist
     index = faiss.IndexFlatL2(embedding_model.get_sentence_embedding_dimension())
     document_texts = []
     with open(index_path, "wb") as f:
-        pickle.dump((index, document_texts), f)
+        pickle.dump(index, f)
         print("Created new FAISS index and saved to faiss_index.pkl")
 
 def upload_files(files):
@@ -80,8 +88,11 @@ def upload_files(files):
 
     # Save the updated index and documents
     with open(index_path, "wb") as f:
-        pickle.dump((index, document_texts), f)
-        print("Saved updated FAISS index and documents to faiss_index.pkl")
+        pickle.dump(index, f)
+        print("Saved updated FAISS index to faiss_index.pkl")
+    with open(document_texts_path, "wb") as f:
+        pickle.dump(document_texts, f)
+        print("Saved updated document texts to document_texts.pkl")
 
     return "Files processed successfully"
 
@@ -119,6 +130,7 @@ with gr.Blocks() as demo:
 
 demo.launch()
 
+