Update app.py

app.py

@@ -317,17 +317,17 @@ def view_messages_ui():
             st.write(f"Message: {message_dict['message_content']}")
             st.write("---")
 
-# Streamlit interface for messenger using tabs
 def messenger_ui():
     st.title("Messenger")
-    tabs = st.columns(2)
-    selected_tab = tabs[0].button("Send Message")
-    tabs[1].button("View Messages")
+    selected_tab = st.session_state.get("messenger_selected_tab", "Send Message")
+    selected_tab = st.radio("Select Tab", ["Send Message", "View Messages"], index=0 if selected_tab == "Send Message" else 1)
 
-    if selected_tab:
+    if selected_tab == "Send Message":
         send_message_ui()
+        st.session_state.messenger_selected_tab = "Send Message"
     else:
         view_messages_ui()
+        st.session_state.messenger_selected_tab = "View Messages"
 #end of messaging section 
 
 def tour_guide_ui():