Update app.py

app.py

@@ -1,9 +1,19 @@
 import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
-import random, time
+import random, time, os
 
 DIRECTIONS = [(-1,-1),(-1,0),(-1,1),(0,-1),(0,1),(1,-1),(1,0),(1,1)]
 
+history_log = []
+
+# Attempt to load larger font
+try:
+    FONT_L = ImageFont.truetype("DejaVuSans-Bold.ttf", 28)
+    FONT_S = ImageFont.truetype("DejaVuSans-Bold.ttf", 16)
+except:
+    FONT_L = ImageFont.load_default()
+    FONT_S = ImageFont.load_default()
+
 def initialize_board():
     b = [[0]*8 for _ in range(8)]
     b[3][3], b[4][4] = 1, 1
@@ -40,25 +50,23 @@ def count_score(board):
 def is_game_over(board):
     return not any(get_flips(board,r,c,p) for p in (-1,1) for r in range(8) for c in range(8))
 
-FONT_L = ImageFont.load_default()
-FONT_S = ImageFont.load_default()
-
 def board_to_image(board, state):
     board_state, player, last_user, last_ai, history = state
-    size, cell = 360, 360//8
-    img=Image.new('RGB',(size,size+cell*2),'darkgreen')
+    size, cell = 400, 400//8
+    img=Image.new('RGB',(size,size+cell*4),'darkgreen')
     draw=ImageDraw.Draw(img)
-    # Scoreboard
+
     b,w=count_score(board_state)
     draw.rectangle([0,0,size,cell],fill='navy')
     draw.text((10,2),f"BLACK: {b}",font=FONT_L,fill='white')
-    draw.text((size-160,2),f"WHITE: {w}",font=FONT_L,fill='white')
-    # Game over
+    draw.text((size-180,2),f"WHITE: {w}",font=FONT_L,fill='white')
+
     if is_game_over(board_state):
-        winner = "Draw" if b==w else ("Black Wins" if b>w else "White Wins")
-        draw.rectangle([0,cell,size,cell*2],fill='maroon')
-        draw.text((size//2-80,cell+2),winner,font=FONT_L,fill='yellow')
-    # Board
+        winner = "Draw" if b==w else ("BLACK WINS" if b>w else "WHITE WINS")
+        draw.rectangle([0,cell,size,cell*2],fill='darkred')
+        draw.text((size//2-110,cell+2),winner,font=FONT_L,fill='yellow')
+        draw.text((size//2-90,cell+30),"Click 'New Game' to restart",font=FONT_S,fill='white')
+
     for r in range(8):
         for c in range(8):
             x0,y0=c*cell,cell*2+r*cell; x1,y1=x0+cell,y0+cell
@@ -67,50 +75,54 @@ def board_to_image(board, state):
                 draw.ellipse([x0+cell*0.4,y0+cell*0.4,x0+cell*0.6,y0+cell*0.6],fill='yellow')
             if board_state[r][c]==1: draw.ellipse([x0+4,y0+4,x1-4,y1-4],fill='white')
             if board_state[r][c]==-1: draw.ellipse([x0+4,y0+4,x1-4,y1-4],fill='black')
-    # Last moves
+
     for mark,clr in ((last_user,'red'),(last_ai,'blue')):
         if mark:
             mr,mc=mark; x0,y0=mc*cell,cell*2+mr*cell; x1,y1=x0+cell,y0+cell
             draw.rectangle([x0,y0,x1,y1],outline=clr,width=4)
-    # History
-    y=cell*2+8*cell+2
+
+    y=cell*2+8*cell+10
     for res in history[-5:]:
         draw.text((10,y),res,font=FONT_S,fill='white'); y+=cell//2
+
     return img
 
 def click_handler(evt, state):
-    # Ensure state initialized
     if state is None:
         state = (initialize_board(), -1, None, None, [])
     board, player, lu, la, history = state
     x, y = evt.index
-    cell = 360 // 8
+    cell = 400 // 8
     c, r = int(x // cell), int((y - cell * 2) // cell)
-    # Player move
+
     if 0 <= r < 8 and 0 <= c < 8 and not is_game_over(board):
         if apply_move(board, r, c, player):
             lu, player = (r, c), -player
-    # Update after player
+
     yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
-    # Check end
+
     if is_game_over(board):
         winner = "Draw" if count_score(board)[0] == count_score(board)[1] else ("Black" if count_score(board)[0] > count_score(board)[1] else "White")
         history.append(f"Game {len(history)+1}: {winner}")
+        history_log.append(f"Game {len(history_log)+1}: {winner}")
         yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
         return
-    # AI thinking indicator
-    yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
-    time.sleep(2)
-    # AI move
+
+    thinking_img = board_to_image(board, (board, player, lu, la, history))
+    draw = ImageDraw.Draw(thinking_img)
+    draw.text((150, 400 - 25), "AI Thinking...", font=FONT_S, fill='cyan')
+    yield thinking_img, (board, player, lu, la, history)
+    time.sleep(1.5)
+
     ai_mv = choose_move(board, player)
     if ai_mv:
         apply_move(board, ai_mv[0], ai_mv[1], player)
         la, player = ai_mv, -player
     yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
 
-# def reset_handler placeholders for replacement
-#def reset_handler():
-    return initialize_board(),-1,None,None,[]
+def reset_handler():
+    new_board = initialize_board()
+    return new_board, -1, None, None, history_log.copy()
 
 with gr.Blocks() as demo:
     state=gr.State((initialize_board(),-1,None,None,[]))
@@ -118,4 +130,4 @@ with gr.Blocks() as demo:
     new_btn=gr.Button("New Game")
     img.select(click_handler,inputs=[state],outputs=[img,state])
     new_btn.click(fn=reset_handler,outputs=[state,img])
-    demo.launch()
+    demo.launch()