from PyQt5.QtWidgets import *
from PyQt5.QtGui import QPixmap
import sys
from PyQt5 import QtCore
from functions.get_img_map import get_img_map


class Window(QMainWindow):
    def __init__(self):
        super().__init__()
        self.acceptDrops()
        self.setWindowTitle("Image")
        self.setGeometry(0, 0, 600, 450)
        self.label = QLabel(self)
        self.delta = 0.02
        get_img_map(self.delta)
        self.pixmap = QPixmap('image.png')
        self.label.setPixmap(self.pixmap)
        self.label.resize(self.pixmap.width(), self.pixmap.height())
        self.show()

    def keyPressEvent(self, event):
        try:
            if event.key() == QtCore.Qt.Key_Up:
                self.delta += 0.01
                self.update()
            if event.key() == QtCore.Qt.Key_Down:
                self.delta -= 0.01
                self.update()
            event.accept()
        except Exception as e:
            print(e)

    def update(self):
        get_img_map(self.delta)
        self.pixmap = QPixmap('image.png')
        self.label.setPixmap(self.pixmap)
        self.label.resize(self.pixmap.width(), self.pixmap.height())
        self.show()


App = QApplication(sys.argv)
window = Window()
sys.exit(App.exec())