Update app.py
Browse files
app.py
CHANGED
@@ -1,19 +1,16 @@
|
|
1 |
import gradio as gr
|
2 |
from PIL import Image, ImageDraw
|
3 |
|
4 |
-
# Othello (Reversi) logic implementation
|
5 |
def initialize_board():
|
6 |
board = [[0 for _ in range(8)] for _ in range(8)]
|
7 |
-
board[3][3], board[4][4] = 1, 1
|
8 |
-
board[3][4], board[4][3] = -1, -1
|
9 |
return board
|
10 |
|
11 |
-
# Directions for flipping
|
12 |
DIRECTIONS = [(-1, -1), (-1, 0), (-1, 1),
|
13 |
(0, -1), (0, 1),
|
14 |
(1, -1), (1, 0), (1, 1)]
|
15 |
|
16 |
-
# Check if a move is valid and collect flips
|
17 |
def get_flips(board, row, col, player):
|
18 |
if board[row][col] != 0:
|
19 |
return []
|
@@ -23,12 +20,12 @@ def get_flips(board, row, col, player):
|
|
23 |
buffer = []
|
24 |
while 0 <= r < 8 and 0 <= c < 8 and board[r][c] == -player:
|
25 |
buffer.append((r, c))
|
26 |
-
r += dr
|
|
|
27 |
if buffer and 0 <= r < 8 and 0 <= c < 8 and board[r][c] == player:
|
28 |
flips.extend(buffer)
|
29 |
return flips
|
30 |
|
31 |
-
# Apply move if valid
|
32 |
def apply_move(board, row, col, player):
|
33 |
flips = get_flips(board, row, col, player)
|
34 |
if not flips:
|
@@ -38,7 +35,6 @@ def apply_move(board, row, col, player):
|
|
38 |
board[r][c] = player
|
39 |
return True
|
40 |
|
41 |
-
# Generate PIL image representation
|
42 |
def board_to_image(board, img_size=320):
|
43 |
cell_size = img_size // 8
|
44 |
img = Image.new('RGB', (img_size, img_size), 'green')
|
@@ -54,39 +50,37 @@ def board_to_image(board, img_size=320):
|
|
54 |
draw.ellipse([x0+4, y0+4, x1-4, y1-4], fill='black')
|
55 |
return img
|
56 |
|
57 |
-
# Count score
|
58 |
def count_score(board):
|
59 |
black = sum(cell == -1 for row in board for cell in row)
|
60 |
white = sum(cell == 1 for row in board for cell in row)
|
61 |
return black, white
|
62 |
|
63 |
-
|
64 |
-
def play_move(evt, state):
|
65 |
board, player = state
|
66 |
img_size = 320
|
67 |
cell_size = img_size / 8
|
68 |
-
x, y = evt['x'], evt['y']
|
69 |
col, row = int(x // cell_size), int(y // cell_size)
|
|
|
|
|
70 |
if not apply_move(board, row, col, player):
|
71 |
return board_to_image(board), f"Invalid move at ({row+1},{col+1})", state
|
72 |
-
# Switch player
|
73 |
player = -player
|
74 |
black_score, white_score = count_score(board)
|
75 |
status = f"Black: {black_score} | White: {white_score} | {'Black' if player == -1 else 'White'} to move"
|
76 |
return board_to_image(board), status, (board, player)
|
77 |
|
78 |
-
# Initialize Gradio app
|
79 |
def main():
|
80 |
with gr.Blocks() as demo:
|
81 |
gr.HTML("<h2>Othello (Reversi) Game</h2>")
|
82 |
-
state = gr.State((initialize_board(), -1))
|
83 |
-
|
84 |
status = gr.Text(value="Black: 2 | White: 2 | Black to move", interactive=False)
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
|
|
89 |
demo.launch()
|
90 |
|
91 |
if __name__ == "__main__":
|
92 |
-
main()
|
|
|
1 |
import gradio as gr
|
2 |
from PIL import Image, ImageDraw
|
3 |
|
|
|
4 |
def initialize_board():
|
5 |
board = [[0 for _ in range(8)] for _ in range(8)]
|
6 |
+
board[3][3], board[4][4] = 1, 1
|
7 |
+
board[3][4], board[4][3] = -1, -1
|
8 |
return board
|
9 |
|
|
|
10 |
DIRECTIONS = [(-1, -1), (-1, 0), (-1, 1),
|
11 |
(0, -1), (0, 1),
|
12 |
(1, -1), (1, 0), (1, 1)]
|
13 |
|
|
|
14 |
def get_flips(board, row, col, player):
|
15 |
if board[row][col] != 0:
|
16 |
return []
|
|
|
20 |
buffer = []
|
21 |
while 0 <= r < 8 and 0 <= c < 8 and board[r][c] == -player:
|
22 |
buffer.append((r, c))
|
23 |
+
r += dr
|
24 |
+
c += dc
|
25 |
if buffer and 0 <= r < 8 and 0 <= c < 8 and board[r][c] == player:
|
26 |
flips.extend(buffer)
|
27 |
return flips
|
28 |
|
|
|
29 |
def apply_move(board, row, col, player):
|
30 |
flips = get_flips(board, row, col, player)
|
31 |
if not flips:
|
|
|
35 |
board[r][c] = player
|
36 |
return True
|
37 |
|
|
|
38 |
def board_to_image(board, img_size=320):
|
39 |
cell_size = img_size // 8
|
40 |
img = Image.new('RGB', (img_size, img_size), 'green')
|
|
|
50 |
draw.ellipse([x0+4, y0+4, x1-4, y1-4], fill='black')
|
51 |
return img
|
52 |
|
|
|
53 |
def count_score(board):
|
54 |
black = sum(cell == -1 for row in board for cell in row)
|
55 |
white = sum(cell == 1 for row in board for cell in row)
|
56 |
return black, white
|
57 |
|
58 |
+
def play_move(x, y, state):
|
|
|
59 |
board, player = state
|
60 |
img_size = 320
|
61 |
cell_size = img_size / 8
|
|
|
62 |
col, row = int(x // cell_size), int(y // cell_size)
|
63 |
+
if not (0 <= row < 8 and 0 <= col < 8):
|
64 |
+
return board_to_image(board), f"Click inside the board.", state
|
65 |
if not apply_move(board, row, col, player):
|
66 |
return board_to_image(board), f"Invalid move at ({row+1},{col+1})", state
|
|
|
67 |
player = -player
|
68 |
black_score, white_score = count_score(board)
|
69 |
status = f"Black: {black_score} | White: {white_score} | {'Black' if player == -1 else 'White'} to move"
|
70 |
return board_to_image(board), status, (board, player)
|
71 |
|
|
|
72 |
def main():
|
73 |
with gr.Blocks() as demo:
|
74 |
gr.HTML("<h2>Othello (Reversi) Game</h2>")
|
75 |
+
state = gr.State((initialize_board(), -1))
|
76 |
+
image_box = gr.Image(value=board_to_image(initialize_board()), interactive=True, label="Click to place a stone")
|
77 |
status = gr.Text(value="Black: 2 | White: 2 | Black to move", interactive=False)
|
78 |
+
|
79 |
+
def click_handler(evt: gr.SelectData, state):
|
80 |
+
return play_move(evt.index[0], evt.index[1], state)
|
81 |
+
|
82 |
+
image_box.select(click_handler, inputs=[state], outputs=[image_box, status, state])
|
83 |
demo.launch()
|
84 |
|
85 |
if __name__ == "__main__":
|
86 |
+
main()
|