Update src/streamlit_app.py

src/streamlit_app.py

@@ -146,24 +146,61 @@ if st.button("Reset Game"):
     st.session_state.history = []
     st.rerun()
 
-board_placeholder = st.empty()
+# Create two columns for layout
+col_board, col_pgn = st.columns([2, 1])
 
-def render_board():
-    try:
-        last_move = board.peek() if board.move_stack else None
-        svg_board = chess.svg.board(board=board, lastmove=last_move, size=400)
-        board_placeholder.markdown(f'<div style="display: flex; justify-content: center;">{svg_board}</div>', unsafe_allow_html=True)
-    except Exception as e:
-        st.error(f"Error rendering board: {e}")
+with col_board:
+    board_placeholder = st.empty()
+    
+    def render_board():
+        try:
+            last_move = board.peek() if board.move_stack else None
+            svg_board = chess.svg.board(board=board, lastmove=last_move, size=400)
+            board_placeholder.markdown(f'<div style="display: flex; justify-content: center;">{svg_board}</div>', unsafe_allow_html=True)
+        except Exception as e:
+            st.error(f"Error rendering board: {e}")
 
-render_board()
+    render_board()
+
+    col1, col2 = st.columns(2)
+    with col1:
+        st.write(f"**Turn:** {'White' if board.turn == chess.WHITE else 'Black'}")
+    with col2:
+        if board.is_check():
+            st.write("**Check!**")
 
-col1, col2 = st.columns(2)
-with col1:
-    st.write(f"**Turn:** {'White' if board.turn == chess.WHITE else 'Black'}")
-with col2:
-    if board.is_check():
-        st.write("**Check!**")
+with col_pgn:
+    st.write("### Game History")
+    pgn_placeholder = st.empty()
+    
+    def render_pgn():
+        if history:
+            try:
+                game = chess.pgn.Game()
+                game.headers["Event"] = "Human vs o2"
+                game.headers["White"] = "Human"
+                game.headers["Black"] = "o2" if agent_loaded else "Random AI"
+                node = game
+                temp_board = chess.Board()
+                for uci in history:
+                    move = chess.Move.from_uci(uci)
+                    if move in temp_board.legal_moves:
+                        node = node.add_main_variation(move)
+                        temp_board.push(move)
+                    else:
+                        break
+                pgn_placeholder.code(str(game), language="pgn")
+            except Exception as e:
+                move_pairs = []
+                for i in range(0, len(history), 2):
+                    white_move = history[i]
+                    black_move = history[i+1] if i+1 < len(history) else ""
+                    move_pairs.append(f"{i//2 + 1}. {white_move} {black_move}")
+                pgn_placeholder.code("\n".join(move_pairs))
+        else:
+            pgn_placeholder.text("No moves yet")
+    
+    render_pgn()
 
 if not board.is_game_over() and board.turn == chess.WHITE:
     st.write("### Your Turn (White)")
@@ -194,6 +231,8 @@ if not board.is_game_over() and board.turn == chess.WHITE:
                     try:
                         board.push(parsed_move)
                         history.append(parsed_move.uci())
+                        render_board()  # Update board immediately
+                        render_pgn()    # Update PGN immediately
                         st.success(f"You played: {board.san(parsed_move)} ({parsed_move.uci()})")
                         st.rerun()
                     except Exception as e:
@@ -208,6 +247,8 @@ if not board.is_game_over() and board.turn == chess.WHITE:
                 random_move = random.choice(legal_moves)
                 board.push(random_move)
                 history.append(random_move.uci())
+                render_board()  # Update board immediately
+                render_pgn()    # Update PGN immediately
                 st.rerun()
 
 if not board.is_game_over() and board.turn == chess.BLACK:
@@ -224,6 +265,8 @@ if not board.is_game_over() and board.turn == chess.BLACK:
                 move_san = board.san(move)
                 board.push(move)
                 history.append(move.uci())
+                render_board()  # Update board immediately
+                render_pgn()    # Update PGN immediately
                 st.success(f"{ai_name} played: {move_san} ({move.uci()})")
                 st.rerun()
             else:
@@ -231,33 +274,6 @@ if not board.is_game_over() and board.turn == chess.BLACK:
         except Exception as e:
             st.error(f"Error during {ai_name} move: {e}")
 
-if history:
-    st.write("### Game History")
-    try:
-        game = chess.pgn.Game()
-        game.headers["Event"] = "Human vs o2"
-        game.headers["White"] = "Human"
-        game.headers["Black"] = "o2" if agent_loaded else "Random AI"
-        node = game
-        temp_board = chess.Board()
-        for uci in history:
-            move = chess.Move.from_uci(uci)
-            if move in temp_board.legal_moves:
-                node = node.add_main_variation(move)
-                temp_board.push(move)
-            else:
-                st.warning(f"Invalid move in history: {uci}")
-                break
-        st.code(str(game), language="pgn")
-    except Exception as e:
-        st.error(f"Error generating PGN: {e}")
-        move_pairs = []
-        for i in range(0, len(history), 2):
-            white_move = history[i]
-            black_move = history[i+1] if i+1 < len(history) else ""
-            move_pairs.append(f"{i//2 + 1}. {white_move} {black_move}")
-        st.code("\n".join(move_pairs))
-
 if board.is_game_over():
     st.write("### Game Over!")
     result = board.result()