Update app.py

app.py

@@ -349,16 +349,14 @@ if "menu" not in st.session_state:
 
 # sidebar with menu navigation
 with st.sidebar:
-    st.subheader("DO NOT NAVIGATE between pages when agent is still generating messages in the chat.  Wait for query to complete first.\n")
-    selected = option_menu("Main Menu", st.session_state["menu"], 
+    st.subheader("DO NOT NAVIGATE between pages when agent is still generating messages in the chat.  Wait for query to complete first.")
+    st.write("")
+    page = option_menu("Main Menu", st.session_state["menu"], 
         icons=['house', 'gear', 'gear', 'gear'], menu_icon="cast", default_index=0)
     st.spinner("test")
-
     
-col1, col2 = st.columns(2)
-# with col1:
 
-if selected == "Chatbot":
+if page == "Chatbot":
     st.header("Chat")
     
     # Store the conversation in the session state.
@@ -432,16 +430,20 @@ if selected == "Chatbot":
             st.markdown(response)
 
 
-    # with col2:
-    #     st.write("hi")
+if page == "Chat Config":
+    st.header(page)
+
+
+if page == "Document, Retriever, Web Scraping Config":
+    st.header(page)
 
 
-if selected == "Source Documents for Last Query":
-    st.header("Source Documents for Last Query")
+if page == "Source Documents for Last Query":
+    st.header(page)
     try:
         st.subheader(st.session_state['source_documents'][0])
         for doc in st.session_state['source_documents'][1:]:
             st.write("Source: " + doc.metadata['source'])
             st.write(doc)
     except:
-        st.write("No source documents retrieved yet. Please run a user query before coming back to this page.")
+        st.write("No source documents retrieved yet. Please run a user query before coming back to this page.")