This Jupyter notebook-like system you're describing here is basically Runme. It do not yet have an IntelliJ/Jetbrains plugin; however, the Kernel Architecture outside of the headless CLI allows for integration into any IDE or UI.

It's all open source. Please get involved! We'd be happy to have you.