Update app.py

app.py

@@ -667,23 +667,27 @@ def streamlit_ui():
         epsilon_decay = st.slider("Epsilon Decay", 0.0, 1.0, EPSILON_DECAY)
         num_episodes = st.slider("Número de Episodios", 1, 500, NUM_EPISODES)
 
+    # Seleccionar modelo
+    model_directory = 'models/highscore/'
+    model_files = os.listdir(model_directory)
+    selected_model_file = st.selectbox('Elige un modelo para cargar', model_files)
+
     # Mostrar métricas
     score_col, highscore_col = st.columns(2)
     with score_col:
-        st.metric(label="Puntaje Actual", value=0)  # Actualiza este valor en tiempo real
+        score = st.empty()  # Usar .empty() para actualizar más tarde
     with highscore_col:
-        st.metric(label="Puntaje Más Alto", value=0)  # Actualiza este valor en tiempo real
-
-    # Botón para iniciar el juego
-    if st.button('Iniciar Juego con IA'):
-        # Iniciar juego aquí
-        pass
+        high_score = st.empty()  # Usar .empty() para actualizar más tarde
 
     # Gráfico de recompensas
-    st.line_chart(data)  # Actualiza 'data' con las métricas en tiempo real
+    data = []  # Inicializa data como lista vacía
+    reward_chart = st.line_chart(data)
 
-    # Área para logs y salidas
-    st.text_area("Logs del Juego", "Aquí se mostrarán los logs...")
+    # 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'
 
 # Ejecutar UI
 streamlit_ui()