import gym
import gym_donkeycar.envs
import gym_donkeycar.envs.donkey_env  # Force registration
import os

host = os.environ.get('DONKEY_SIM_HOST', '127.0.0.1')
port = int(os.environ.get('DONKEY_SIM_PORT', '9091'))

print(f"Attempting to connect to donkey sim at {host}:{port}")

env = gym.make("donkey-generated-roads-v0")

observation = env.reset()
for t in range(10):
    action = env.action_space.sample()
    observation, reward, done, info = env.step(action)
    print(f"Step {t}: reward {reward}, done {done}, info {info}")
    if done:
        observation = env.reset()
env.close()