from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import (QAction, QApplication, QWidget,QGridLayout)
#from board import Board
#from clearButton import ClearButton
#from trainWidget import TrainWidget
#from testWidget import TestWidget
from ubuwidgets import (Button, ComboWidget, Board)

class MainWindow(QWidget):
    def __init__(self,parent=None):
        super(MainWindow, self).__init__(parent)
        self.scribbleArea = Board()
        self.clearButton = Button(self, "Clear Image", self.on_clear)
        self.trainButton = Button(self, "Train", self.on_train)
        self.saveCombo = ComboWidget(self, "Label: ", "Save", self.on_save)
        self.testCombo = ComboWidget(self, "Result: ", "Test", self.on_test, True)
        grid = QGridLayout()
        grid.addWidget(self.scribbleArea,1,0)
        grid.addWidget(self.clearButton,2,0)
        grid.addWidget(self.trainButton,3,0)
        grid.addWidget(self.saveCombo,4,0)
        grid.addWidget(self.testCombo,5,0)
        self.setLayout(grid)
        self.setWindowTitle("number ubu")
        self.resize(512, 700)

    def on_clear(self):
        self.scribbleArea.clearImage()

    def on_train(self):
        print('training..')
        
    def on_save(self):
        print('on_save')
        self.scribbleArea.saveImage(self.saveCombo.text())
        self.scribbleArea.clearImage()
        
    def on_test(self):
        print("testing...")
        
        
if __name__ == '__main__':
    import sys

    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())