import sys
import os
import PySide6
from PySide6.QtWidgets import QApplication, QMainWindow
from mainwindow import Ui_MainWindow
import folium
import io

dirname = os.path.dirname(PySide6.__file__)
plugin_path = os.path.join(dirname, 'plugins', 'platforms')
os.environ['QT_QPA_PLATFORM_PLUGIN_PATH'] = plugin_path


class MainWindow(QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        self.show_map()


    def show_map(self):
        # создание объекты карты
        m = folium.Map(
            location=[52.286387, 104.280660],
            zoom_start=11,
            tiles="CartoDB Voyager"
        )

        ###

        folium.Marker(
            location=[52.286387, 104.280660],
            popup="Это Иркутск",
            icon=folium.Icon(
                icon="cow",
                prefix="fa",
                color="lightred",
                icon_color="black"
            )
        ).add_to(m)

        folium.CircleMarker(
            location=[52.226387, 104.220660],
            popup="Это уже не Иркутск",
            fill=True,
            radius=30,
            color="red",
            fill_color="blue",
            fill_opacity=0.9,
        ).add_to(m)

        folium.Marker(
            location=[52.226387, 104.220660],
            popup="Это СНТ",
            icon=folium.DivIcon(
                html='<span style="font-size: 24px; background: yellow;">Привет</span>'
            )
        ).add_to(m)

        bounds = [
            (52.2599889, 104.2616272),
            (52.2615387, 104.2702103),
            (52.2636006, 104.2692447),
            (52.2649138, 104.2670131),
            (52.2641127, 104.2594814),
            (52.2599626, 104.2616057),
        ]

        folium.Polygon(
            bounds,
            color="red",
            fill=True,
            popup="ИРНИТУ"
        ).add_to(m)

        ###

        # подключение карты к интерфейсу
        m_data = io.BytesIO()
        m.save(m_data, close_file=False)
        self.ui.web.setHtml(m_data.getvalue().decode())



if __name__ == "__main__":
    app = QApplication(sys.argv)

    window = MainWindow()
    window.show()

    sys.exit(app.exec())