Drawing robots on a grid? That sounds close to a simple game. So Pygame may be the right graphical API for you.