Buttons always return to False so it’s not a good idea to have other widgets afterwards.
- A demo of exactly that is in this video tutorial: Streamlit Shorts: How to combine a button, checkbox and radio buttion - YouTube
- You can check this similar thread for another example.