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

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.df = pd.read_csv("./data.csv")

        self.show_map()


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

        ###

        items = self.df.loc[self.df['population'] >= 100000]
        for item in items.itertuples():
            icon = folium.Icon()
            if item.population >= 1000000:
                icon = folium.Icon(
                    color="red",
                    icon="atom",
                    prefix='fa',
                )

            folium.Marker(
                location=[item.latitude_dd, item.longitude_dd],
                popup=item.settlement,
                icon=icon
            ).add_to(m)

            folium.Marker(
                location=[item.latitude_dd, item.longitude_dd],
                icon=folium.DivIcon(
                    icon_size=[80, 35],
                    html=f'<div style="background: yellow; text-align: center">{item.settlement}<br>{item.population} чел.<span/>'
                )
            ).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())