uci color differently when it's own turn

[sugizo]

code

import chess
import chess.pgn
import time
from IPython.display import display, HTML, clear_output

def who(player):
    return "White" if player == chess.WHITE else "Black"

def display_board(board, use_svg):
    if use_svg:
        return board._repr_svg_()
    else:
        return "<pre>" + str(board) + "</pre>"

def play_pgn(file_pgn, pause = 9.9, visual = "svg"):
    """
    visual: "simple" | "svg" | None
    """
    use_svg = (visual == "svg")
    pgn = open(file_pgn)
    read_game = chess.pgn.read_game(pgn)
    read_game_mainline_moves = read_game.mainline_moves()
    board = read_game.board()
    game = chess.pgn.Game()

    try:
        for uci in read_game_mainline_moves:
            board.push(uci)
            board_stop = display_board(board, use_svg)
            len_board_move_stack = len(board.move_stack)

            if len_board_move_stack == 1:
                node = game.add_variation(uci)
            else:
                node = node.add_variation(uci)
            
            mainline_moves = game.mainline_moves()

            html = """
            <table>
            <tr>
            <th rowspan = "7" style = "width:45%%;text-align: center">%s</th>
            <th style = "text-align: left">Mainline Moves</th>
            <td style = "text-align: left">%s</td>
            </tr>
            </table>
            """ % (board_stop, mainline_moves)

            if visual is not None:
                if visual == "svg":
                    clear_output(wait = True)

                display(HTML(html) )

                if visual == "svg":
                    time.sleep(pause)

    except KeyboardInterrupt:
        msg = "Game Interrupted!"
        return (msg, board)

play_pgn("pgn/chess-scholar_mate_1.pgn", pause = 9.9)

pgn content
1. e4 e5 2. Qh5 Nc6 3. Bc4 Nf6 4. Qxf7#

tested environment
jupyter notebook or google colab

objective
current code situation when turn 2 (len(board.move_stack) == 2) show
1. e4 e5
objective
1. e4 e5 2. Qh5 Nc6 3. Bc4 Nf6 4. Qxf7#
e5 is with colored (red or whatever)

question
how to achieve it (uci color differently when it's own turn) in python-chess ?

best regards