mainMenu.py

import os
import sys
import cv2
from PyQt5 import QtWidgets
from PyQt5.QtWidgets import QApplication, QDialog
from PyQt5.uic import loadUi
from takePicPage import App
from page2 import App2
import face_recognition
import main
import RPi.GPIO as GPIO


video_capture = cv2.VideoCapture(main.url)
os.chdir('imgs')
for file in os.listdir():
    if file.endswith('.jpg'):
        this_img = face_recognition.load_image_file(file)
        if file == 'unknown.jpg':
            continue
        try:
            main.valid_imgs.append(this_img)
            main.valid_imgs_encodings.append(
                face_recognition.face_encodings(this_img)[0])
        except IndexError:
            main.showMessage("this image doesn't have any faces")
            continue

main.GPIOsetup()

class welcomeScreen(QDialog):
    def __init__(self):
        os.chdir('..')
        super(welcomeScreen, self).__init__()
        loadUi("welcomeScreen.ui", self)
        self._new_window = None
        self.member.clicked.connect(self.memberFunc)
        self.mode_1.clicked.connect(self.mode_1Func)
        self.mode_2.clicked.connect(self.mode_2Func)
        self.mode_3.clicked.connect(self.mode_3Func)
        os.chdir('imgs')

    def memberFunc(self):
        print("add member")
        main.qtStack.setCurrentIndex(1)
        print("Video Played")
    def mode_1Func(self):
        print("mode 1")
        main.mode = 1
        main.qtStack.setCurrentIndex(2)
        print("entered")
    def mode_2Func(self):
        print("mode 2")
        main.mode = 2
        main.qtStack.setCurrentIndex(2)
        print("entered")
    def mode_3Func(self):
        print("mode 3")
        main.mode = 3
        main.qtStack.setCurrentIndex(2)
        print("entered")


app = QApplication(sys.argv)
welcome = welcomeScreen()
main.qtStack = QtWidgets.QStackedWidget()
main.qtStack.addWidget(welcome)
main.qtStack.addWidget(App())
main.qtStack.addWidget(App2())
main.qtStack.setFixedHeight(400)
main.qtStack.setFixedWidth(600)
main.qtStack.show()
try:
    sys.exit(app.exec_())
except:
    print("Exiting")