import os
import gymnasium as gym
from gym_donkeycar.envs.donkey_env import DonkeyEnv
from autoresearch.envs.gymnasium import GymnasiumEnv

# Choose your desired track:
TRACK = os.environ.get("DONKEYCAR_TRACK", "generated_road")  # You can override with env var

print(f"[AutoResearch] Using DonkeyCar track: {TRACK}")
donkey_env = DonkeyEnv(level=TRACK)

# Wrap it as an autoresearch env
aresearch_env = GymnasiumEnv(donkey_env)

# Import and run an agent from autoresearch
from autoresearch.agents.dqn import main as dqn_main
# Run for a few episodes with default settings
if __name__ == '__main__':
    dqn_main(env=aresearch_env)