Update app.py

app.py

@@ -16,7 +16,6 @@ board_split = ' | '
 nums = {str(n): empty_field * n for n in range(1, 9)}
 nums_rev = {v:k for k,v in reversed(nums.items())}
 
-board = chess.Board()
 
 def encode_fen(fen):
     # decompress fen representation
@@ -45,19 +44,17 @@ def predict_move(fen, top_k=3):
     return p['label']
 
 def btn_load(inp_fen):
-    global board
-    
     if inp_fen:
         board = chess.Board(inp_fen)
     else:
-        board.reset()
+        board = chess.Board()
     
     with open('board.svg', 'w') as f:
         f.write(str(chess.svg.board(board)))
     return 'board.svg', board.fen()
 
-def btn_play(inp_move, inp_notation):
-    global board
+def btn_play(inp_fen, inp_move, inp_notation):
+    board = chess.Board(inp_fen)
     
     if inp_move:
         if inp_notation == 'UCI': mv = chess.Move.from_uci(inp_move) #board.push_uci(inp_move)
@@ -73,12 +70,17 @@ def btn_play(inp_move, inp_notation):
     return 'board.svg', board.fen(), ''
 
 with gr.Blocks() as block:
-    gr.Markdown('# Play YoloChess - Policy Network v0.3')
+    gr.Markdown(
+    '''
+    # Play YoloChess - Policy Network v0.3
+    110M Parameter Transformer (BERT-base architecture) trained for text classification from scratch on expert games in FEN notation. 
+    '''
+    )
     with gr.Row() as row:
         with gr.Column():
             move = gr.Textbox(label='human player move')
             notation = gr.Radio(["SAN", "UCI"], value="SAN", label='move notation')
-            fen = gr.Textbox(placeholder=board.fen(), label='FEN')
+            fen = gr.Textbox(value=chess.Board().fen(), label='FEN')
             with gr.Row():
                 load_btn = gr.Button("Load")
                 play_btn = gr.Button("Play")
@@ -86,7 +88,7 @@ with gr.Blocks() as block:
             position_output = gr.Image(label='board')
 
     load_btn.click(fn=btn_load, inputs=fen, outputs=[position_output, fen])
-    play_btn.click(fn=btn_play, inputs=[move, notation], outputs=[position_output, fen, move])
+    play_btn.click(fn=btn_play, inputs=[fen, move, notation], outputs=[position_output, fen, move])
     
     gr.Markdown(
         '''