# -*- coding: utf-8 -*-
"""
FrozenLake play using simple input.
"""
# !pip install gymnasium
# !pip install pygame
import numpy as np
import matplotlib.pyplot as plt
import gymnasium as gym

env = gym.make('FrozenLake-v1',is_slippery=True,
               render_mode='rgb_array')

state=env.reset()
screen=env.render()
plt.imshow(screen)
plt.show()
#env.reward_range
#env.metadata
#env.spec
#env.action_space
commandes={"G":0,"H":3,"B":1,"D":2}
for _ in range(100):
    prompt=input('ou doit aller le bonhomme (G/D/H/B): ')
    prompt=commandes[prompt]
    print('commande=',prompt)
    env.step(prompt)
    screen=env.render()
#    if len(screen>0):
    plt.imshow(screen)
    plt.show()