# Import the necessary modules required
import sys

'''
from PySide2.QtWidgets import *
from PySide2.QtGui import *
from PySide2.QtCore import *
'''


from PySide6.QtWidgets import *
from PySide6.QtGui import *
from PySide6.QtCore import *



# Main widget class
class MainWidget(QWidget):
    def __init__(self, parentObj = None):
        super(MainWidget,self).__init__()
        self.setFixedSize(150, 150)
        self.show()

    # paintEvent() - Function called on the update of the widget.
    def paintEvent(self, event: QPaintEvent):

        gradient = QLinearGradient(10, 10, 100, 100)

        gradient.setColorAt(0, Qt.red) # Start color
        gradient.setColorAt(0.5, Qt.green)
        gradient.setColorAt(1, Qt.blue) # End color

        painter = QPainter()
        painter.begin(self)


        painter.setBrush(gradient)
        painter.setPen(Qt.NoPen)
        painter.drawRect(10,10, 100, 100)

        print("\n\nGradient stop points:")
        print(gradient.stops())

        painter.end()





if __name__ =='__main__':
    # Exception Handling
    try:
        myApp = QApplication(sys.argv)
        myWindow = MainWidget()
        myApp.exec_()
        sys.exit(0)
    except NameError:
        print("Name Error:", sys.exc_info()[1])
    except SystemExit:
        print("Closing Window...")
    except Exception:
        print(sys.exc_info()[1])