We could explore the [VSCode Notebook API](https://code.visualstudio.com/api/extension-guides/notebook) and see if we could build something out from it.