Update app.py

app.py

@@ -541,6 +541,8 @@ class Game:
  step = 1
  current_state = self.get_state()
  self.run = True
+ # Al final de cada episodio, actualiza la interfaz de Streamlit
+ st_placeholder.text(f'Escenario: {episode}, Resultado: {episode_reward}')
  while self.run is True:
 
  for event in pygame.event.get():
@@ -678,16 +680,14 @@ def streamlit_ui():
  score = st.empty() # Usar .empty() para actualizar más tarde
  with highscore_col:
  high_score = st.empty() # Usar .empty() para actualizar más tarde
-
- # Gráfico de recompensas
- data = [] # Inicializa data como lista vacía
- reward_chart = st.line_chart(data)
+ # Placeholder para mostrar el número de escenario y el resultado
+ episode_info = st.empty()
 
  # Botón para iniciar el juego
  if st.button('Iniciar Juego con IA'):
  model_path = os.path.join(model_directory, selected_model_file)
  game = Game(EPSILON_INIT, load_model=True, model_path=model_path)
- game.play_auto() # Asegúrate de que esta función actualice 'score', 'high_score' y 'data'
+ game.play_auto(episode_info)
 
 # Ejecutar UI
 streamlit_ui()