Update app.py

app.py

@@ -135,15 +135,22 @@ def initialize_vector_store(documents, persist_dir):
     """
     Initialize the vector store with the provided documents.
     """
-    embeddings = HuggingFaceEmbeddings()
-    vectordb = Chroma.from_documents(
-        documents=documents, 
-        embedding_function=embeddings.embed_query,  # Use .embed_query explicitly
-        persist_directory=persist_dir
-    )
-    vectordb.persist()  # Persist the vector store to disk
-    return vectordb
+    try:
+        # Initialize HuggingFace embeddings
+        embeddings = HuggingFaceEmbeddings()
+
+        # Create a Chroma vector store with the embeddings
+        vectordb = Chroma.from_documents(
+            documents=documents, 
+            embedding=embeddings,  # Pass embeddings directly
+            persist_directory=persist_dir
+        )
+        vectordb.persist()  # Persist the vector store to disk
+        return vectordb
 
+    except Exception as e:
+        st.error(f"Failed to initialize the vector store: {e}")
+        st.stop()
 
 def create_retriever(vectordb):
     """