nicegui/nice_gui/elements/plot.py

from nice_gui.elements.element import Element
import justpy as jp
import matplotlib.pyplot as plt
from .element import Element

class Plot(Element):

    def __init__(self, close: bool = True):

        self.close = close
        self.fig = plt.figure()

        view = jp.Matplotlib()
        view.set_figure(self.fig)

        super().__init__(view)

    def __enter__(self):

        plt.figure(self.fig)

        return self

    def __exit__(self, *_):

        self.view.set_figure(plt.gcf())

        if self.close:
            self.fig.close()