Update app.py

app.py

@@ -17,8 +17,7 @@ import math
 import time
 from tqdm import tqdm
 from datetime import datetime
-
-
+import matplotlib.pyplot as plt
 
 
 SCREEN_HEIGHT = 600
@@ -533,7 +532,7 @@ class Game:
             pygame.display.update()
 
 
-    def play_auto(self):
+    def play_auto(self,episode_info):
         try:
             points_label = 0
             for episode in tqdm(range(1, NUM_EPISODES + 1), ascii=True, unit='episodes'):
@@ -542,7 +541,7 @@ class Game:
                 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}')
+                episode_info.text(f'Escenario: {episode}, Puntuación actual: {self.points}, Recompensa del episodio: {episode_reward}')
                 while self.run is True:
 
                     for event in pygame.event.get():
@@ -658,6 +657,15 @@ class Game:
                 print(f"Modelo duplicado guardado como: {final_model_filename}")
 
 
+def plot_rewards(ep_rewards):
+    plt.figure(figsize=(10, 6))
+    plt.plot(ep_rewards)
+    plt.title("Recompensas por Episodio")
+    plt.xlabel("Episodio")
+    plt.ylabel("Recompensa")
+    st.pyplot(plt)
+
+
 # Streamlit UI
 def streamlit_ui():
     st.title('Juego del Dinosaurio con IA')
@@ -688,6 +696,10 @@ def streamlit_ui():
         model_path = os.path.join(model_directory, selected_model_file)
         game = Game(EPSILON_INIT, load_model=True, model_path=model_path)
         game.play_auto(episode_info)
+    
+    # Llama a esta función después de que `play_auto` haya terminado
+    if len(game.ep_rewards) > 0:
+        plot_rewards(game.ep_rewards)
 
 # Ejecutar UI
 streamlit_ui()