improve interface

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import random
 import chess
 import chess.svg
-from transformers import AutoModelForSequenceClassification, AutoTokenizer, AutoConfig, pipeline
+from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
 token = os.environ['auth_token']
 
@@ -44,38 +44,58 @@ def predict_move(fen, top_k=3):
     # discard illegal moves (https://python-chess.readthedocs.io/en/latest/core.html#chess.Board.legal_moves), then select top_k
     return p['label']
 
-def play_yolochess(inp_color, inp_notation, inp_move, inp_custom_fen, state):
+def btn_load(inp_fen):
     global board
-    if inp_custom_fen:
-        board = chess.Board(fen=inp_custom_fen)
-    if (inp_color == 'white' and board.turn == chess.BLACK) or (inp_color == 'black' and board.turn == chess.WHITE):
-        move = predict_move(board.fen())
-        board.push_uci(move)
+    
+    if inp_fen:
+        board = chess.Board(inp_fen)
     else:
-        if inp_move:
-            if inp_notation == 'UCI':
-                board.push_uci(inp_move)
-            if inp_notation == 'SAN':
-                board.push_san(inp_move)
+        board.reset()
+    
     with open('board.svg', 'w') as f:
         f.write(str(chess.svg.board(board)))
-    print(state)
     return 'board.svg', board.fen()
 
-iface = gr.Interface(
-    fn=play_yolochess, 
-    inputs=[
-        gr.Radio(["white", "black"], value="white", label='human player color'),
-        gr.Radio(["SAN", "UCI"], value="SAN", label='move notation'),
-        gr.Textbox(label='human player move'),
-        gr.Textbox(placeholder=board.fen(), label='starting position FEN'),
-        "state"
-    ], 
-    outputs=[
-        gr.Image(label='position'),
-        "state"
-    ],
-    title='Play YoloChess - Policy Network v0.3',
-    allow_flagging="never",
-)
-iface.launch()
+def btn_play(inp_move, inp_notation):
+    global board
+    
+    if inp_move:
+        if inp_notation == 'UCI': mv = chess.Move.from_uci(inp_move) #board.push_uci(inp_move)
+        elif inp_notation == 'SAN': mv = board.parse_san(inp_move) #chess.Move.from_san(inp_move) #board.push_san(inp_move)
+    else:
+        mv = chess.Move.from_uci(predict_move(board.fen()))
+    
+    board.push(mv)
+        
+    with open('board.svg', 'w') as f:
+        f.write(str(chess.svg.board(board, lastmove=mv)))
+    
+    return 'board.svg', board.fen(), ''
+
+with gr.Blocks() as block:
+    gr.Markdown('# Play YoloChess - Policy Network v0.3')
+    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')
+            with gr.Row():
+                load_btn = gr.Button("Load")
+                play_btn = gr.Button("Play")
+        with gr.Column():
+            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])
+    
+    gr.Markdown(
+        '''
+        - Click "Load" button to start and reset board.  
+        - Click "Play" button to get Engine move.  
+        - Enter a "human player move" in UCI or SAN notation and click "Play" to move a piece.  
+        - Output "ERROR" generally occurs on illegal moves (Human or Engine).
+        - Enter "FEN" to start from a custom position.
+        '''
+    )
+
+block.launch()