Update app.py

app.py

@@ -19,6 +19,8 @@ from tqdm import tqdm
 from datetime import datetime
 
 
+
+
 SCREEN_HEIGHT = 600
 SCREEN_WIDTH = 1100
 
@@ -353,6 +355,7 @@ class Cloud:
 
 class Game:
     def __init__(self, epsilon, load_model=False, model_path=None):
+        os.environ["SDL_VIDEODRIVER"] = "dummy"  # Establece el driver de video de SDL a dummy
         pygame.init()
         self.SCREEN = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
 
@@ -653,10 +656,16 @@ class Game:
                 print(f"Modelo duplicado guardado como: {final_model_filename}")
 
 
-# Streamlit UI
+# Suponiendo que tienes un directorio con tus modelos
+model_directory = 'models/highscore/'
+model_files = os.listdir(model_directory)  # Obtiene una lista de archivos de modelo
+
+# Streamlit UI para seleccionar el modelo
 st.title('Juego del Dinosaurio con IA')
+selected_model_file = st.selectbox('Elige un modelo para cargar', model_files)
 
+# Botón para iniciar el juego con el modelo seleccionado
 if st.button('Iniciar Juego con IA'):
-    model_path = 'models/highscore/4245_BestScore_Final_2023-12-10_18-43-53_model.pth'  # Reemplaza con la ruta al modelo que quieras cargar
+    model_path = os.path.join(model_directory, selected_model_file)
     game = Game(EPSILON_INIT, load_model=True, model_path=model_path)
     game.play_auto()