ui_module.py

# Wardrobe AI - The Console UI. We use PyQT5 for this purpose
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import *                                  
from PyQt5.QtWidgets import *
from PyQt5.QtGui import *
from recognition_module import *

import time


class Ui_MainWindow(object):
    """
   This class generates the main window of our UI. 
   It has functions managing functionalities to add, edit, delete and generate predictions.
   This calls the corresponding individual widgets.
    """
    def __init__(self):
        """
        Initialize lists . We would store outfit information of topwear, bottomwear, footwear in these lists.
        """
        self.top_list = []
        self.bottom_list = []
        self.foot_list = []
        self.season = None
        
        
    def PREDICT_OUTFIT(self):
        """
        Predict function to add an images and classify to top, bottom and shoes. Generate the outfit recommendation
        """
        _translate = QtCore.QCoreApplication.translate
        directory1 = QFileDialog.getOpenFileName(None, "Select file", "H:/")

        # Invoke cloth_classification from recognition module to inspect the image and classify if category = top/bottom/foot
        category, info, res_place_holder = cloth_classification(directory1[0])
        
        # if the result is top, then add an item to the "top" list on GUI.
        if category == "top":
            item = QtWidgets.QListWidgetItem(info)
            self.TOP_LIST.addItem(item)
            self.top_list.append(res_place_holder)
        # if the result is bottom, then add an item to the "bottom" list on GUI.
        elif category == "bottom":
            item = QtWidgets.QListWidgetItem(info)
            self.BOTTOM_LIST.addItem(item)
            self.bottom_list.append(res_place_holder)
        # if the result is shoes, then add an item to the "shoes" list on GUI.
        elif category == "foot":
            item = QtWidgets.QListWidgetItem(info)
            self.SHOE_LIST.addItem(item)
            self.foot_list.append(res_place_holder)
            
    def TOP_LIST_EDIT(self):
            """
           Top edit button: To edit topwear name 
            """
            
            selected_items = self.TOP_LIST.selectedItems()
            text, okPressed = QtWidgets.QInputDialog.getText(self.AddTopButton, "EDIT","Please Edit This Top:", QtWidgets.QLineEdit.Normal, selected_items[0].text())
            for i in selected_items:
                self.TOP_LIST.takeItem(self.TOP_LIST.row(i)) 
            if okPressed and text != '':
                item = QtWidgets.QListWidgetItem(text)
                self.TOP_LIST.addItem(item)
               
    def TOP_LIST_DEL(self):
            """
            Top remove button: To remove topwear from the list 
            """
            selected_items = self.TOP_LIST.selectedItems()
            for i in selected_items:
                self.TOP_LIST.takeItem(self.TOP_LIST.row(i))
            text = selected_items[0].text()
            
            path = text.split(", ")[-1]
            for i in self.top_list:
                if(i[-1] == path):
                    self.top_list.remove(i)
            
    #########################################################################################          
    def BOTTOM_LIST_EDIT(self):
            """
            Bottom edit button: To edit bottompart name
            """
            selected_items = self.BOTTOM_LIST.selectedItems()
            text, okPressed = QtWidgets.QInputDialog.getText(self.AddBottomButton, "EDIT","Please Edit This Bottom:", QtWidgets.QLineEdit.Normal, selected_items[0].text())
            for i in selected_items:
                self.BOTTOM_LIST.takeItem(self.BOTTOM_LIST.row(i))
            if okPressed and text != '':
                item = QtWidgets.QListWidgetItem(text)
                self.BOTTOM_LIST.addItem(item)   

    def BOTTOM_LIST_DEL(self):
            """
            Bottom remove button: To remove bottompart from the list 
            """
            selected_items = self.BOTTOM_LIST.selectedItems()
            for i in selected_items:
                self.BOTTOM_LIST.takeItem(self.BOTTOM_LIST.row(i))
            text = selected_items[0].text()
            
            path = text.split(", ")[-1]
            for i in self.bottom_list:
                if(i[-1] == path):
                    self.bottom_list.remove(i)
                    
    ###########################################################################################         
           
    def SHOE_LIST_EDIT(self):
            """
            Shoes edit button: To edit footwear name
            """
            selected_items = self.SHOE_LIST.selectedItems()
            text, okPressed = QtWidgets.QInputDialog.getText(self.AddShoeButton, "EDIT","Please Edit This Shoes:", QtWidgets.QLineEdit.Normal, selected_items[0].text())
            for i in selected_items:
                self.SHOE_LIST.takeItem(self.SHOE_LIST.row(i))
            if okPressed and text != '':
                item = QtWidgets.QListWidgetItem(text)
                self.SHOE_LIST.addItem(item)   
                
    def SHOE_LIST_DEL(self):
            """
            Shoes remove button: To remove footwear from the list 
            """
            selected_items = self.SHOE_LIST.selectedItems()
            for i in selected_items:
                self.SHOE_LIST.takeItem(self.SHOE_LIST.row(i))
            text = selected_items[0].text()
            
            path = text.split(", ")[-1]
            for i in self.foot_list:
                if(i[-1] == path):
                    self.foot_list.remove(i)
    ######################################################################################################
    def Generate(self):
        """
        Incorporates information of month(season) into the outfit prediction.
        """
        top_right_season = [i for i in self.top_list if i[3] == self.season ]   #select topwears for the given season
         # If topwear is not present for the given season, then pick one at random
        if top_right_season != []:
            ad_top = top_right_season[np.random.randint(len(top_right_season))]
        else:
            ad_top = self.top_list[np.random.randint(len(self.top_list))]  

        helper_bot = [i for i in self.bottom_list if i[4] == ad_top[4] ]
        helper_sho = [i for i in self.foot_list if i[4] == ad_top[4] ]

        if helper_bot==[]:
            ad_bot = self.bottom_list[np.random.randint(len(self.bottom_list))]
        else:
            bot_right_season = [i for i in helper_bot if i[3] == self.season]
            
            if bot_right_season != []:
                ad_bot = bot_right_season[np.random.randint(len(bot_right_season))]
            else:
                ad_bot = helper_bot[np.random.randint(len(helper_bot))]   
        if helper_sho==[]:
            ad_sho = self.foot_list[np.random.randint(len(self.foot_list))]
        else:
            sho_right_season = [i for i in helper_sho if i[3] == self.season]
            
            if sho_right_season != []:
                ad_sho = sho_right_season[np.random.randint(len(sho_right_season))]
            else:
                ad_sho = helper_sho[np.random.randint(len(helper_sho))]
        
        self.listWidget_1.setPixmap(QtGui.QPixmap(ad_top[-1]).scaled(281,300))
        self.listWidget_2.setPixmap(QtGui.QPixmap(ad_bot[-1]).scaled(281,300))
        self.listWidget_3.setPixmap(QtGui.QPixmap(ad_sho[-1]).scaled(281,300))

    
#####################################################################################
    def index_changed(self, ind): 
        """
         Fetches the season when user changes the month in UI
        """
        to_month = ind
        if to_month in [2,3,4]:
            toseason = "Spring"
        elif to_month in [5,6,7]:
            toseason = "Summer"
        elif to_month in [8,9,10]:
            toseason = "Fall"
        else:
            toseason = "Winter"
        self.season = toseason
       
        
        
#####################################################################################

    # UI elements setup
    
    def placeWidgets(self, MainWindow):
        """
        Add items into GUI.
        """
        MainWindow.setObjectName("MainWindow")
        MainWindow.resize(880, 669)
        MainWindow.setWindowIcon(QtGui.QIcon('ui_images/icon.jpg'))
        self.centralwidget = QtWidgets.QWidget(MainWindow)
        self.centralwidget.setObjectName("centralwidget")
        
#####################################################################################
        #Widgets for Tops
        
        """
        Our model outputs 5 attributes each for topwear, bottomwear and footwear.
        We choose a Pyqt - widget list to display these attributes. 
        
        """
        self.TOP_LIST = QtWidgets.QListWidget(self.centralwidget) # Widget list to display Topwear predictions
        self.TOP_LIST.setGeometry(QtCore.QRect(10, 30, 281, 181))
        self.TOP_LIST.setObjectName("TOP_LIST")

        # Defining properties for Top
        self.AddTopButton = QtWidgets.QPushButton(self.centralwidget)
        self.AddTopButton.setGeometry(QtCore.QRect(10, 210, 141, 41))
        self.AddTopButton.setAutoFillBackground(False)
        self.AddTopButton.setCheckable(False)
        self.AddTopButton.setObjectName("AddTopButton")

        self.DeleteTopButton = QtWidgets.QPushButton(self.centralwidget)
        self.DeleteTopButton.setGeometry(QtCore.QRect(150, 210, 141, 41))
        self.DeleteTopButton.setCheckable(False)
        self.DeleteTopButton.setChecked(False)
        self.DeleteTopButton.setObjectName("DeleteTopButton")
        
#####################################################################################
        # Widgets for bottom
        
        self.AddBottomButton = QtWidgets.QPushButton(self.centralwidget) # Widget list to display Bottomwear predictions
        self.AddBottomButton.setGeometry(QtCore.QRect(300, 210, 141, 41))
        self.AddBottomButton.setObjectName("AddBottomButton")
        self.BOTTOM_LIST = QtWidgets.QListWidget(self.centralwidget)
        self.BOTTOM_LIST.setGeometry(QtCore.QRect(300, 30, 281, 181))
        self.BOTTOM_LIST.setObjectName("BOTTOM_LIST")
        self.DeleteBottomButton = QtWidgets.QPushButton(self.centralwidget)
        self.DeleteBottomButton.setGeometry(QtCore.QRect(440, 210, 141, 41))
        self.DeleteBottomButton.setObjectName("DeleteBottomButton")
        
#####################################################################################
        # Widgets for footwear
        
        self.AddShoeButton = QtWidgets.QPushButton(self.centralwidget) # Widget list to display Footwear predictions
        self.AddShoeButton.setGeometry(QtCore.QRect(590, 210, 141, 41))
        self.AddShoeButton.setObjectName("AddShoeButton")
        self.SHOE_LIST = QtWidgets.QListWidget(self.centralwidget)
        self.SHOE_LIST.setGeometry(QtCore.QRect(590, 30, 281, 181))
        self.SHOE_LIST.setObjectName("SHOE_LIST")
        self.DeleteShoeButton = QtWidgets.QPushButton(self.centralwidget)
        self.DeleteShoeButton.setGeometry(QtCore.QRect(730, 210, 141, 41))
        self.DeleteShoeButton.setObjectName("DeleteShoeButton")
        
#####################################################################################
        # Widgets for generate button
        
        self.GenerateButton = QtWidgets.QPushButton(self.centralwidget)
        self.GenerateButton.setGeometry(QtCore.QRect(660, 270, 141, 41))
        self.GenerateButton.setObjectName("GenerateButton")
        self.AddOutfitButton = QtWidgets.QPushButton(self.centralwidget)
        self.AddOutfitButton.setGeometry(QtCore.QRect(80, 270, 141, 41))
        self.AddOutfitButton.setObjectName("AddOutfitButton")
        self.combo = QtWidgets.QComboBox(self.centralwidget)
        self.combo.setGeometry(QtCore.QRect(370, 270, 141, 41))
        self.combo.addItems(['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec'])
        self.combo.setCurrentIndex(6)
        self.combo.currentIndexChanged.connect(self.index_changed)
        
        
#####################################################################################
        
        
        self.TopLabel = QtWidgets.QLabel(self.centralwidget)
        self.TopLabel.setGeometry(QtCore.QRect(140, 10, 60, 16))
        self.TopLabel.setTextFormat(QtCore.Qt.RichText)
        self.TopLabel.setObjectName("TopLabel")
        self.label = QtWidgets.QLabel(self.centralwidget)
        self.label.setGeometry(QtCore.QRect(420, 10, 60, 16))
        self.label.setObjectName("label")
        self.label_2 = QtWidgets.QLabel(self.centralwidget)
        self.label_2.setGeometry(QtCore.QRect(710, 10, 60, 16))
        self.label_2.setObjectName("label_2")
        
#####################################################################################
        # Widgets for base display
        
        self.listWidget_1 = QtWidgets.QLabel(self.centralwidget)
        self.listWidget_1.setGeometry(QtCore.QRect(10, 370, 281, 300))
        self.listWidget_1.setObjectName("listWidget_1")
        self.listWidget_1.setPixmap(QtGui.QPixmap("ui_images/tshirts.jpg").scaled(281,300))
        self.listWidget_2 = QtWidgets.QLabel(self.centralwidget)
        self.listWidget_2.setGeometry(QtCore.QRect(300, 370, 281, 300))
        self.listWidget_2.setObjectName("listWidget_2")
        self.listWidget_2.setPixmap(QtGui.QPixmap("ui_images/pants.jpg").scaled(281,300))
        self.listWidget_3 = QtWidgets.QLabel(self.centralwidget)
        self.listWidget_3.setGeometry(QtCore.QRect(590, 370, 281, 300))
        self.listWidget_3.setObjectName("listWidget_3")
        self.listWidget_3.setPixmap(QtGui.QPixmap("ui_images/shoes.jpg").scaled(281,300))
        
#####################################################################################        
        # MainWindow widget
      
        MainWindow.setCentralWidget(self.centralwidget)
        self.menubar = QtWidgets.QMenuBar(MainWindow)
        self.menubar.setGeometry(QtCore.QRect(0, 0, 880, 22))
        self.menubar.setObjectName("menubar")
        MainWindow.setMenuBar(self.menubar)
        self.statusbar = QtWidgets.QStatusBar(MainWindow)
        self.statusbar.setObjectName("statusbar")
        MainWindow.setStatusBar(self.statusbar)

        self.retranslateUi(MainWindow)
        QtCore.QMetaObject.connectSlotsByName(MainWindow)
        #####################################################################################
        
        # Establish a connection between buttons and functions
        self.AddTopButton.clicked.connect(self.TOP_LIST_EDIT)
        self.DeleteTopButton.clicked.connect(self.TOP_LIST_DEL)
        
        self.AddBottomButton.clicked.connect(self.BOTTOM_LIST_EDIT)
        self.DeleteBottomButton.clicked.connect(self.BOTTOM_LIST_DEL)
        
        self.AddShoeButton.clicked.connect(self.SHOE_LIST_EDIT)
        self.DeleteShoeButton.clicked.connect(self.SHOE_LIST_DEL)
        self.AddOutfitButton.clicked.connect(self.PREDICT_OUTFIT)            
        self.GenerateButton.clicked.connect(self.Generate)         

        
        #####################################################################################

    def retranslateUi(self, MainWindow):
        """
        Mapping classes on GUI to the button names we provide 
        
        """
        _translate = QtCore.QCoreApplication.translate # base function used to translate MainWindow elements
        #Translate elements of Main UI class to text. Replacement done basis the object used to invoke the method.
        MainWindow.setWindowTitle(_translate("MainWindow", "Welcome to WardrobeAI!"))  
        self.AddTopButton.setText(_translate("MainWindow", "Edit outfit"))
        self.DeleteTopButton.setText(_translate("MainWindow", "Remove outfit"))

        self.AddBottomButton.setText(_translate("MainWindow", "Edit outfit "))
        self.DeleteBottomButton.setText(_translate("MainWindow", "Remove outfit"))

        self.AddShoeButton.setText(_translate("MainWindow", "Edit outfit "))
        self.DeleteShoeButton.setText(_translate("MainWindow", "Remove outfit"))

        self.GenerateButton.setText(_translate("MainWindow", "Generate outfit "))
        self.AddOutfitButton.setText(_translate("MainWindow", "Add outfit"))

        self.TopLabel.setText(_translate("MainWindow", "Topwear"))
        self.label.setText(_translate("MainWindow", "Bottomwear"))
        self.label_2.setText(_translate("MainWindow", "Shoes"))
        
def main():
    
    import sys
    app = QtWidgets.QApplication(sys.argv)
    splash = QSplashScreen(QtGui.QPixmap("ui_images/bg_closet.png"))
    splash.show()
    QTimer.singleShot(1000,splash.close)
    time.sleep(3)
    #Use Fusion style from Pyqt5 factory
    QtWidgets.QApplication.setStyle(QtWidgets.QStyleFactory.create('fusion'))
    MainWindow = QtWidgets.QMainWindow()
    # Setting colour and background colour for main window UI
    MainWindow.setStyleSheet("color: black;"
                             "selection-background-color: blue;"
                             "selection-color: white;"
                             "background-color: #AA96B7;")

    ui = Ui_MainWindow() #Initialize all widget elements and base GUI
    ui.placeWidgets(MainWindow) #Place widget elements
    MainWindow.show() #display whole GUI
    sys.exit(app.exec_())
    

if __name__ == "__main__":
    main()