# Form implementation generated from reading ui file 'interface.ui'
#
# Created by: PyQt6 UI code generator 6.11.0
#
# WARNING: Any manual changes made to this file will be lost when pyuic6 is
# run again.  Do not edit this file unless you know what you are doing.


from PyQt6 import QtCore, QtGui, QtWidgets
import serial
import time


class Ui_MainWindow(object):
    def setupUi(self, MainWindow):
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(500, 420)

        self.centralwidget = QtWidgets.QWidget(parent=MainWindow)
        self.centralwidget.setObjectName("centralwidget")

        self.label = QtWidgets.QLabel(parent=self.centralwidget)
        self.label.setGeometry(QtCore.QRect(50, 40, 150, 40))
        font = QtGui.QFont()
        font.setPointSize(20)
        self.label.setFont(font)
        self.label.setObjectName("label")

        self.TextBox = QtWidgets.QLineEdit(parent=self.centralwidget)
        self.TextBox.setGeometry(QtCore.QRect(50, 100, 220, 40))
        self.TextBox.setObjectName("TextBox")

        self.register_2 = QtWidgets.QPushButton(parent=self.centralwidget)
        self.register_2.setGeometry(QtCore.QRect(300, 100, 150, 40))
        font = QtGui.QFont()
        font.setPointSize(14)
        self.register_2.setFont(font)
        self.register_2.setObjectName("register_2")

        self.listView = QtWidgets.QListView(parent=self.centralwidget)
        self.listView.setGeometry(QtCore.QRect(50, 180, 400, 200))
        self.listView.setObjectName("listView")

        MainWindow.setCentralWidget(self.centralwidget)

        self.menubar = QtWidgets.QMenuBar(parent=MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 500, 22))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)

        self.statusbar = QtWidgets.QStatusBar(parent=MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

        self.retranslateUi(MainWindow)

        # List model
        self.ids = []
        self.model = QtCore.QStringListModel()
        self.listView.setModel(self.model)

        # Serial connection
        self.serial_port = None
        self.connect_serial()

        # Button and textbox actions
        self.register_2.clicked.connect(self.register_id)
        self.TextBox.returnPressed.connect(self.register_id)

        QtCore.QMetaObject.connectSlotsByName(MainWindow)

    def retranslateUi(self, MainWindow):
        _translate = QtCore.QCoreApplication.translate
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))
        self.label.setText(_translate("MainWindow", "Write ID:"))
        self.register_2.setText(_translate("MainWindow", "Register"))

    def connect_serial(self):
        try:
            self.serial_port = serial.Serial("COM10", 115200, timeout=1)
            time.sleep(2)
            print("Connected to ESP32 on COM10")
        except Exception as error:
            print("Could not connect to COM10")
            print(error)

    def register_id(self):
        user_id = self.TextBox.text().strip()

        if user_id:
            self.ids.append(user_id)
            self.model.setStringList(self.ids)
            self.TextBox.clear()

            last_index = self.model.index(len(self.ids) - 1)
            self.listView.scrollTo(last_index)

            self.send_to_esp32("REGISTER")

    def send_to_esp32(self, command):
        if self.serial_port and self.serial_port.is_open:
            self.serial_port.write((command + "\n").encode())
            print(f"Sent to ESP32: {command}")
        else:
            print("Serial port is not connected.")


if __name__ == "__main__":
    import sys

    app = QtWidgets.QApplication(sys.argv)
    MainWindow = QtWidgets.QMainWindow()
    ui = Ui_MainWindow()
    ui.setupUi(MainWindow)
    MainWindow.show()
    sys.exit(app.exec())