Update app.py

app.py

@@ -73,15 +73,20 @@ class ArkanoidEnv(gym.Env):
         super(ArkanoidEnv, self).__init__()
         self.action_space = gym.spaces.Discrete(3)  # 0: stay, 1: move left, 2: move right
         self.observation_space = gym.spaces.Box(low=0, high=SCREEN_WIDTH, shape=(5 + BRICK_ROWS * BRICK_COLS * 2,), dtype=np.float32)
+        self.seed_value = None
         self.reset()
 
-    def reset(self):
+    def reset(self, seed=None, options=None):
+        if seed is not None:
+            random.seed(seed)
+            np.random.seed(seed)
+            self.seed_value = seed
         self.paddle = Paddle()
         self.ball = Ball()
         self.bricks = [Brick(x, y) for y in range(BRICK_HEIGHT, BRICK_HEIGHT * (BRICK_ROWS + 1), BRICK_HEIGHT) for x in range(BRICK_WIDTH, SCREEN_WIDTH - BRICK_WIDTH, BRICK_WIDTH)]
         self.done = False
         self.score = 0
-        return self._get_state()
+        return self._get_state(), {}
 
     def step(self, action):
         if action == 0:
@@ -154,7 +159,7 @@ def evaluate_model(model):
 def play_game():
     env = ArkanoidEnv()
     model = DQN.load("arkanoid_model")
-    obs = env.reset()
+    obs = env.reset()[0]
     done = False
     frames = []
     while not done: