import matplotlib
matplotlib.use('Qt5Agg')
import Graph 
from PyQt5 import QtWidgets

from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg, NavigationToolbar2QT as NavigationToolbar


class GraphWidget(QtWidgets.QWidget):
    def __init__(self, x, y, labels, mult_plots=False):
        super().__init__()
        graph = Graph.Graph(x,y,labels, mult_subplots=mult_plots)

        sc = FigureCanvasQTAgg(graph.figure)

        # Create toolbar, passing canvas as first parament, parent (self, the MainWindow) as second.
        toolbar = NavigationToolbar(sc, self)

        layout = QtWidgets.QVBoxLayout()
        layout.addWidget(toolbar)
        layout.addWidget(sc)

        self.setLayout(layout)