Update app.py

app.py

@@ -419,22 +419,29 @@ def get_trust_tip_and_suggestion():
 
 
 def load_main_data_source():
+    """
+    Load the main data source and return a list of Document objects.
+    """
     try:
         data_path = "./data_source/time_to_rethink_trust_book.md"
         if not os.path.exists(data_path):
-            st.error("Data source file not found.")
+            st.error(f"Data source file not found at: {data_path}")
             return []
 
         with open(data_path, "r") as f:
             main_content = f.read()
 
-        # Split main content into chunks
+        # Split the content into chunks
         text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
         main_texts = text_splitter.split_text(main_content)
-        st.write(f"Split into {len(main_texts)} chunks.")
 
         # Create Document objects
         main_documents = [Document(page_content=text) for text in main_texts]
+        if not main_documents:
+            st.error("No documents created. Data source may be empty.")
+            return []
+        
+        st.write(f"Loaded {len(main_documents)} documents from the data source.")
         return main_documents
     except Exception as e:
         st.error(f"Error loading main data source: {e}")
@@ -442,19 +449,26 @@ def load_main_data_source():
 
 
 
+
 def refresh_faiss_index(documents=None):
     """
     Refresh or initialize the FAISS index with updated documents.
     """
     try:
+        if documents is None or not isinstance(documents, list):
+            st.error("No valid documents provided to refresh the FAISS index.")
+            return
+
         embeddings = OpenAIEmbeddings()
 
         # Initialize FAISS if it doesn't exist
         if "faiss_db" not in st.session_state:
             st.session_state["faiss_db"] = FAISS.from_documents(documents, embeddings)
-        elif documents:
+            st.success("FAISS database initialized.")
+        else:
             # Add/Update specific documents in the index
             st.session_state["faiss_db"].add_documents(documents, embeddings)
+            st.success("Documents added to FAISS database.")
     except Exception as e:
         st.error(f"Error refreshing FAISS index: {e}")