How can I correctly setup the visualization in Jupyter?

I want to get started with the NGSolve tutorials using Jupyter but I am facing some issue due to which I cannot render the geometry/mesh/solution.

I have
jupyter-notebook-lab-extension and webgui-jupyter-widgets installed and enabled. However, when I try to run this it says Error displaying widget:

mesh = Mesh(unit_square.GenerateMesh(maxh=0.2))
Draw (mesh);

Any tips on how I can troubleshoot this problem ?