Update app.py

app.py

@@ -35,7 +35,7 @@ class TradingEnv(gym.Env):
     
     def _get_observation(self):
         window_data = self.data[self.current_step-self.window_size:self.current_step]
-        obs = window_data[['Close', 'EMA']].values
+        obs = window_data[['close', 'EMA']].values
         obs = (obs - obs.min()) / (obs.max() - obs.min())
         return obs
     
@@ -48,9 +48,9 @@ class TradingEnv(gym.Env):
             done = True
         else:
             if action == 1:  
-                reward = self.data['Close'].iloc[self.current_step] - self.data['Close'].iloc[self.current_step - 1]
+                reward = self.data['close'].iloc[self.current_step] - self.data['close'].iloc[self.current_step - 1]
             elif action == 2: 
-                reward = self.data['Close'].iloc[self.current_step - 1] - self.data['Close'].iloc[self.current_step]
+                reward = self.data['close'].iloc[self.current_step - 1] - self.data['close'].iloc[self.current_step]
 
         return self._get_observation(), reward, done, {}