Update app.py

app.py

@@ -54,7 +54,7 @@ class TradingEnv(gym.Env):
             elif action == 2: 
                 reward = self.data['close'].iloc[self.current_step - 1] - self.data['close'].iloc[self.current_step]
 
-        return self._get_observation(), reward, done, {}
+        return self._get_observation(), reward, done, {}, {}
 
 def fetch_data(symbol='ETH', tsym='USD', start_date='2021-01-01', api_key='YOUR_API_KEY'):
     start_date = datetime.strptime(start_date, '%Y-%m-%d')
@@ -92,7 +92,7 @@ def load_or_train_model(env):
     return model
 
 def run_model(model, env, data):
-    obs = env.reset()
+    obs, _ = env.reset()
     dates = data.index[50:]
     prices = data['close'][50:]
     emas = data['EMA'][50:]
@@ -101,7 +101,7 @@ def run_model(model, env, data):
     for date, price, ema in zip(dates, prices, emas):
         action, _ = model.predict(obs)
         actions.append(action[0])
-        obs, _, done, _ = env.step(action)
+        obs, _, done, _, _ = env.step(action)
         if done:
             break