def METHOD_NAME():
    game_state, set_game_state = reactpy.hooks.use_state(GameState.init)
    if game_state == GameState.play:
        return GameLoop(grid_size=6, block_scale=50, set_game_state=set_game_state)
    start_button = reactpy.html.button(
        {"on_click": lambda event: set_game_state(GameState.play)}, "Start"
    )
    if game_state == GameState.won:
        menu = reactpy.html.div(reactpy.html.h3("You won!"), start_button)
    elif game_state == GameState.lost:
        menu = reactpy.html.div(reactpy.html.h3("You lost"), start_button)
    else:
        menu = reactpy.html.div(reactpy.html.h3("Click to play"), start_button)
    menu_style = reactpy.html.style(
        """
        .snake-game-menu h3 {
            margin-top: 0px !important;
        }
        """
    )
    return reactpy.html.div({"class_name": "snake-game-menu"}, menu_style, menu)