Hi @Alexboiboi,
I really like this idea, though it would take a bit of work. The challenge is that Jupyter’s architecture is designed so that a kernel can’t directly ask which front-end it’s running in (e.g. IPython console, QtConsole, classic notebook, JupyterLab, etc.). The way around this, for JupyterLab, is to make this information available to the kernel from a JupyterLab extension.
I added an issue comment with the idea at https://github.com/plotly/plotly.py/issues/1473#issuecomment-479872266. Feel free to chime in to add any other thoughts you have on the idea!
-Jon