Update app.py

app.py

@@ -85,6 +85,8 @@ def main():
         st.session_state.conversation = None
     if "chat_history" not in st.session_state:
         st.session_state.chat_history = None
+    if "vectorstore_saved" not in st.session_state:
+        st.session_state.vectorstore_saved = False
     
     if st.session_state.conversation is None:
         if os.path.isdir("./chroma_db"):
@@ -92,6 +94,7 @@ def main():
                 with st.spinner("Loading vector store..."):
                     vectorstore = Chroma(persist_directory="./chroma_db", embedding_function=OpenAIEmbeddings())
                     st.session_state.conversation = get_conversation_chain(vectorstore)
+                    st.session_state.vectorstore_saved = True
 
     if st.session_state.conversation is None:
         pdf_doc = getpdfdoc()
@@ -110,7 +113,11 @@ def main():
 
     if st.session_state.conversation is not None:
         st.header("Ask questions from 48 Laws of Power:books:")
-        user_question = st.chat_input("Ask a question about your documents:")
+        if st.session_state.vectorstore_saved:
+            st.info("Chroma saved")
+        else:
+            st.info("Chroma failed to save")
+        user_question = st.chat_input("Ask a question related 48 Laws of Power:")
         if user_question:
             handle_userinput(user_question)