This is now its own package! Install the widget and track the development here: https://github.com/plotly/jupyterlab-dash