main.py

from SolveSudoku import isSafe, FindEmptyLocation
import Boards
from Box import Box
import pygame
pygame.init()


class GUI:

    def __init__(self, grid, width, height, window):
        self.height = height
        self.width = width
        self.window = window
        self.grid = grid
        self.rows = 9
        self.cols = 9
        self.boxes = [[Box(grid[i][j], i, j, self.width, self.height) for j in range(self.cols)]
            for i in range(self.rows)]

    def draw_grid(self):
        space = self.width / 9
        for i in range(self.rows + 1):
            if i % 3 == 0 and i != 0:
                thickness = 4
            else:
                thickness = 1
            pygame.draw.line(self.window, (255, 255, 255), (0, i * space), (self.width, i * space), thickness)
            pygame.draw.line(self.window, (255, 255, 255), (i * space, 0), (i * space, self.height), thickness)

        for i in range(self.rows):
            for j in range(self.cols):
                self.boxes[i][j].draw_boxes(self.window)

    def solve(self):
        box = FindEmptyLocation(self.grid)
        if box:
            r, c = box
        else:
            return True
        for i in range(1, 10):
            if isSafe(self.grid, r, c, i):
                self.grid[r][c] = i
                self.boxes[r][c].add_num(i)
                self.boxes[r][c].update_box(self.window, True)
                pygame.display.update()
                pygame.time.delay(100)

                if self.solve():
                    return True
                self.grid[r][c] = 0
                self.boxes[r][c].add_num(0)
                self.boxes[r][c].update_box(self.window, False)
                pygame.display.update()
                pygame.time.delay(100)
        return False



def redrawWindow(win, grid):
    win.fill((0, 0, 0))
    grid.draw_grid()


def main():
    g = Boards.get_board()

    window = pygame.display.set_mode((540, 600))
    pygame.display.set_caption("Sudoku")
    grid = GUI(g, 540, 540, window)

    run = True
    while run:

        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                run = False
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_SPACE:
                    grid.solve()

        redrawWindow(window, grid)
        pygame.display.update()


main()
pygame.quit()