Update jupyter_app.py

Author: deepyaman

jupyter_dash/jupyter_app.py

@@ -17,10 +17,32 @@
 from ansi2html import Ansi2HTMLConverter
 import uuid
 
+from .comms import _dash_comm, _jupyter_config, _request_jupyter_config
 
-from werkzeug.debug.tbtools import get_current_traceback
 
-from .comms import _dash_comm, _jupyter_config, _request_jupyter_config
+def _get_skip(error: Exception, divider=2):
+
+    try:
+        # pylint: disable=import-outside-toplevel
+        from werkzeug.debug import tbtools
+    except ImportError:
+        tbtools = None
+
+    # werkzeug<2.1.0
+    if hasattr(tbtools, "get_current_traceback"):
+        tb = tbtools.get_current_traceback()
+        text = tb.plaintext.splitlines()
+
+    if hasattr(tbtools, "DebugTraceback"):
+        tb = tbtools.DebugTraceback(error)  # pylint: disable=no-member
+        text = tb.render_traceback_text().splitlines()
+    
+    skip = 0
+    for i, line in enumerate(text):
+        if "%% callback invoked %%" in line:
+            skip = int((i + 1) / divider)
+            break
+    return skip
 
 
 class JupyterDash(dash.Dash):
@@ -356,18 +378,12 @@ def _config_callback_exception_handling(
     ):
 
         @self.server.errorhandler(Exception)
-        def _wrap_errors(_):
+        def _wrap_errors(error):
             """Install traceback handling for callbacks"""
             self._traceback = sys.exc_info()[2]
 
             # Compute number of stack frames to skip to get down to callback
-            tb_werkzeug = get_current_traceback()
-            skip = 0
-            if dev_tools_prune_errors:
-                for i, line in enumerate(tb_werkzeug.plaintext.splitlines()):
-                    if "%% callback invoked %%" in line:
-                        skip = int((i + 1) / 2)
-                        break
+            skip = _get_skip(error) if dev_tools_prune_errors else 0
 
             # Customized formatargvalues function so we can place function parameters
             # on separate lines