GUI.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Image classification"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "#import libraries\n",
    "import os\n",
    "import keras\n",
    "from keras.preprocessing import image\n",
    "from keras.applications.imagenet_utils import decode_predictions, preprocess_input\n",
    "from keras.applications import vgg16\n",
    "from keras.models import Model\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "#model\n",
    "model = keras.applications.vgg16.VGG16(weights='imagenet', include_top=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Model: \"vgg16\"\n",
      "_________________________________________________________________\n",
      " Layer (type)                Output Shape              Param #   \n",
      "=================================================================\n",
      " input_1 (InputLayer)        [(None, 224, 224, 3)]     0         \n",
      "                                                                 \n",
      " block1_conv1 (Conv2D)       (None, 224, 224, 64)      1792      \n",
      "                                                                 \n",
      " block1_conv2 (Conv2D)       (None, 224, 224, 64)      36928     \n",
      "                                                                 \n",
      " block1_pool (MaxPooling2D)  (None, 112, 112, 64)      0         \n",
      "                                                                 \n",
      " block2_conv1 (Conv2D)       (None, 112, 112, 128)     73856     \n",
      "                                                                 \n",
      " block2_conv2 (Conv2D)       (None, 112, 112, 128)     147584    \n",
      "                                                                 \n",
      " block2_pool (MaxPooling2D)  (None, 56, 56, 128)       0         \n",
      "                                                                 \n",
      " block3_conv1 (Conv2D)       (None, 56, 56, 256)       295168    \n",
      "                                                                 \n",
      " block3_conv2 (Conv2D)       (None, 56, 56, 256)       590080    \n",
      "                                                                 \n",
      " block3_conv3 (Conv2D)       (None, 56, 56, 256)       590080    \n",
      "                                                                 \n",
      " block3_pool (MaxPooling2D)  (None, 28, 28, 256)       0         \n",
      "                                                                 \n",
      " block4_conv1 (Conv2D)       (None, 28, 28, 512)       1180160   \n",
      "                                                                 \n",
      " block4_conv2 (Conv2D)       (None, 28, 28, 512)       2359808   \n",
      "                                                                 \n",
      " block4_conv3 (Conv2D)       (None, 28, 28, 512)       2359808   \n",
      "                                                                 \n",
      " block4_pool (MaxPooling2D)  (None, 14, 14, 512)       0         \n",
      "                                                                 \n",
      " block5_conv1 (Conv2D)       (None, 14, 14, 512)       2359808   \n",
      "                                                                 \n",
      " block5_conv2 (Conv2D)       (None, 14, 14, 512)       2359808   \n",
      "                                                                 \n",
      " block5_conv3 (Conv2D)       (None, 14, 14, 512)       2359808   \n",
      "                                                                 \n",
      " block5_pool (MaxPooling2D)  (None, 7, 7, 512)         0         \n",
      "                                                                 \n",
      " flatten (Flatten)           (None, 25088)             0         \n",
      "                                                                 \n",
      " fc1 (Dense)                 (None, 4096)              102764544 \n",
      "                                                                 \n",
      " fc2 (Dense)                 (None, 4096)              16781312  \n",
      "                                                                 \n",
      " predictions (Dense)         (None, 1000)              4097000   \n",
      "                                                                 \n",
      "=================================================================\n",
      "Total params: 138,357,544\n",
      "Trainable params: 138,357,544\n",
      "Non-trainable params: 0\n",
      "_________________________________________________________________\n"
     ]
    }
   ],
   "source": [
    "model.summary()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "def load_image(path):\n",
    "    img = image.load_img(path, target_size=model.input_shape[1:3])\n",
    "    x = image.img_to_array(img)\n",
    "    x = np.expand_dims(x, axis=0)\n",
    "    x = preprocess_input(x)\n",
    "    return img, x\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "#Model for feature extraction\n",
    "feat_extractor = Model(inputs=model.input, outputs=model.get_layer(\"fc2\").output)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "keeping 183 images to analyze\n"
     ]
    }
   ],
   "source": [
    "#recursively crawl the folder specified by image_path looking for images of the extensions inside of image_extensions \n",
    "#and then limiting them to a random subset of maximum max_num_images images.\n",
    "\n",
    "images_path = 'dataset'\n",
    "image_extensions = ['.jpg', '.png', '.jpeg']   # case-insensitive (upper/lower doesn't matter)\n",
    "max_num_images = 1000\n",
    "\n",
    "images = [os.path.join(dp, f) for dp, dn, filenames in os.walk(images_path) for f in filenames if os.path.splitext(f)[1].lower() in image_extensions]\n",
    "if max_num_images < len(images):\n",
    "    images = [images[i] for i in sorted(random.sample(xrange(len(images)), max_num_images))]\n",
    "\n",
    "print(\"keeping %d images to analyze\" % len(images))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "finished extracting features for 183 images\n"
     ]
    }
   ],
   "source": [
    "#a loop which will open each image, extract its feature vector, and append it to a list called features which will contain our activations for each image\n",
    "import time\n",
    "tic = time.perf_counter()\n",
    "\n",
    "\n",
    "features = []\n",
    "for i, image_path in enumerate(images):\n",
    "    if i % 500 == 0:\n",
    "        toc = time.perf_counter()\n",
    "        elap = toc-tic;\n",
    "        tic = time.perf_counter()\n",
    "#         print(\"analyzing image %d / %d. Time: %4.4f seconds.\" % (i, len(images),elap))\n",
    "    img, x = load_image(image_path);\n",
    "    feat = feat_extractor.predict(x)[0]\n",
    "    features.append(feat)\n",
    "\n",
    "print('finished extracting features for %d images' % len(images))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "PCA(n_components=30)"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#Alone, these activations provide a good representation, but it is a good idea to do one more step before using these as our feature vectors, \n",
    "#which is to do a principal component analysis (PCA) to reduce the dimensionality of our feature vectors down to 30. We apply PCA for two reasons: \n",
    "#1) the 4096-bit feature vector may have some redundancy in it, such that multiple elements in the vector are highly correlated or similar. \n",
    "#This would skew similarity comparisons towards those over-represented features. \n",
    "#2) Operating over 4096 elements is inefficient both in terms of space/memory requirements and processor speed, and \n",
    "#it would be better for us if we can reduce the length of these vectors but maintain the same effective representation. \n",
    "#PCA allows us to do this by reducing the dimensionality down of the feature vectors from 4096 to much less, \n",
    "#but maintain a representation which is still faithful to the original data, by preserving the relative inter-point distance.\n",
    "#Thus, PCA reduces the amount of redundancy in our features (from duplicate or highly-correlated features), \n",
    "#speeds up computation over them, and reduces the amount of memory they take up.\n",
    "\n",
    "from sklearn.decomposition import PCA\n",
    "features = np.array(features)\n",
    "pca = PCA(n_components=30)\n",
    "pca.fit(features)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "pca_features = pca.transform(features)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {},
   "outputs": [],
   "source": [
    "from scipy.spatial import distance\n",
    "def get_closest_images(query_image_idx, num_results=10):\n",
    "    distances = [ distance.cosine(pca_features[query_image_idx], feat) for feat in pca_features ]\n",
    "    idx_closest = sorted(range(len(distances)), key=lambda k: distances[k])[1:num_results+1]\n",
    "    return idx_closest\n",
    "\n",
    "def get_concatenated_images(indexes, thumb_height):\n",
    "    thumbs = []\n",
    "    paths = []\n",
    "    for idx in indexes:\n",
    "        img = image.load_img(images[idx])\n",
    "        img = img.resize((int(img.width * thumb_height / img.height), thumb_height))\n",
    "        paths.append(images[idx])\n",
    "        thumbs.append(img)\n",
    "    concat_image = np.concatenate([np.asarray(t) for t in thumbs], axis=1)\n",
    "    return concat_image,paths\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## LOGIN WINDOW"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 25,
   "metadata": {},
   "outputs": [],
   "source": [
    "from PyQt5 import QtCore, QtGui, QtWidgets\n",
    "from PyQt5.QtWidgets import QFileDialog,QWidget\n",
    "from PyQt5.QtGui import QPixmap\n",
    "\n",
    "class Ui_login_Form(object):\n",
    "    \n",
    "    def openSearchwindow(self):\n",
    "        user_id = self.username_lineEdit.text()\n",
    "        password = self.password_lineEdit.text()\n",
    "        \n",
    "        users = ['Dhanush','Pradyumna','Satwik']\n",
    "        passwords = ['AIE19002','AIE9043','AIE19059']\n",
    "        \n",
    "        if user_id == users[0] and passwords[0] or users[1] and passwords[1] or users[2] and passwords[2]:\n",
    "            self.window = QtWidgets.QMainWindow()\n",
    "            self.ui = Ui_SearchWindow()\n",
    "            self.ui.setupUi(self.window)\n",
    "            self.window.show()\n",
    "        else:\n",
    "            self.caution_label.setText(\"Invalid Credentials\")\n",
    "        \n",
    "        \n",
    "        \n",
    "    def setupUi(self, Form):\n",
    "        Form.setObjectName(\"Form\")\n",
    "        Form.resize(773, 634)\n",
    "        self.gridLayout = QtWidgets.QGridLayout(Form)\n",
    "        self.gridLayout.setObjectName(\"gridLayout\")\n",
    "        self.widget = QtWidgets.QWidget(Form)\n",
    "        self.widget.setStyleSheet(\"\")\n",
    "        self.widget.setObjectName(\"widget\")\n",
    "        \n",
    "        self.img_label = QtWidgets.QLabel(self.widget)\n",
    "        self.img_label.setGeometry(QtCore.QRect(0, 0, 401, 611))\n",
    "        self.img_label.setStyleSheet(\"border-image: url(shopping.png);\\n\"\"border-top-left-radius :50px;\")\n",
    "        self.img_label.setText(\"\")\n",
    "        self.img_label.setObjectName(\"img_label\")\n",
    "\n",
    "        self.label_2 = QtWidgets.QLabel(self.widget)\n",
    "        self.label_2.setGeometry(QtCore.QRect(400, 0, 341, 611))\n",
    "        self.label_2.setStyleSheet(\"background-color: rgb(255, 255, 255);\\n\"\"border-bottom-right-radius :50px;\")\n",
    "        self.label_2.setText(\"\")\n",
    "        self.label_2.setObjectName(\"label_2\")\n",
    "\n",
    "        self.label_3 = QtWidgets.QLabel(self.widget)\n",
    "        self.label_3.setGeometry(QtCore.QRect(480, 50, 191, 51))\n",
    "        font = QtGui.QFont()\n",
    "        font.setPointSize(22)\n",
    "        font.setBold(True)\n",
    "        font.setWeight(75)\n",
    "        self.label_3.setFont(font)\n",
    "        self.label_3.setAlignment(QtCore.Qt.AlignCenter)\n",
    "        self.label_3.setObjectName(\"label_3\")\n",
    "\n",
    "        self.username_lineEdit = QtWidgets.QLineEdit(self.widget)\n",
    "        self.username_lineEdit.setGeometry(QtCore.QRect(420, 170, 300, 80))\n",
    "        font = QtGui.QFont()\n",
    "        font.setPointSize(15)\n",
    "        self.username_lineEdit.setFont(font)\n",
    "        self.username_lineEdit.setStyleSheet(\"background-color: rgb(255, 255, 255);\\n\"\"border:none;\\n\"\"border-bottom: 3px solid rgba(46,82,101,200);\\n\"\"color:rgba(0,0,0,240);\\n\"\"padding-bottom:7px;\")\n",
    "        self.username_lineEdit.setObjectName(\"username_lineEdit\")\n",
    "\n",
    "        self.password_lineEdit = QtWidgets.QLineEdit(self.widget)\n",
    "        self.password_lineEdit.setGeometry(QtCore.QRect(420, 340, 300, 80))\n",
    "        font = QtGui.QFont()\n",
    "        font.setPointSize(15)\n",
    "        self.password_lineEdit.setFont(font)\n",
    "        self.password_lineEdit.setStyleSheet(\"background-color: rgb(255, 255, 255);\\n\"\"border:none;\\n\"\"border-bottom: 3px solid rgba(46,82,101,200);\\n\"\"color:rgba(0,0,0,240);\\n\"\"padding-bottom:7px;\")\n",
    "        self.password_lineEdit.setEchoMode(QtWidgets.QLineEdit.Password)\n",
    "        self.password_lineEdit.setObjectName(\"password_lineEdit\")\n",
    "\n",
    "        self.login_pushButton = QtWidgets.QPushButton(self.widget)\n",
    "        self.login_pushButton.setGeometry(QtCore.QRect(480, 480, 181, 51))\n",
    "        font = QtGui.QFont()\n",
    "        font.setPointSize(15)\n",
    "        font.setBold(True)\n",
    "        font.setWeight(75)\n",
    "        self.login_pushButton.setFont(font)\n",
    "        self.login_pushButton.setStyleSheet(\"background-color: rgb(153, 105, 199);\\n\"\"border-radius:6px;\\n\"\"\")\n",
    "        self.login_pushButton.setObjectName(\"login_pushButton\")\n",
    "        self.login_pushButton.clicked.connect(self.openSearchwindow)\n",
    "        \n",
    "        self.caution_label = QtWidgets.QLabel(self.widget)\n",
    "        self.caution_label.setGeometry(QtCore.QRect(460, 550, 271, 31))\n",
    "        self.caution_label.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "        self.caution_label.setText(\"\")\n",
    "        self.caution_label.setObjectName(\"caution_label\")\n",
    "\n",
    "        self.label_2.raise_()\n",
    "        self.img_label.raise_()\n",
    "        self.label_3.raise_()\n",
    "        self.username_lineEdit.raise_()\n",
    "        self.password_lineEdit.raise_()\n",
    "        self.login_pushButton.raise_()\n",
    "        self.caution_label.raise_()\n",
    "        self.gridLayout.addWidget(self.widget, 0, 1, 1, 1)\n",
    "\n",
    "        self.retranslateUi(Form)\n",
    "        QtCore.QMetaObject.connectSlotsByName(Form)\n",
    "\n",
    "    def retranslateUi(self, Form):\n",
    "        _translate = QtCore.QCoreApplication.translate\n",
    "        Form.setWindowTitle(_translate(\"Form\", \"Form\"))\n",
    "        self.label_3.setText(_translate(\"Form\", \"Log In\"))\n",
    "        self.username_lineEdit.setPlaceholderText(_translate(\"Form\", \"User Name\"))\n",
    "        self.password_lineEdit.setPlaceholderText(_translate(\"Form\", \"Password\"))\n",
    "        self.login_pushButton.setText(_translate(\"Form\", \"Log In\"))\n",
    "\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    import sys\n",
    "    app = QtWidgets.QApplication(sys.argv)\n",
    "    Form = QtWidgets.QWidget()\n",
    "    ui = Ui_login_Form()\n",
    "    ui.setupUi(Form)\n",
    "    Form.show()\n",
    "    app.exec_()\n",
    "    del app\n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Search Window"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [],
   "source": [
    "from PyQt5 import QtCore, QtGui, QtWidgets\n",
    "from PyQt5.QtWidgets import QFileDialog,QWidget\n",
    "from PyQt5.QtGui import QPixmap\n",
    "\n",
    "class Ui_SearchWindow(object):\n",
    "            \n",
    "    def setupUi(self, SearchWindow):\n",
    "        SearchWindow.setObjectName(\"SearchWindow\")\n",
    "        SearchWindow.resize(865, 711)\n",
    "        self.centralwidget = QtWidgets.QWidget(SearchWindow)\n",
    "        self.centralwidget.setObjectName(\"centralwidget\")\n",
    "\n",
    "        self.img_label = QtWidgets.QLabel(self.centralwidget)\n",
    "        self.img_label.setGeometry(QtCore.QRect(10, 10, 431, 661))\n",
    "        self.img_label.setStyleSheet(\"border-image: url(shopping.png);\\n\"\"border-top-left-radius :50px;\")\n",
    "        self.img_label.setText(\"\")\n",
    "        self.img_label.setObjectName(\"img_label\")\n",
    "\n",
    "        self.label_2 = QtWidgets.QLabel(self.centralwidget)\n",
    "        self.label_2.setGeometry(QtCore.QRect(440, 10, 401, 651))\n",
    "        self.label_2.setStyleSheet(\"background-color: rgb(255, 255, 255);\\n\"\"border-bottom-right-radius :50px;\")\n",
    "        self.label_2.setText(\"\")\n",
    "        self.label_2.setObjectName(\"label_2\")\n",
    "\n",
    "        self.load_pushButton = QtWidgets.QPushButton(self.centralwidget)\n",
    "        self.load_pushButton.setGeometry(QtCore.QRect(510, 410, 251, 71))\n",
    "        font = QtGui.QFont()\n",
    "        font.setPointSize(15)\n",
    "        font.setBold(True)\n",
    "        font.setWeight(75)\n",
    "        self.load_pushButton.setFont(font)\n",
    "        self.load_pushButton.setStyleSheet(\"background-color: rgb(153, 105, 199);\\n\"\"border-radius:6px;\\n\"\"\")\n",
    "        self.load_pushButton.setObjectName(\"load_pushButton\")\n",
    "        self.load_pushButton.clicked.connect(self.loadimage)\n",
    "\n",
    "        self.search_pushButton = QtWidgets.QPushButton(self.centralwidget)\n",
    "        self.search_pushButton.setGeometry(QtCore.QRect(510, 520, 251, 71))\n",
    "        font = QtGui.QFont()\n",
    "        font.setPointSize(15)\n",
    "        font.setBold(True)\n",
    "        font.setWeight(75)\n",
    "        self.search_pushButton.setFont(font)\n",
    "        self.search_pushButton.setStyleSheet(\"background-color: rgb(153, 105, 199);\\n\"\"border-radius:6px;\\n\"\"\")\n",
    "        self.search_pushButton.setObjectName(\"search_pushButton\")\n",
    "        self.search_pushButton.clicked.connect(self.loadwindow)\n",
    "\n",
    "        self.product_display_label = QtWidgets.QLabel(self.centralwidget)\n",
    "        self.product_display_label.setGeometry(QtCore.QRect(480, 50, 321, 311))\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Ignored, QtWidgets.QSizePolicy.Ignored)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.product_display_label.sizePolicy().hasHeightForWidth())\n",
    "        self.product_display_label.setSizePolicy(sizePolicy)\n",
    "        self.product_display_label.setText(\"\")\n",
    "        self.product_display_label.setScaledContents(True)\n",
    "        self.product_display_label.setObjectName(\"product_display_label\")\n",
    "\n",
    "        SearchWindow.setCentralWidget(self.centralwidget)\n",
    "        self.menubar = QtWidgets.QMenuBar(SearchWindow)\n",
    "        self.menubar.setGeometry(QtCore.QRect(0, 0, 865, 26))\n",
    "        self.menubar.setObjectName(\"menubar\")\n",
    "        SearchWindow.setMenuBar(self.menubar)\n",
    "        self.statusbar = QtWidgets.QStatusBar(SearchWindow)\n",
    "        self.statusbar.setObjectName(\"statusbar\")\n",
    "        SearchWindow.setStatusBar(self.statusbar)\n",
    "\n",
    "        self.retranslateUi(SearchWindow)\n",
    "        QtCore.QMetaObject.connectSlotsByName(SearchWindow)\n",
    "        \n",
    "        \n",
    "    def loadimage(self):\n",
    "        file_name = QtWidgets.QFileDialog.getOpenFileName(None, 'Open file', 'F:\\\\',\"Image files (*.jpg *.png *.jpeg)\")\n",
    "        self.pixmap = QPixmap(file_name[0])\n",
    "        self.product_display_label.setPixmap(self.pixmap)\n",
    "        f = open(\"query_imagepath.txt\", \"w\")\n",
    "        f.write(str(file_name[0]))\n",
    "        f.close()\n",
    "        \n",
    "        with open('query_imagepath.txt') as f:\n",
    "            contents = f.read()\n",
    "            \n",
    "        new_image, x = load_image(str(contents))\n",
    "        new_features = feat_extractor.predict(x)\n",
    "\n",
    "        # project it into pca space\n",
    "        new_pca_features = pca.transform(new_features)[0]\n",
    "\n",
    "        # calculate its distance to all the other images pca feature vectors\n",
    "        distances = [ distance.cosine(new_pca_features, feat) for feat in pca_features ]\n",
    "        idx_closest = sorted(range(len(distances)), key=lambda k: distances[k])[0:10]  \n",
    "        results_image,paths_final = get_concatenated_images(idx_closest, 200)\n",
    "        \n",
    "        \n",
    "        \n",
    "        textfile = open(\"temp.txt\",\"w\")\n",
    "        for ele in paths_final:\n",
    "            textfile.write(ele+\"\\n\")\n",
    "        textfile.close()    \n",
    "        \n",
    "        \n",
    "    \n",
    "    def loadwindow(self):\n",
    "        self.window = QtWidgets.QMainWindow()\n",
    "        self.ui = Ui_loadWindow()\n",
    "        self.ui.setupUi(self.window)\n",
    "        self.window.show()\n",
    "        \n",
    "\n",
    "        \n",
    "    def retranslateUi(self, SearchWindow):\n",
    "        _translate = QtCore.QCoreApplication.translate\n",
    "        SearchWindow.setWindowTitle(_translate(\"SearchWindow\", \"SearchWindow\"))\n",
    "        self.load_pushButton.setText(_translate(\"SearchWindow\", \"Load Product\"))\n",
    "        self.search_pushButton.setText(_translate(\"SearchWindow\", \"Search\"))\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    import sys\n",
    "    app = QtWidgets.QApplication(sys.argv)\n",
    "    SearchWindow = QtWidgets.QMainWindow()\n",
    "    ui = Ui_SearchWindow()\n",
    "    ui.setupUi(SearchWindow)\n",
    "    SearchWindow.show()\n",
    "    app.exec_()\n",
    "    del app\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Load Window"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {},
   "outputs": [],
   "source": [
    "from PyQt5 import QtCore, QtGui, QtWidgets\n",
    "from PyQt5.QtCore import *\n",
    "from PyQt5.QtGui import *\n",
    "from PyQt5.QtWidgets import *\n",
    "counter = 0\n",
    "class Ui_loadWindow(object):\n",
    "    \n",
    "    def setupUi(self, loadWindow):\n",
    "        \n",
    "        loadWindow.setObjectName(\"loadWindow\")\n",
    "        loadWindow.resize(818, 289)\n",
    "        loadWindow.setStyleSheet(\"background-color: rgb(95, 114, 190);\")\n",
    "        self.centralwidget = QtWidgets.QWidget(loadWindow)\n",
    "        self.centralwidget.setObjectName(\"centralwidget\")\n",
    "        self.label = QtWidgets.QLabel(self.centralwidget)\n",
    "        self.label.setGeometry(QtCore.QRect(310, 10, 261, 101))\n",
    "        font = QtGui.QFont()\n",
    "        font.setFamily(\"Segoe UI Black\")\n",
    "        font.setPointSize(25)\n",
    "        font.setBold(True)\n",
    "        font.setWeight(75)\n",
    "\n",
    "        self.label.setFont(font)\n",
    "        self.label.setStyleSheet(\"color: rgb(254, 121, 199);\")\n",
    "        self.label.setObjectName(\"label\")\n",
    "\n",
    "        self.timer = QtCore.QTimer()\n",
    "        self.timer.timeout.connect(self.progress)\n",
    "        self.timer.start(35)\n",
    "\n",
    "        self.progressBar = QtWidgets.QProgressBar(self.centralwidget)\n",
    "        self.progressBar.setGeometry(QtCore.QRect(50, 150, 741, 41))\n",
    "        self.progressBar.setStyleSheet(\"QProgressBar {\\n\"\"\\n\"\"    background-color: rgb(98,114,164);\\n\"\"    border-style : none;\\n\"\"    border-radius : 10px;\\n\"\"   colour: rgb(200,200,200)\\n\"\"\\n\"\"}\\n\"\"QProgressBar::chunk{\\n\"\"\\n\"\"    background-color: qlineargradient(spread:pad, x1:0, y1:0, x2:1, y2:0, stop:0  #5f72be, stop:1 #9921e8);\\n\"\"    border-radius: 10px;\\n\"\"}\")\n",
    "        self.progressBar.setProperty(\"value\", 24)\n",
    "        self.progressBar.setObjectName(\"progressBar\")\n",
    "\n",
    "\n",
    "\n",
    "        loadWindow.setCentralWidget(self.centralwidget)\n",
    "        self.statusbar = QtWidgets.QStatusBar(loadWindow)\n",
    "        self.statusbar.setObjectName(\"statusbar\")\n",
    "        loadWindow.setStatusBar(self.statusbar)\n",
    "\n",
    "        self.retranslateUi(loadWindow)\n",
    "        QtCore.QMetaObject.connectSlotsByName(loadWindow)\n",
    "\n",
    "    def progress(self):\n",
    "        global counter\n",
    "        self.progressBar.setValue(counter)\n",
    "        if counter > 100:\n",
    "            self.timer.stop()\n",
    "            self.window = QtWidgets.QMainWindow()\n",
    "            self.ui = Ui_ProductWindow()\n",
    "            self.ui.setupUi(self.window)\n",
    "            self.window.show()\n",
    "            loadWindow.close()\n",
    "            \n",
    "            \n",
    "        counter += 1\n",
    "        \n",
    "\n",
    "    def retranslateUi(self, loadWindow):\n",
    "        _translate = QtCore.QCoreApplication.translate\n",
    "        loadWindow.setWindowTitle(_translate(\"loadWindow\", \"loadWindow\"))\n",
    "        self.label.setText(_translate(\"loadWindow\", \"Loading..\"))\n",
    "\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    import sys\n",
    "    app = QtWidgets.QApplication(sys.argv)\n",
    "    loadWindow = QtWidgets.QMainWindow()\n",
    "    ui = Ui_loadWindow()\n",
    "    ui.setupUi(loadWindow)\n",
    "    loadWindow.show()\n",
    "    app.exec_()\n",
    "    del app\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Product Display Window"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "fileName = 'Images Dataset Product Description.xlsx'\n",
    "data_only = pd.read_excel(fileName, index_col=False)\n",
    "data = pd.DataFrame(data_only)\n",
    "data.drop(data.columns[[0,2]], axis=1, inplace=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [],
   "source": [
    "def getLabelData(queryIMAGE):\n",
    "    fileName = 'Images Dataset Product Description.xlsx'\n",
    "    data_only = pd.read_excel(fileName, index_col=False)\n",
    "    data = pd.DataFrame(data_only)\n",
    "    data.drop(data.columns[[0,2]], axis=1, inplace=True)\n",
    "    \n",
    "    columns = ['PATH : ','PRODUCT : ', 'COLOR : ', 'TYPE : ', 'BRAND : ', 'RATING : ', 'PRICE : ']\n",
    "    datalist = data.values.tolist()\n",
    "    ansList = []\n",
    "    outString = ''\n",
    "    for rowList in datalist:\n",
    "        if queryIMAGE in rowList:\n",
    "            ansList = rowList\n",
    "              \n",
    "    return ansList\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "from PyQt5 import QtCore, QtGui, QtWidgets\n",
    "from PyQt5.QtWidgets import * \n",
    "from PyQt5.QtGui import * \n",
    "from PyQt5.QtCore import *\n",
    "\n",
    "class Ui_ProductWindow(QtWidgets.QWidget):\n",
    "    \n",
    "    global items_list\n",
    "    items_list = []\n",
    "    def opendialog(self):\n",
    "        sender_button = self.sender()  \n",
    "        product_index = str(sender_button.objectName()).split('_')[1]\n",
    "        quanities = ['1','2','3','4','5']\n",
    "        quantity, response = QtWidgets.QInputDialog.getItem(self, 'Add to cart', 'Quantity',quanities)\n",
    "        if response == False:\n",
    "            pass\n",
    "        else:\n",
    "            items_list.append([product_index,quantity])\n",
    "        f = open(\"cartdetails.txt\", \"w\")\n",
    "        for l in items_list:\n",
    "            for ele in l:\n",
    "                f.write(str(ele)+'\\t')\n",
    "            f.write('\\n')\n",
    "        f.close()\n",
    "    \n",
    "            \n",
    "                          \n",
    "    \n",
    "    def setupUi(self, ProductWindow):\n",
    "        ProductWindow.setObjectName(\"ProductWindow\")\n",
    "        ProductWindow.resize(889, 712)\n",
    "\n",
    "        self.centralwidget = QtWidgets.QWidget(ProductWindow)\n",
    "        self.centralwidget.setObjectName(\"centralwidget\")\n",
    "        self.centralwidget.setStyleSheet(\"background-color: rgb(181, 137, 214);\")\n",
    "        self.verticalLayout = QtWidgets.QVBoxLayout(self.centralwidget)\n",
    "        self.verticalLayout.setObjectName(\"verticalLayout\")\n",
    "\n",
    "        self.scrollArea = QtWidgets.QScrollArea(self.centralwidget)\n",
    "        self.scrollArea.setWidgetResizable(True)\n",
    "        self.scrollArea.setObjectName(\"scrollArea\")\n",
    "\n",
    "        self.scrollAreaWidgetContents = QtWidgets.QWidget()\n",
    "        self.scrollAreaWidgetContents.setGeometry(QtCore.QRect(0, 0, 844, 2585))\n",
    "        self.scrollAreaWidgetContents.setObjectName(\"scrollAreaWidgetContents\")\n",
    "\n",
    "        self.formLayout = QtWidgets.QFormLayout(self.scrollAreaWidgetContents)\n",
    "        self.formLayout.setObjectName(\"formLayout\")\n",
    "\n",
    "        self.pushButton_1 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_1.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_1.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_1.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_1.setObjectName(\"pushButton_1\")\n",
    "        self.formLayout.setWidget(0, QtWidgets.QFormLayout.LabelRole, self.pushButton_1)   \n",
    "        \n",
    "        self.pushButton_2 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_2.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_2.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_2.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_2.setObjectName(\"pushButton_2\")\n",
    "        self.formLayout.setWidget(1, QtWidgets.QFormLayout.LabelRole, self.pushButton_2)\n",
    "\n",
    "        self.pushButton_3 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Fixed, QtWidgets.QSizePolicy.Fixed)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_3.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_3.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_3.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_3.setObjectName(\"pushButton_3\")\n",
    "        self.formLayout.setWidget(2, QtWidgets.QFormLayout.LabelRole, self.pushButton_3)\n",
    "\n",
    "        self.pushButton_4 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Preferred)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_4.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_4.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_4.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_4.setObjectName(\"pushButton_4\")\n",
    "        self.formLayout.setWidget(3, QtWidgets.QFormLayout.LabelRole, self.pushButton_4)\n",
    "\n",
    "        self.pushButton_5 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Preferred)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_5.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_5.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_5.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_5.setObjectName(\"pushButton_5\")\n",
    "        self.formLayout.setWidget(4, QtWidgets.QFormLayout.LabelRole, self.pushButton_5)\n",
    "\n",
    "        self.pushButton_6 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Preferred)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_6.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_6.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_6.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_6.setObjectName(\"pushButton_6\")\n",
    "        self.formLayout.setWidget(5, QtWidgets.QFormLayout.LabelRole, self.pushButton_6)\n",
    "\n",
    "        self.pushButton_7 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Preferred)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_7.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_7.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_7.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_7.setObjectName(\"pushButton_7\")\n",
    "        self.formLayout.setWidget(6, QtWidgets.QFormLayout.LabelRole, self.pushButton_7)\n",
    "\n",
    "        self.pushButton_8 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Preferred)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_8.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_8.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_8.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_8.setObjectName(\"pushButton_8\")\n",
    "        self.formLayout.setWidget(7, QtWidgets.QFormLayout.LabelRole, self.pushButton_8)\n",
    "\n",
    "        self.pushButton_9 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Preferred)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_9.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_9.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_9.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_9.setObjectName(\"pushButton_9\")\n",
    "        self.formLayout.setWidget(8, QtWidgets.QFormLayout.LabelRole, self.pushButton_9)\n",
    "\n",
    "        self.pushButton_10 = QtWidgets.QPushButton(self.scrollAreaWidgetContents)\n",
    "        sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Minimum, QtWidgets.QSizePolicy.Preferred)\n",
    "        sizePolicy.setHorizontalStretch(0)\n",
    "        sizePolicy.setVerticalStretch(0)\n",
    "        sizePolicy.setHeightForWidth(self.pushButton_10.sizePolicy().hasHeightForWidth())\n",
    "        self.pushButton_10.setSizePolicy(sizePolicy)\n",
    "        self.pushButton_10.setMinimumSize(QtCore.QSize(250, 250))\n",
    "        self.pushButton_10.setObjectName(\"pushButton_10\")\n",
    "        self.formLayout.setWidget(9, QtWidgets.QFormLayout.LabelRole, self.pushButton_10)\n",
    "\n",
    "            \n",
    "        finals = []\n",
    "\n",
    "        with open('temp.txt', 'r') as file:\n",
    "            for row in file.readlines():\n",
    "                row = row.split('\\\\')\n",
    "                finals.append(row[1][:-1])\n",
    "                \n",
    "            \n",
    "       \n",
    "        self.pushButton_1.setStyleSheet(\"border-image:url(dataset/\"+finals[0]+\")\")\n",
    "        self.pushButton_2.setStyleSheet(\"border-image:url(dataset/\"+finals[1]+\")\")\n",
    "        self.pushButton_3.setStyleSheet(\"border-image:url(dataset/\"+finals[2]+\")\")\n",
    "        self.pushButton_4.setStyleSheet(\"border-image:url(dataset/\"+finals[3]+\")\")\n",
    "        self.pushButton_5.setStyleSheet(\"border-image:url(dataset/\"+finals[4]+\")\")\n",
    "        self.pushButton_6.setStyleSheet(\"border-image:url(dataset/\"+finals[5]+\")\")\n",
    "        self.pushButton_7.setStyleSheet(\"border-image:url(dataset/\"+finals[6]+\")\")\n",
    "        self.pushButton_8.setStyleSheet(\"border-image:url(dataset/\"+finals[7]+\")\")\n",
    "        self.pushButton_9.setStyleSheet(\"border-image:url(dataset/\"+finals[8]+\")\")\n",
    "        self.pushButton_10.setStyleSheet(\"border-image:url(dataset/\"+finals[9]+\")\")\n",
    "        \n",
    "        \n",
    "        self.pushButton_1.clicked.connect(self.opendialog)\n",
    "        self.pushButton_2.clicked.connect(self.opendialog)\n",
    "        self.pushButton_3.clicked.connect(self.opendialog)\n",
    "        self.pushButton_4.clicked.connect(self.opendialog)\n",
    "        self.pushButton_5.clicked.connect(self.opendialog)\n",
    "        self.pushButton_6.clicked.connect(self.opendialog)\n",
    "        self.pushButton_7.clicked.connect(self.opendialog)\n",
    "        self.pushButton_8.clicked.connect(self.opendialog)\n",
    "        self.pushButton_9.clicked.connect(self.opendialog)\n",
    "        self.pushButton_10.clicked.connect(self.opendialog)\n",
    "                    \n",
    "    \n",
    "        self.label_1 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_1.setObjectName(\"label_1\")\n",
    "        self.formLayout.setWidget(0, QtWidgets.QFormLayout.FieldRole, self.label_1)\n",
    "        self.label_1.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_2 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_2.setObjectName(\"label_2\")\n",
    "        self.formLayout.setWidget(1, QtWidgets.QFormLayout.FieldRole, self.label_2)\n",
    "        self.label_2.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_3 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_3.setObjectName(\"label_3\")\n",
    "        self.formLayout.setWidget(2, QtWidgets.QFormLayout.FieldRole, self.label_3)\n",
    "        self.label_3.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_4 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_4.setObjectName(\"label_4\")\n",
    "        self.formLayout.setWidget(3, QtWidgets.QFormLayout.FieldRole, self.label_4)\n",
    "        self.label_4.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_5 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_5.setObjectName(\"label_5\")\n",
    "        self.formLayout.setWidget(4, QtWidgets.QFormLayout.FieldRole, self.label_5)\n",
    "        self.label_5.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_6 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_6.setObjectName(\"label_6\")\n",
    "        self.formLayout.setWidget(5, QtWidgets.QFormLayout.FieldRole, self.label_6)\n",
    "        self.label_6.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_7 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_7.setObjectName(\"label_7\")\n",
    "        self.formLayout.setWidget(6, QtWidgets.QFormLayout.FieldRole, self.label_7)\n",
    "        self.label_7.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_8 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_8.setObjectName(\"label_8\")\n",
    "        self.formLayout.setWidget(7, QtWidgets.QFormLayout.FieldRole, self.label_8)\n",
    "        self.label_8.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_9 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_9.setObjectName(\"label_9\")\n",
    "        self.formLayout.setWidget(8, QtWidgets.QFormLayout.FieldRole, self.label_9)\n",
    "        self.label_9.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "\n",
    "        self.label_10 = QtWidgets.QLabel(self.scrollAreaWidgetContents)\n",
    "        self.label_10.setObjectName(\"label_10\")\n",
    "        self.formLayout.setWidget(9, QtWidgets.QFormLayout.FieldRole, self.label_10)\n",
    "        self.label_10.setStyleSheet(\"background-color: rgb(255, 255, 255);\")\n",
    "            \n",
    "        \n",
    "        \n",
    "\n",
    "        self.scrollArea.setWidget(self.scrollAreaWidgetContents)\n",
    "        self.verticalLayout.addWidget(self.scrollArea)\n",
    "\n",
    "        self.next_pushButton = QtWidgets.QPushButton(self.centralwidget)\n",
    "        self.next_pushButton.setMinimumSize(QtCore.QSize(0, 50))\n",
    "        self.next_pushButton.setStyleSheet(\"background-color: rgb(153, 105, 199);\\n\"\"border-radius:6px;\\n\"\"\")\n",
    "        self.next_pushButton.setObjectName(\"next_pushButton\")\n",
    "        self.next_pushButton.clicked.connect(self.viewsummary)\n",
    "        \n",
    "        \n",
    "        self.verticalLayout.addWidget(self.next_pushButton)\n",
    "\n",
    "        ProductWindow.setCentralWidget(self.centralwidget)\n",
    "        self.menubar = QtWidgets.QMenuBar(ProductWindow)\n",
    "        self.menubar.setGeometry(QtCore.QRect(0, 0, 889, 26))\n",
    "        self.menubar.setObjectName(\"menubar\")\n",
    "        ProductWindow.setMenuBar(self.menubar)\n",
    "        self.statusbar = QtWidgets.QStatusBar(ProductWindow)\n",
    "        self.statusbar.setObjectName(\"statusbar\")\n",
    "        ProductWindow.setStatusBar(self.statusbar)\n",
    "        self.retranslateUi(ProductWindow)\n",
    "        QtCore.QMetaObject.connectSlotsByName(ProductWindow)\n",
    "        \n",
    "    \n",
    "        _translate = QtCore.QCoreApplication.translate\n",
    "\n",
    "        self.label_1.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[0])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[0])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[0])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[0])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[0])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[0])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_2.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[1])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[1])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[1])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[1])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[1])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[1])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_3.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[2])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[2])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[2])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[2])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[2])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[2])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_4.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[3])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[3])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[3])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[3])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[3])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[3])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_5.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[4])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[4])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[4])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[4])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[4])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[4])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_6.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[5])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[5])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[5])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[5])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[5])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[5])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_7.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[6])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[6])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[6])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[6])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[6])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[6])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_8.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[7])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[7])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[7])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[7])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[7])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[7])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_9.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[8])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[8])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[8])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[8])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[8])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[8])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "        self.label_10.setText(_translate(\"ProductWindow\",\"<html><head/><body><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRODUCT : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[9])[1] +\n",
    "                                      \"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">COLOUR : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[9])[2] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"TYPE : </span><span style=\\\" font-size:12pt;\\\">\"+ getLabelData(finals[9])[3] +\"</span></p><p align=\\\"justify\\\"><span style=\\\" font-size:12pt; font-weight:600;\\\">BRAND : </span><span style=\\\" font-size:12pt;\\\">\"\n",
    "                                      \"\"+ getLabelData(finals[9])[4] +\"</span></p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">\"\n",
    "                                      \"RATING : </span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[9])[5]) +\"</span>\"\n",
    "                                      \"</p><p><span style=\\\" font-size:12pt; font-weight:600;\\\">PRICE : \"\n",
    "                                      \"</span><span style=\\\" font-size:12pt;\\\">\"+ str(getLabelData(finals[9])[6]) +\"</span></p></body></html>\"))\n",
    "        \n",
    "    \n",
    "\n",
    "        \n",
    "    def viewsummary(self):\n",
    "        cartWindow.main_window()\n",
    "        \n",
    "        \n",
    "\n",
    "    def retranslateUi(self, ProductWindow):\n",
    "        _translate = QtCore.QCoreApplication.translate\n",
    "        ProductWindow.setWindowTitle(_translate(\"ProductWindow\", \"ProductWindow\"))\n",
    "        self.pushButton_1.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_2.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_3.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_4.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_5.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_6.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_7.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_8.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_9.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.pushButton_10.setText(_translate(\"ProductWindow\", \"\"))\n",
    "        self.next_pushButton.setText(_translate(\"ProductWindow\", \"Proceed to Buy\"))\n",
    "\n",
    "\n",
    "if __name__ == \"__main__\":\n",
    "    import sys\n",
    "    app = QtWidgets.QApplication(sys.argv)\n",
    "    ProductWindow = QtWidgets.QMainWindow()\n",
    "    ui = Ui_ProductWindow()\n",
    "    ui.setupUi(ProductWindow)\n",
    "    ProductWindow.show()\n",
    "    app.exec_()\n",
    "    del app\n",
    "\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Cart"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "global cartDFFinal\n",
    "def readProductDetails(cartDetailsFile, finalDetailsFile):\n",
    "    cartdetailsLines = []\n",
    "    with open(cartDetailsFile, 'r') as file:\n",
    "        cartdetailsLines =  file.read()\n",
    "    cartdetailsLineStrings = cartdetailsLines.split('\\n')\n",
    "    cartdetailsProductStrings = []\n",
    "    for strDetails in cartdetailsLineStrings:\n",
    "        cartdetailsProductStrings.append(strDetails.split('\\t'))\n",
    "    cartdetailsProductStrings.pop()\n",
    "    cartdetails = []\n",
    "    for st in cartdetailsProductStrings:\n",
    "        cartdetails.append(list(st[0:2]))\n",
    "    finals = []\n",
    "    with open(finalDetailsFile, 'r') as file:\n",
    "        for row in file.readlines():\n",
    "            row = row.split('\\\\')\n",
    "            finals.append(row[1][:-1])\n",
    "    return cartdetails, finals\n",
    "\n",
    "def getBillDetails(cartDetailsFile, finalDetailsFile):     \n",
    "    cartDETAILS, finalsLIST = readProductDetails(cartDetailsFile, finalDetailsFile)\n",
    "    fileName = 'Images Dataset Product Description.xlsx'\n",
    "    data_only = pd.read_excel(fileName, index_col=False)\n",
    "    data = pd.DataFrame(data_only)\n",
    "    data.drop(data.columns[[0]], axis=1, inplace=True)\n",
    "    dataLISTS = data.values.tolist()  \n",
    "    billDetails = []\n",
    "    for l in cartDETAILS:        \n",
    "        finalsIndex = int(l[0]) - 1\n",
    "        queryIMAGE = finalsLIST[finalsIndex]\n",
    "        ansList = []   \n",
    "        for rowList in dataLISTS:\n",
    "            if queryIMAGE in rowList:\n",
    "                ansList = rowList\n",
    "        smallList = ansList[2:6]\n",
    "        smallList.append(ansList[7])\n",
    "        quantity = int(l[1])\n",
    "        smallList.append(quantity)\n",
    "        amt = int(ansList[7].split(' ')[1])\n",
    "        totalPrice ='₹ '+str(quantity*amt)+' /-'\n",
    "        smallList.append(totalPrice)\n",
    "        billDetails.append(smallList)                \n",
    "    return billDetails\n",
    "\n",
    "cartFile  = 'cartdetails.txt'\n",
    "finalFile = 'temp.txt'\n",
    "bill = getBillDetails(cartFile, finalFile)\n",
    "cartDF = pd.DataFrame(bill, columns = ['PRODUCT', 'COLOR', 'TYPE', 'BRAND', 'PRICE', 'QUANTITY', 'TOTAL PRICE'])\n",
    "totalBillQuantityListValues = cartDF['QUANTITY'].to_list()\n",
    "totalBillQuantity = sum(totalBillQuantityListValues)\n",
    "totalBillQuantityString = str(totalBillQuantity)\n",
    "totalBillAmountList = cartDF['TOTAL PRICE'].to_list()\n",
    "totalBillAmountListValues= [int(a.split(' ')[1]) for a in totalBillAmountList]\n",
    "totalBillAmount = sum(totalBillAmountListValues)\n",
    "totalBillAmountString = '₹ '+str(totalBillAmount)+' /-'\n",
    "withFinalPrice = [' ', ' ', ' ', ' ', 'TOTAL : ', totalBillQuantityString, totalBillAmountString]\n",
    "bill.append(withFinalPrice)\n",
    "cartDFFinal = pd.DataFrame(bill, columns = ['PRODUCT', 'COLOR', 'TYPE', 'BRAND', 'PRICE', 'QUANTITY', 'TOTAL PRICE'])\n",
    "\n",
    "\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {},
   "outputs": [],
   "source": [
    "import sys\n",
    "from PyQt5 import QtCore, QtGui, QtWidgets\n",
    "from PyQt5.QtCore import *\n",
    "from PyQt5.QtGui import *\n",
    "from PyQt5.QtWidgets import *\n",
    "    \n",
    "    \n",
    "    \n",
    "class cartWindow(QWidget):\n",
    "    \n",
    "    def __init__(self):\n",
    "        QWidget.__init__(self)\n",
    "\n",
    "        tablemodel = pandasModel(cartDFFinal)\n",
    "        tableview = QTableView()\n",
    "        tableview.setModel(tablemodel)\n",
    "\n",
    "        layout = QVBoxLayout(self)\n",
    "        layout.addWidget(tableview)\n",
    "        self.setLayout(layout)\n",
    "        \n",
    "    def main_window(self):\n",
    "        self.show()   \n",
    "        \n",
    "    \n",
    "        \n",
    "\n",
    "class pandasModel(QAbstractTableModel):\n",
    "\n",
    "    def __init__(self, data):\n",
    "        QAbstractTableModel.__init__(self)\n",
    "        self._data = data\n",
    "\n",
    "    def rowCount(self, parent=None):\n",
    "        return self._data.shape[0]\n",
    "\n",
    "    def columnCount(self, parnet=None):\n",
    "        return self._data.shape[1]\n",
    "\n",
    "    def data(self, index, role=Qt.DisplayRole):\n",
    "        if index.isValid():\n",
    "            if role == Qt.DisplayRole:\n",
    "                return str(self._data.iloc[index.row(), index.column()])\n",
    "        return None\n",
    "\n",
    "    def headerData(self, col, orientation, role):\n",
    "        if orientation == Qt.Horizontal and role == Qt.DisplayRole:\n",
    "            return self._data.columns[col]\n",
    "        return None\n",
    "    \n",
    "    \n",
    "    \n",
    "\n",
    "\n",
    "if __name__ == '__main__':\n",
    "    app = QtWidgets.QApplication(sys.argv)\n",
    "    cartWindow = cartWindow()\n",
    "    cartWindow.resize(1000,500)\n",
    "    cartWindow.show()\n",
    "    app.exec_()\n",
    "    del app\n",
    "    \n",
    "    \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.8.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}