Update app.py

app.py

@@ -13,6 +13,7 @@ from langchain.llms import OpenAI
 from langchain.chains.question_answering import load_qa_chain
 from langchain.callbacks import get_openai_callback
 import os
+import traceback
 
 import pandas as pd
 import pydeck as pdk
@@ -43,10 +44,9 @@ api_key = os.getenv("OPENAI_API_KEY")
 
 
 # Updated caching mechanism using st.cache_data
-@st.cache_data(persist="disk")  # Using persist="disk" to save cache across sessions
+@st.cache_data(persist="disk")
 def load_vector_store(file_path, store_name, force_reload=False):
-
-        # Check if we need to force reload the vector store (e.g., when the PDF changes)
+    try:
         if force_reload or not os.path.exists(f"{store_name}.pkl"):
             text_splitter = RecursiveCharacterTextSplitter(
                 chunk_size=1000,
@@ -64,9 +64,15 @@ def load_vector_store(file_path, store_name, force_reload=False):
         else:
             with open(f"{store_name}.pkl", "rb") as f:
                 VectorStore = pickle.load(f)
-    
+        
         return VectorStore
 
+    except Exception as e:
+        st.error(f"An error occurred: {e}")
+        traceback.print_exc()  # This will print the full traceback to the console
+        return None
+
+        
 # Utility function to load text from a PDF
 def load_pdf_text(file_path):
     pdf_reader = PdfReader(file_path)