How hard would it be to make this work in JupyterLab?

I tried it, but the regular JupyterLab exception style remains. I guess they do something similar where they intercept the output and have their own format. Any ideas how to make it you pretty_errors?