Source code for solidipes.viewers.notebook

import streamlit as st
from IPython.display import Markdown, display

from .. import loaders
from ..utils import solidipes_logging as logging
from ..utils import viewer_backends
from .viewer import Viewer

################################################################
logger = logging.getLogger()
################################################################




class Notebook(Viewer):
    """Viewer for notebooks"""

    def __init__(self, data=None):
        self.compatible_data_types = [loaders.Notebook]
        #: Text to display
        self.notebook = None
        super().__init__(data)



    def add(self, data_container):
        """Append text to the viewer"""
        self.check_data_compatibility(data_container)

        if isinstance(data_container, loaders.DataContainer):
            self.notebook = data_container.notebook




    def show(self):
        if viewer_backends.current_backend == "jupyter notebook":
            from nbconvert import PythonExporter

            python_exporter = PythonExporter()
            script, resources = python_exporter.from_notebook_node(self.notebook)
            display(Markdown(script))

        elif viewer_backends.current_backend == "streamlit":
            from nbconvert import HTMLExporter

            html_exporter = HTMLExporter(template_name="classic")
            (body, resources) = html_exporter.from_notebook_node(self.notebook)
            st.components.v1.html(body, height=1000, scrolling=True)
            # st.markdown(resources, unsafe_allow_html=True)
        else:  # python
            from nbconvert import PythonExporter

            python_exporter = PythonExporter()
            script, resources = python_exporter.from_notebook_node(self.notebook)
            print(script)