Update app.py

app.py

@@ -89,6 +89,49 @@ def create_gif(image_list, gif_path, duration):
         gif_path, save_all=True, append_images=pil_images[1:], duration=duration, loop=0
     )
 
+def render_init(board):
+    svg = chess.svg.board(board=board).encode("utf-8")
+    png = cairosvg.svg2png(bytestring=svg)
+    image = PILImage.open(io.BytesIO(png))
+    
+    # Calculate the size of the new image
+    width, height = image.size
+    new_width = 3 * width
+    
+    # Create a new blank image with the desired dimensions
+    new_image = PILImage.new('RGB', (new_width, height), 'white')
+    
+    # Calculate the position to paste the chess board image
+    left = width  # One third of the width
+    upper = 0
+    
+    # Paste the chess board image into the new image
+    new_image.paste(image, (left, upper))
+
+    return new_image
+
+def render_new(board):
+    last_move = board.peek()
+    svg = chess.svg.board(board=board, arrows=[(last_move.from_square, last_move.to_square)]).encode("utf-8")
+    png = cairosvg.svg2png(bytestring=svg)
+    image = PILImage.open(io.BytesIO(png))
+
+    # Calculate the size of the new image
+    width, height = image.size
+    new_width = 3 * width
+    
+    # Create a new blank image with the desired dimensions
+    new_image = PILImage.new('RGB', (new_width, height), 'white')
+    
+    # Calculate the position to paste the chess board image
+    left = width  # One third of the width
+    upper = 0
+    
+    # Paste the chess board image into the new image
+    new_image.paste(image, (left, upper))
+
+    return new_image
+    
 def save_result_file(
     pgn_id, model_id_white, model_id_black, termination, result, auth_token, gist_id
 ):
@@ -201,9 +244,7 @@ def update(model_id_white, model_id_black):
     result = None
 
     # Render first image
-    svg = chess.svg.board(board=board).encode("utf-8")
-    png = cairosvg.svg2png(bytestring=svg)
-    image = PILImage.open(io.BytesIO(png))
+    image = render_init(board)
     yield image
 
     # Time budget
@@ -248,10 +289,7 @@ def update(model_id_white, model_id_black):
             pgn_moves += move_str
 
             # Render the board to an image
-            last_move = board.peek()
-            svg = chess.svg.board(board=board, arrows=[(last_move.from_square, last_move.to_square)]).encode("utf-8")
-            png = cairosvg.svg2png(bytestring=svg)
-            image = PILImage.open(io.BytesIO(png))
+            image = render_new(board)
             board_images.append(np.array(image))
 
             # Deduct the time taken for the move from the model's time budget