Update app.py

app.py

@@ -28,20 +28,20 @@ st.set_page_config(page_title="MBAL Chatbot", page_icon="🛡️", layout="wide"
 
 HF_TOKEN = st.secrets["HF_TOKEN"]
 
-# Initialize your models, databases, and other components here
-# @st.cache_resource
-# def init_chroma():
-#     persist_directory = "chroma_db"
-#     chroma_client = chromadb.PersistentClient(path=persist_directory)
-#     chroma_collection = chroma_client.get_or_create_collection("my_collection")
-#     return chroma_client, chroma_collection
-
-# @st.cache_resource
-# def init_vectorstore():
-#     persist_directory = "chroma_db"
-#     embeddings = HuggingFaceEmbeddings()
-#     vectorstore = Chroma(persist_directory=persist_directory, embedding_function=embeddings, collection_name="my_collection")
-#     return vectorstore
+Initialize your models, databases, and other components here
+@st.cache_resource
+def init_chroma():
+    persist_directory = "chroma_db"
+    chroma_client = chromadb.PersistentClient(path=persist_directory)
+    chroma_collection = chroma_client.get_or_create_collection("my_collection")
+    return chroma_client, chroma_collection
+
+@st.cache_resource
+def init_vectorstore():
+    persist_directory = "chroma_db"
+    embeddings = HuggingFaceEmbeddings()
+    vectorstore = Chroma(persist_directory=persist_directory, embedding_function=embeddings, collection_name="my_collection")
+    return vectorstore
 @st.cache_resource
 def setup_vector():
     # Đọc dữ liệu từ file Excel
@@ -84,7 +84,8 @@ def setup_vector():
 # Initialize components
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3", token=HF_TOKEN)
 chroma_client, chroma_collection = init_chroma()
-vectorstore = init_vectorstore()
+init_vectorstore()
+vectorstore = setup_vector()
 
 # Initialize memory buffer
 memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)