Update app.py

app.py

@@ -77,7 +77,7 @@ class MyCallbackHandler(BaseCallbackHandler):
     def on_tool_end(self, output, **kwargs):
         """Run when tool ends running."""
         #print("\n\nTool End: ", output)
-        tool_output = f"[Tool Output] {output}  \n  \nI am processing the output from the tool..."
+        tool_output = f":blue[[Tool Output]] {output}  \n  \nI am processing the output from the tool..."
         st.session_state.messages.append(
             {"role": "assistant", "content": tool_output}
         )
@@ -344,12 +344,13 @@ if "menu" not in st.session_state:
             "Chatbot", 
             "Chat Config",
             "Document and Retriever Config",
-            "Source Documents  \n (for Last Query, Click Only After Full Execution)",      
+            "Source Documents for Last Query",      
         ]
 
 with st.sidebar:
     selected = option_menu("Main Menu", st.session_state["menu"], 
         icons=['house', 'gear', 'gear'], menu_icon="cast", default_index=0)
+    st.write("DO NOT navigate between pages when agent is generate messages in the chat.")
 
 col1, col2 = st.columns(2)
 # with col1:
@@ -432,7 +433,7 @@ if selected == "Chatbot":
     #     st.write("hi")
 
 
-if selected == "Source Documents  \n (for Last Query, Click Only After Full Execution)":
+if selected == "Source Documents for Last Query":
     st.header("Source Documents for Last Query")
     try:
         st.subheader(st.session_state['source_documents'][0])