How to Run All executable code blocks in Jupyter Book - missing "Restart & Run All"?

[rsenft1]

I'm trying to figure out how to have a code block in a jupyter book with a "Restart and Run all" button (in addition to the typical Run and Restart buttons). I see this is definitely possible with sphinx-thebe, but I can't figure out how to make this work with a jupyter book. I need to be able to have a user change variable values in a visible code block then trigger a bunch of hidden code blocks to run. Here's an image of what I need to get from the sphinx-thebe site. When I try downloading the page and using it in my jupyter book, I don't get the Restart & Run All button. Any suggestions?