import os
import sys

from PySide6.QtGui import QIcon
from PySide6.QtWidgets import QApplication, QMainWindow, QPushButton, QSystemTrayIcon


# Subclass QMainWindow to customize your application's main window
class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("My App")

        button = QPushButton("Press Me!")

        # Set the central widget of the Window.
        self.setCentralWidget(button)

        self.systemTrayIcon = QSystemTrayIcon(self)
        iconFilePath = os.path.join(os.path.dirname(__file__), "icon.png")
        icon = QIcon()
        icon.addFile(iconFilePath)
        self.systemTrayIcon.setIcon(icon)
        self.systemTrayIcon.show()

        button.clicked.connect(
            lambda: self.systemTrayIcon.showMessage("Example", "msg", icon, 1000)
        )


app = QApplication(sys.argv)

window = MainWindow()
window.show()

app.exec()