Upload app.py

app.py

@@ -58,18 +58,21 @@ def play_nn(curr_fen, model):
 
 def make_move(move):
     global board
-    if not board.outcome():
-        if len(move) > 3:
-            turn = "White"
-            new_turn = "Black"
-            if chess.Move.from_uci(move) in board.legal_moves:
-                board.push(chess.Move.from_uci(move))
-                ai_move = str(play_nn(board.fen(), model))
-                board.push(chess.Move.from_uci(ai_move))
-                message = f"{turn} move: {move},move accepted\nAI moved: {ai_move}\n\n{turn}'s move."
-            else:
-                message = f"{turn} move: Illegal move. Please try again."
-            return message, display_board(board)
+    if len(move) > 3:
+        turn = "White"
+        new_turn = "Black"
+        if chess.Move.from_uci(move) in board.legal_moves:
+            board.push(chess.Move.from_uci(move))
+            if board.outcome():
+                return f"White won", display_board(board)
+            ai_move = str(play_nn(board.fen(), model))
+            board.push(chess.Move.from_uci(ai_move))
+            if board.outcome():
+                return f"Black won", display_board(board)
+            message = f"{turn} move: {move},move accepted\nAI moved: {ai_move}\n\n{turn}'s move."
+        else:
+            message = f"{turn} move: Illegal move. Please try again."
+        return message, display_board(board)
     else:
         winner = "White" if board.outcome() == chess.WHITE else "Black"
         return f"{winner} won", display_board(board)
@@ -86,7 +89,7 @@ with gr.Blocks() as demo:
     gr.Markdown("## Chess board game")
     with gr.Row():
         with gr.Column():
-            move_input = gr.Textbox(placeholder="Enter your move in UCI format", lines=1)
+            move_input = gr.Textbox(placeholder="Enter your move in UCI format (e.g. e2e3)", lines=1)
             move_button = gr.Button("Make move")
             reset_button = gr.Button("Reset Board")
             status_output = gr.Textbox(label='Status', interactive=False)