Update app.py

app.py

@@ -1,15 +1,15 @@
 import gradio as gr
 from PIL import Image, ImageDraw, ImageFont
-import random, time, os
+import random, time
 
 DIRECTIONS = [(-1,-1),(-1,0),(-1,1),(0,-1),(0,1),(1,-1),(1,0),(1,1)]
 
 history_log = []
 
-# Attempt to load larger font
+# Load custom fonts or fall back
 try:
-    FONT_L = ImageFont.truetype("DejaVuSans-Bold.ttf", 28)
-    FONT_S = ImageFont.truetype("DejaVuSans-Bold.ttf", 16)
+    FONT_L = ImageFont.truetype("DejaVuSans-Bold.ttf", 36)
+    FONT_S = ImageFont.truetype("DejaVuSans-Bold.ttf", 20)
 except:
     FONT_L = ImageFont.load_default()
     FONT_S = ImageFont.load_default()
@@ -53,19 +53,19 @@ def is_game_over(board):
 def board_to_image(board, state):
     board_state, player, last_user, last_ai, history = state
     size, cell = 400, 400//8
-    img=Image.new('RGB',(size,size+cell*4),'darkgreen')
+    img=Image.new('RGB',(size,size+cell*5),'darkgreen')
     draw=ImageDraw.Draw(img)
 
     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-180,2),f"WHITE: {w}",font=FONT_L,fill='white')
+    draw.text((size-200,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")
+        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')
+        draw.text((size//2-120,cell+2),winner,font=FONT_L,fill='yellow')
+        draw.text((size//2-120,cell+30),"Click 'New Game' to restart",font=FONT_S,fill='white')
 
     for r in range(8):
         for c in range(8):
@@ -82,8 +82,10 @@ def board_to_image(board, state):
             draw.rectangle([x0,y0,x1,y1],outline=clr,width=4)
 
     y=cell*2+8*cell+10
-    for res in history[-5:]:
-        draw.text((10,y),res,font=FONT_S,fill='white'); y+=cell//2
+    if history:
+        draw.text((10,y),"History:",font=FONT_S,fill='white'); y+=cell//2
+        for res in history[-5:]:
+            draw.text((10,y),res,font=FONT_S,fill='white'); y+=cell//2
 
     return img
 
@@ -91,7 +93,6 @@ def click_handler(evt, state):
     if state is None:
         state = (initialize_board(), -1, None, None, [])
     board, player, lu, la, history = state
-    # Handle evt.index callable vs attribute
     if callable(evt.index):
         x, y = evt.index()
     else:
@@ -106,7 +107,7 @@ def click_handler(evt, state):
     yield board_to_image(board, (board, player, lu, la, history)), (board, player, lu, la, history)
 
     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")
+        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)
@@ -133,5 +134,5 @@ with gr.Blocks() as demo:
     img=gr.Image(value=board_to_image(initialize_board(),(initialize_board(),-1,None,None,[])),interactive=True)
     new_btn=gr.Button("New Game")
     img.select(click_handler,inputs=[state],outputs=[img,state])
-    new_btn.click(fn=reset_handler,outputs=[state,img])
+    new_btn.click(fn=lambda: (initialize_board(), -1, None, None, history_log.copy()), outputs=[state, img])
     demo.launch()