diff --git a/pystencils/display_utils.py b/pystencils/display_utils.py index cb09197bf175b43c4896072693ea11253976691f..77e63ba2db377164fef22ea19c952a630a7ccda3 100644 --- a/pystencils/display_utils.py +++ b/pystencils/display_utils.py @@ -30,7 +30,13 @@ def highlight_cpp(code: str): from pygments.lexers import CppLexer css = HtmlFormatter().get_style_defs('.highlight') - css_tag = "<style>{css}</style>".format(css=css) + try: + dark_css = HtmlFormatter(style="stata-dark").get_style_defs('.highlight') + except ModuleNotFoundError: + dark_css = css + + css_tag = "<style>{css} @media (prefers-color-scheme: dark) {{ {dark_css} }}</style>".format(css=css, + dark_css=dark_css) display(HTML(css_tag)) return HTML(highlight(code, CppLexer(), HtmlFormatter()))