import gymnasium as gym
import gym_donkeycar.envs
import gym_donkeycar.envs.donkey_env
import os

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

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

# Try gymnasium-style make
env = gym.make("donkey-generated-roads-v0")

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