orientation last

src/play_interface.py

@@ -1,6 +1,8 @@
 """Interface to play against the model.
 """
 
+from typing import Optional
+
 import huggingface_hub
 import chess
 import chess.svg
@@ -22,6 +24,7 @@ inference_fn = client.text_generation
 
 def plot_board(
     board: chess.Board,
+    orientation: bool = chess.WHITE,
 ):
     try:
         last_move = board.peek()
@@ -34,7 +37,7 @@ def plot_board(
         check = None
     svg_board = chess.svg.board(
         board,
-        orientation=board.turn,
+        orientation=orientation,
         check=check,
         size=350,
         arrows=arrows,
@@ -46,10 +49,13 @@ def plot_board(
 
 def render_board(
     current_board: chess.Board,
+    orientation: Optional[bool] = chess.WHITE,
 ):
     fen = current_board.fen()
     pgn = current_board.root().variation_san(current_board.move_stack)
-    image_board = plot_board(current_board)
+    if orientation is None:
+        orientation = current_board.turn
+    image_board = plot_board(current_board, orientation=orientation)
     return fen, pgn, "", image_board
 
 def play_user_move(
@@ -91,7 +97,7 @@ def try_play_move(
                     }
                 )
                 run.finish()
-            return *render_board(current_board), current_board
+            return *render_board(current_board, orientation=not current_board.turn), current_board
     except:
         gr.Warning("Invalid move")
         return *render_board(current_board), current_board