python - Hide sidebar buttons when switching tabs - Stack Overflow

Let's say i have this small streamlit application with a sidebar that contains 4 buttons for example (Button 1, ..., Button 4). And in the main panel, I have two tabs (Tab 1 and Tab 2).

Now I want this :

• When I select Tab 1, ONLY display Button 1 and Button 2 and hide Button 3 and Button 4.

• When I select Tab 2, ONLY display Button 3 and Button 4 and hide Button 1 and Button 2.

I tried this with session_state to store the current tab but it's not working. Any ideas ?

import streamlit as st

def display_buttons(tab):
    if tab == "Tab 1":
        st.sidebar.button("Button 1")
        st.sidebar.button("Button 2")
    elif tab == "Tab 2":
        st.sidebar.button("Button 3")
        st.sidebar.button("Button 4")

if 'active_tab' not in st.session_state:
    st.session_state['active_tab'] = 'Tab 1'

tab1, tab2 = st.tabs(["Tab 1", "Tab 2"])

if tab1:
    st.session_state['active_tab'] = 'Tab 1'
if tab2:
    st.session_state['active_tab'] = 'Tab 2'

with tab1:
    st.write("This is Tab 1")
    if st.session_state['active_tab'] == 'Tab 1':
        display_buttons("Tab 1")

with tab2:
    st.write("This is Tab 2")
    if st.session_state['active_tab'] == 'Tab 2':
        display_buttons("Tab 2")