Nested Buttons (session state)

Using Streamlit
, ,
1

Hello, I am trying to figure out how I a make certain buttons disappear after the user clicks on either yes or no. Essentially I would like the “begin calibration” button to be the only thing on the screen if the user clicks “yes” and nothing to appear if the user clicks “no”

import streamlit as st
import time


if "button_clicked" not in st.session_state:
        st.session_state.button_clicked = False

def begin_callback():
        # "Yes" button was clicked
        st.session_state.button_clicked = True

def no_callback():
        # "No" button was clicked
        alert = st.warning("Please go back to the main page")
        time.sleep(3)
        alert.empty()

def yes_callback():
        st.button("Begin Calibration")
        

st.write("Would you like to begin data collection?")

if not st.session_state.button_clicked:
        st.button("Click Here to Begin Data Collection", on_click=begin_callback)

else:
        st.warning("BEFORE SELECTING YES: place one sensor on your sternum and the other on your lower back. Once you have done that, please click yes to begin calibration.")
        st.button("yes", on_click=yes_callback)
        st.button("no", on_click=no_callback)