Update app.py

app.py

@@ -1,19 +1,16 @@
 import streamlit as st
 import pandas as pd
 
-st.title("Topic Modeling to Determine Climate Anxiety Among Youth")
-
-st.header("Our Mission and Plan", divider='red')
-st.markdown("We're using **BERTopic** and **LDA** as baseline models.")
+#st.header("Our Mission and Plan", divider='red')
+#st.markdown("We're using **BERTopic** and **LDA** as baseline models.")
 
 if "page" not in st.session_state:
     st.session_state.page = "Home"
 
 def navigate_to(page):
     st.session_state.page = page
-
-st.subheader("Select a Section:")
-col1, col2, col3, col4 = st.columns(4)
+    
+col1, col2, col3, col4, col5 = st.columns(5)
 
 with col1:
     if st.button("Home"):
@@ -31,10 +28,14 @@ with col4:
     if st.button("LDA"):
         navigate_to("LDA")
 
+with col5:
+    if st.button("Contact Info"):
+        navigate_to("Contact Info")
 st.divider()
 
 if st.session_state.page == "Home":
     st.header("Home Page")
+    st.title("Topic Modeling to Determine Climate Anxiety Among Youth")
 
 elif st.session_state.page == "Sentiment Analysis":
     st.header("Sentiment Analysis")
@@ -45,10 +46,5 @@ elif st.session_state.page == "Topic Modelling":
 elif st.session_state.page == "LDA":
     st.header("Latent Dirichlet Allocation (LDA)")
 
-st.divider()
-
-if st.button("Contact Info"):
-    st.session_state.page = "Contact Info"
-
-if st.session_state.page == "Contact Info":
+elif st.session_state.page == "Contact Info":
     st.header("Contact Information")