<?xml version="1.0" encoding="UTF-8"?>
<ui version="4.0">
 <class>MainWindow</class>
 <widget class="QMainWindow" name="MainWindow">
  <property name="geometry">
   <rect>
    <x>0</x>
    <y>0</y>
    <width>700</width>
    <height>422</height>
   </rect>
  </property>
  <property name="windowTitle">
   <string>MainWindow</string>
  </property>
  <widget class="QWidget" name="centralwidget">
   <layout class="QGridLayout" name="gridLayout">
    <property name="leftMargin">
     <number>10</number>
    </property>
    <property name="topMargin">
     <number>10</number>
    </property>
    <property name="rightMargin">
     <number>10</number>
    </property>
    <property name="bottomMargin">
     <number>10</number>
    </property>
    <item row="0" column="0">
     <widget class="QLabel" name="otwartyPlik">
      <property name="sizePolicy">
       <sizepolicy hsizetype="Preferred" vsizetype="Fixed">
        <horstretch>0</horstretch>
        <verstretch>0</verstretch>
       </sizepolicy>
      </property>
      <property name="autoFillBackground">
       <bool>false</bool>
      </property>
      <property name="styleSheet">
       <string notr="true">background-color: rgb(226, 226, 226);</string>
      </property>
      <property name="text">
       <string/>
      </property>
     </widget>
    </item>
    <item row="1" column="0">
     <widget class="QSplitter" name="splitter">
      <property name="orientation">
       <enum>Qt::Horizontal</enum>
      </property>
      <widget class="QTableView" name="tabela"/>
      <widget class="QLabel" name="wykres">
       <property name="styleSheet">
        <string notr="true">background-color: rgb(230, 230, 230);</string>
       </property>
       <property name="text">
        <string/>
       </property>
      </widget>
     </widget>
    </item>
    <item row="2" column="0">
     <widget class="QPushButton" name="generujButton">
      <property name="enabled">
       <bool>false</bool>
      </property>
      <property name="sizePolicy">
       <sizepolicy hsizetype="Fixed" vsizetype="Fixed">
        <horstretch>0</horstretch>
        <verstretch>0</verstretch>
       </sizepolicy>
      </property>
      <property name="text">
       <string>Generuj Wykres</string>
      </property>
     </widget>
    </item>
   </layout>
  </widget>
  <widget class="QMenuBar" name="menubar">
   <property name="geometry">
    <rect>
     <x>0</x>
     <y>0</y>
     <width>700</width>
     <height>20</height>
    </rect>
   </property>
   <widget class="QMenu" name="menuPlik">
    <property name="title">
     <string>Plik</string>
    </property>
    <addaction name="actionOtworz"/>
    <addaction name="actionZapisz"/>
    <addaction name="actionZamknij"/>
   </widget>
   <addaction name="menuPlik"/>
  </widget>
  <widget class="QStatusBar" name="statusbar"/>
  <action name="actionOtworz">
   <property name="text">
    <string>Otwórz dane</string>
   </property>
  </action>
  <action name="actionZapisz">
   <property name="text">
    <string>Zapisz wykres</string>
   </property>
  </action>
  <action name="actionZamknij">
   <property name="text">
    <string>Zamknij aplikację</string>
   </property>
  </action>
 </widget>
 <resources/>
 <connections/>
</ui>

from PyQt5.QtCore import Qt, QAbstractTableModel

class TabelaModel(QAbstractTableModel):
    """Klasa potomna abstrakcyjnej klasy QAbstractTableModel
    należy zaimplementować poniższe metody.
    """
    def __init__(self, data):
        super().__init__()
        self._data = data

    def rowCount(self, parent=None):
        """Zwraca liczbę rzędów"""
        return self._data.shape[0]

    def columnCount(self, parent=None):
        """Zwraca liczbę kolumn"""
        return self._data.shape[1]

    def data(self, index, role=Qt.DisplayRole):
        """Zwraca dane z tabeli określone przez index"""
        if index.isValid():
            if role == Qt.DisplayRole:
                return str(self._data.iloc[index.row(), index.column()])
        return None

    def headerData(self, col, orientation, role):
        """Ustawia nazwy kolumn w tabeli"""
        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
            return self._data.columns[col]
        return None

import sys
from PyQt5.QtWidgets import QMainWindow, QApplication, QFileDialog 
from PyQt5.uic import loadUi
from PyQt5.QtGui import QPixmap
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
import tabela_model as tab


class Wizualizator(QMainWindow):
    """Główna klasa aplkacji"""
    def __init__(self):
        super().__init__()
        # Import interfejsu użytkownika bezpośrednio
        # z pliku wizualizator.ui
        loadUi("wizualizator.ui", self)
        self.show()
        # Powiązanie sygnałów (akcji wywoływanych na widżetach) 
        # i slotów (funkcji)
        self.actionOtworz.triggered.connect(self.otworz_plik)
        self.actionZamknij.triggered.connect(self.zakoncz)
        self.actionZapisz.triggered.connect(self.zapisz_wykres)
        self.generujButton.clicked.connect(self.generuj_wykres)
        self.tabela.clicked.connect(self.pobierz_wspolrzedne)
        # Status załadowania danych do tabeli potrzebne 
        # do sprawdzenia czy można generować wykres
        self.dane_zaladowane = False

        
    def pobierz_wspolrzedne(self, item):
        """Pobiera ideks wybranej kolumny w tabeli.
        Wykres będzie generowany dla danych w tej kolumnie
        """
        if item.column() < 2:
            self.wybrana_kolumna = 1
        else:
            self.wybrana_kolumna = item.column()

    def otworz_plik(self):      
        """Otwiera plik z użyciem okienka dialogowego"""
        # Otwieranie okna dialogiwego
        plik, _ = QFileDialog.getOpenFileName(self,
                "Otwórz plik", 
                "",
                "Wszystkie pliki (*);;CSV files (*.csv)")       
        # Jeżeli wybrano plik do otwarcia
        if plik:
            # Odczyt danych z pliku csv
            self.dane = pd.read_csv(plik)
            self.dane_zaladowane = True
            # Tworzenie obiektu tabeli 
            dane_tab = tab.TabelaModel(self.dane)
            # Załadowanie tabeli do widżetu
            self.tabela.setModel(dane_tab)
            # Uaktywnienie przycisku generowania wykresu
            self.generujButton.setEnabled(True)
            # W widżecie otwartyPlik zostanie wyświetlona 
            # ścieżka i nazwa otwartego poliku z danymi
            self.otwartyPlik.setText(plik)
            # Zmiana koloru tła widżetu
            self.otwartyPlik.setStyleSheet("background-color: rgb(240, 240, 240)")
            # Domyślnie do wykresu używane sa dane z 2 kolumny
            # (index = 1) 
            self.wybrana_kolumna = 1

    def zapisz_wykres(self):
        """Zapisuje wykres w wybranym formacie"""
        # Okno dialogowe wyboru pliku do zapisania
        plik, _ = QFileDialog.getSaveFileName(
                                        self,
                                        "Zapisz wykres jako",
                                        ".png",
                                        "Wszystkie pliki (*);;"+
                                        "Pliki bitmapowe (*.png *.jpg);;"+
                                        "Pliki wektorowe (*.svg *.pdf)",
                                        )
        # Jeśli została wybrana nazwa pliku, to zapis wykresu
        if plik:
            self.wykres_sb.get_figure().savefig(plik)

    def generuj_wykres(self):
        """Generuje wykres"""
        if self.dane_zaladowane:
            # Dane dla osi x - pierwsza kolumna danych
            dane_x = self.dane.iloc[:,0]
            # Dane dla osi y - wybrana kolumna (przez kliknięcie komórki)
            # Domyślnie jest to 2 kolumna
            dane_y = self.dane.iloc[:,self.wybrana_kolumna]
            # Nazwa pliku używanego przez program do zapisu
            # wykresu, który będzie wyświetlany
            plik = 'wykres.png'
            # Potrzebne aby wykresy się "resetowały" przy zmianie danych
            plt.figure()
            # Tworzenie wykresu z użyciem seaborn
            self.wykres_sb = sns.barplot(x = dane_x,
                                    y = dane_y,
                                    palette = ['brown', 'red', 'darkorange'])
            # Zapis wykresu do pliku, który będzie wyświetlany
            self.wykres_sb.get_figure().savefig(plik)
            # Tworzenie obiektu z obrazem wykresu
            wykres_png = QPixmap(plik)
            # Umieszczenie obrazu wykresu w widżecie wykres
            self.wykres.setPixmap(wykres_png) 
            # Zmiana rozmiaru widżetu wykres do rozmiaru obrazka         
            self.wykres.resize(wykres_png.width(),wykres_png.height())
            # Zmiana rozmiarów okna aplikacji
            self.resize(self.tabela.width() +
                        wykres_png.width() +
                        25,wykres_png.height()+100)
            # Uaktywnienie pozycji w menu umozliwiającej zapis wykresu
            self.actionZapisz.setEnabled(True)
        else:
            print("Brak danych")

    def zakoncz(self):
        """Wyłącza aplikację"""
        self.close()

if __name__ == "__main__":
    """Główna funkcja aplikacji"""
    # Uruchomienie obiektu aplikacji
    aplikacja = QApplication(sys.argv)
    # Tworzenie okna
    okno = Wizualizator()  
    # Wyświetlenie okna
    okno.show()
    # Przy zamknięciu aplikacji, zwalnia się pamięć komputera
    sys.exit(aplikacja.exec_())