from game.Engine import Engine
from game.interface.GUI import GUI
from agents import Agent

MAP_NUMBER = 1

if __name__ == '__main__':
    env = Engine(map_number=MAP_NUMBER)
    state = env.reset()

    agent1 = Agent(
        is_activated=False,
        model_structure=[512, 512, 256, 256],
        input_dim=state.shape,
        actions=env.actions,
    )

    agent2 = Agent(
        is_activated=True,
        model_structure=[256, 128],
        input_dim=state.shape,
        actions=env.actions,
    )

    env = Engine(map_number=MAP_NUMBER, agents=[agent1, agent2])
    gui = GUI(env)
    gui.reset()
    gui.render()

