Update app.py

app.py

@@ -27,49 +27,62 @@ if model is None:
 else:
     if st.button("Start / Végtelen ciklus indítása ▶️"):
         
+        # Helyőrzők létrehozása
         status_text = st.empty()
         video_placeholder = st.empty()
         
         env = gym.make("LunarLander-v3", render_mode="rgb_array")
         episode_count = 0
-        fps = 30  # Rögzített FPS érték
+        fps = 30
         
         while True:
             episode_count += 1
+            
+            # 1. Előkészítés
+            # Töröljük az előző videót, hogy ne látszódjon, amíg az újat számoljuk
+            video_placeholder.empty() 
+            status_text.info(f"🔄 {episode_count}. szimuláció számolása a háttérben...")
+            
             obs, info = env.reset()
             done = False
             truncated = False
             total_reward = 0
             frames = [] 
             
-            status_text.info(f"Szimuláció folyamatban ({episode_count}. kör)... Kérlek várj!")
-            
-            # --- SZIMULÁCIÓ ---
+            # 2. Szimuláció futtatása (láthatatlanul)
             while not done and not truncated:
                 action, _states = model.predict(obs, deterministic=True)
                 obs, reward, done, truncated, info = env.step(action)
                 total_reward += reward
                 frames.append(env.render())
 
-            # --- GIF MEGJELENÍTÉS IDŐZÍTÉSSEL ---
-            status_text.info("Visszajátszás generálása...")
-            
+            # 3. GIF generálása
             gif_path = "replay.gif"
             imageio.mimsave(gif_path, frames, fps=fps)
             
+            # Videó hossza másodpercben
+            video_duration = len(frames) / fps
+            
+            # 4. Videó indítása (Eredmény MÉG NINCS kiírva)
+            status_text.info(f"🎬 {episode_count}. kör visszajátszása...")
+            video_placeholder.image(gif_path, caption=f"{episode_count}. Leszállás")
+            
+            # Várakozás, amíg a felhasználó végignézi a videót
+            time.sleep(video_duration)
+            
+            # 5. Eredmény felfedése (Csak most!)
             if total_reward >= 200:
-                msg = f"✅ SIKER! Pontszám: {total_reward:.2f}"
+                msg = f"✅ SIKER! Profi munka. Pontszám: {total_reward:.2f}"
+                st.balloons() # Csak siker esetén jön lufi
             elif total_reward >= 0:
-                msg = f"⚠️ OK. Pontszám: {total_reward:.2f}"
+                msg = f"⚠️ OK. Biztonságos. Pontszám: {total_reward:.2f}"
             else:
-                msg = f"💥 ZUHANÁS. Pontszám: {total_reward:.2f}"
+                msg = f"💥 ZUHANÁS! Pontszám: {total_reward:.2f}"
             
+            # Kiírjuk az eredményt
             status_text.markdown(f"### {msg}")
-            video_placeholder.image(gif_path, caption=f"{episode_count}. kör visszajátszása")
             
-            # --- A JAVÍTÁS: ---
-            # Kiszámoljuk, hány másodpercig tart a videó
-            video_duration = len(frames) / fps
+            # 6. Szünet az olvasáshoz
+            time.sleep(2)
             
-            # Várunk a videó hossza + 2 másodpercet, mielőtt újat kezdünk
-            time.sleep(video_duration + 2)
+            # Ezután ugrik vissza a ciklus elejére