Update app.py

app.py

@@ -678,6 +678,8 @@ def plot_rewards(ep_rewards):
 
 # Streamlit UI
 def streamlit_ui():
+    global EPSILON_INIT, EPSILON_DECAY, NUM_EPISODES  # Declare global variables at the start of the function
+    
     st.title('Juego del Dinosaurio con IA')
 
     # Barra lateral para configuraciones
@@ -702,23 +704,22 @@ def streamlit_ui():
     episode_info = st.empty()
 
     # Contenedores para métricas y logs
-    metrics_container = st.container()
-    logs_container = st.container()
+    metrics_container = st.expander("Graficas")
+    logs_container = st.expander("Logs")
     episode_info = st.empty()
 
     if st.button('Iniciar Juego con IA'):
-        # Aquí es donde actualizas las variables con los valores de los sliders
-        global EPSILON_INIT, EPSILON_DECAY, NUM_EPISODES
-        EPSILON_INIT = epsilon_init_slider
-        EPSILON_DECAY = epsilon_decay_slider
-        NUM_EPISODES = num_episodes_slider
+        # Actualizar las variables globales con los valores de los sliders
+        EPSILON_INIT = epsilon_init
+        EPSILON_DECAY = epsilon_decay
+        NUM_EPISODES = num_episodes
 
         model_path = os.path.join(model_directory, selected_model_file)
-        game = Game(epsilon_init_slider, load_model=True, model_path=model_path) # Usa el valor del slider aquí también
+        game = Game(epsilon_init, load_model=True, model_path=model_path) # Usa el valor del slider aquí también
         game.play_auto(episode_info, metrics_container, logs_container)
-        
+
     # Añadir la descripción debajo del botón
     st.markdown("Para poder correr el modelo usando Pygame y ver la GUI en tiempo real, descarga el juego y modelo en mi github: [chrome dino game](https://github.com/JuanPabloWakkaWatson/chrome-dino-automl23)")
 
 # Ejecutar UI
-streamlit_ui()
+streamlit_ui()