import sys
from PyQt5 import QtWidgets
from PyQt5.QtWidgets import QApplication, QMainWindow
import time

class MyWindow(QMainWindow):

    def __init__(self):
        super(MyWindow, self).__init__()
        self.setGeometry(50, 50, 500, 500)
        self.setWindowTitle("Tic Tac Toe")
        self.initUI()
        self.turns()

    def initUI(self):
        self.label = QtWidgets.QLabel(self)
        self.label.setText(" ")
        self.label.move(0, 0)

        self.btn1 = QtWidgets.QPushButton(self)
        self.btn2 = QtWidgets.QPushButton(self)
        self.btn3 = QtWidgets.QPushButton(self)
        self.btn4 = QtWidgets.QPushButton(self)
        self.btn5 = QtWidgets.QPushButton(self)
        self.btn6 = QtWidgets.QPushButton(self)
        self.btn7 = QtWidgets.QPushButton(self)
        self.btn8 = QtWidgets.QPushButton(self)
        self.btn9 = QtWidgets.QPushButton(self)

        self.btn1.setText("  ")
        self.btn2.setText("  ")
        self.btn3.setText("  ")
        self.btn4.setText("  ")
        self.btn5.setText("  ")
        self.btn6.setText("  ")
        self.btn7.setText("  ")
        self.btn8.setText("  ")
        self.btn9.setText("  ")

        #20 -> top, 50 -> middle, 80 -> bottom
        self.btn1.move(0, 20)
        self.btn2.move(0, 50)
        self.btn3.move(0, 80)
        self.btn4.move(100, 20)
        self.btn5.move(100, 50)
        self.btn6.move(100, 80)
        self.btn7.move(200, 20)
        self.btn8.move(200, 50)
        self.btn9.move(200, 80)

        self.btn1.clicked.connect(self.clicked1)
        self.btn2.clicked.connect(self.clicked2)
        self.btn3.clicked.connect(self.clicked3)
        self.btn4.clicked.connect(self.clicked4)
        self.btn5.clicked.connect(self.clicked5)
        self.btn6.clicked.connect(self.clicked6)
        self.btn7.clicked.connect(self.clicked7)
        self.btn8.clicked.connect(self.clicked8)
        self.btn9.clicked.connect(self.clicked9)

    #assign each button a task upon clicking
    def clicked1(self):
        self.btn1.setText("X")
    def clicked2(self):
        self.btn2.setText("X")
    def clicked3(self):
        self.btn3.setText("X")
    def clicked4(self):
        self.btn4.setText("X")
    def clicked5(self):
        self.btn5.setText("X")
    def clicked6(self):
        self.btn6.setText("X")
    def clicked7(self):
        self.btn7.setText("X")
    def clicked8(self):
        self.btn8.setText("X")
    def clicked9(self):
        self.btn9.setText("X")


    #def a winner
    def winner_player(self):
        #top horiz
        if self.btn1.text() and self.btn4.text() and self.btn7.text() == "X":
            self.label.setText("Player Wins")
        #mid horiz
        if self.btn2.text() and self.btn5.text() and self.btn8.text() == "X":
            self.label.setText("Player Wins")
        #bot horiz
        if self.btn3.text() and self.btn6.text() and self.btn9.text() == "X":
            self.label.setText("Player Wins")
        #left vert
        if self.btn1.text() and self.btn2.text() and self.btn3.text() == "X":
            self.label.setText("Player Wins")
        #mid vert
        if self.btn4.text() and self.btn5.text() and self.btn6.text() == "X":
            self.label.setText("Player Wins")
        #right vert
        if self.btn2.text() and self.btn2.text() and self.btn2.text() == "X":
            self.label.setText("Player Wins")
        #left -> right diag
        if self.btn1.text() and self.btn5.text() and self.btn9.text() == "X":
            self.label.setText("Player Wins")
        #right -> left diag
        if self.btn3.text() and self.btn5.text() and self.btn7.text() == "X":
            self.label.setText("Player Wins")



    def winner_cpu(self):
        #top horiz
        if self.btn1.text() and self.btn4.text() and self.btn7.text() == "O":
            self.label.setText("CPU Wins")
        #mid horiz
        if self.btn2.text() and self.btn5.text() and self.btn8.text() == "O":
            self.label.setText("CPU Wins")
        #bot horiz
        if self.btn3.text() and self.btn6.text() and self.btn9.text() == "O":
            self.label.setText("CPU Wins")
        #left vert
        if self.btn1.text() and self.btn2.text() and self.btn3.text() == "O":
            self.label.setText("CPU Wins")
        #mid vert
        if self.btn4.text() and self.btn5.text() and self.btn6.text() == "O":
            self.label.setText("CPU Wins")
        #right vert
        if self.btn2.text() and self.btn2.text() and self.btn2.text() == "O":
            self.label.setText("CPU Wins")
        #left -> right diag
        if self.btn1.text() and self.btn5.text() and self.btn9.text() == "O":
            self.label.setText("CPU Wins")
        #right -> left diag
        if self.btn3.text() and self.btn5.text() and self.btn7.text() == "O":
            self.label.setText("CPU Wins")


    def turns(self):
        player_turn = True

        if player_turn == True:
            self.label_turn = QtWdigets.QLabel(self)
            self.label_turn.setText("Pick a box")
            self.label_turn.move(0, 0)

            time.sleep(3)

            self.winner_player()

            player_turn = False
            cpu_turn = True

        if cpu_turn == True:
            self.label_turn = QtWdigets.QLabel(self)
            self.label_turn.setText("Pick a box")
            self.label.move(0, 0)

            time.sleep(3)

            self.winner_cpu()

            player_turn = True
            cpu_turn = False

def window():
    app = QApplication(sys.argv)
    win = MyWindow()
    win.show()
    sys.exit(app.exec_())
window()