# Supported DonkeyCar tracks detected in code:
available_tracks = [
    "generated_road",
    "warehouse",
    "sparkfun_avc",
    "generated_track",
    "mountain_track",
    "roboracingleague_1",
    "waveshare",
    "mini_monaco",
    "warren",
    "thunderhill",
    "circuit_launch",
]
from gym_donkeycar.envs.donkey_env import DonkeyEnv
import sys

print("Available tracks:")
for i, t in enumerate(available_tracks):
    print(f"[{i}] {t}")

choice = input("Enter the number of the track to use: ")
try:
    idx = int(choice.strip())
    track = available_tracks[idx]
    print(f"Loading track: {track}")
except Exception as e:
    print(f"Invalid selection ({e}), using default track: {available_tracks[0]}")
    track = available_tracks[0]

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