state output

src/play_interface.py

@@ -78,7 +78,7 @@ def try_play_move(
 ):
     if current_board.is_game_over():
         gr.Warning("The game is already over")
-        return render_board(current_board)
+        return *render_board(current_board), current_board
     try:
         current_board = play_user_move(move_to_play, current_board)
         if current_board.is_game_over():
@@ -92,10 +92,10 @@ def try_play_move(
                     }
                 )
                 run.finish()
-            return render_board(current_board)
+            return *render_board(current_board), current_board
     except:
         gr.Warning("Invalid move")
-        return render_board(current_board)
+        return *render_board(current_board), current_board
     temperature_retries = [
         (i+1)/10 for i in range(10)
     ]
@@ -108,7 +108,7 @@ def try_play_move(
     else:
         gr.Warning("AI move failed with all temperatures")
         current_board.pop()
-        return render_board(current_board)
+        return *render_board(current_board), current_board
 
 
 with gr.Blocks() as interface:
@@ -161,4 +161,4 @@ with gr.Blocks() as interface:
         inputs=[*static_inputs, state_board],
         outputs=[*static_outputs, gr.State()],
     )
-    interface.load(render_board, inputs=[state_board], outputs=[*static_outputs, gr.State()])
+    interface.load(render_board, inputs=[state_board], outputs=[*static_outputs])