poleCart_manual.py

#import gym

#env = gym.make('MountainCar-v0')
#env = gym.make('CartPole-v0')

#env = gym.make('MsPacman-v0')

import gym # get the environment from openAI
import curses # for keypress

env = gym.make('CartPole-v0') #init environment

observation = env.reset()

screen = curses.initscr()
curses.noecho()
curses.curs_set(0)
screen.keypad(1)
screen.addstr("Play the game")


while True:
	env.render() # make visible
	event = screen.getch()
	if event == curses.KEY_LEFT:
		print("Left Arrow Key pressed")
		action = 0
	elif event == curses.KEY_RIGHT:
		print("Right Arrow Key pressed")
		action = 1
	else:
		break
	env.step(action)
	print action

curses.endwin()