Update app.py

app.py

@@ -4,18 +4,28 @@ from stable_baselines3 import DQN
 import imageio
 import time
 import os
+import base64
 
 # TensorFlow figyelmeztetések elrejtése
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 
-st.set_page_config(page_title="Lunar Lander AI", page_icon="🚀")
+st.set_page_config(page_title="Lunar Lander AI", page_icon="🚀", layout="wide")
+
+# --- SEGÉDFÜGGVÉNY: HTML LETÖLTÉS (Hogy ne akadjon be a loop miatt) ---
+def get_binary_file_downloader_html(bin_file, file_label='File'):
+    with open(bin_file, 'rb') as f:
+        data = f.read()
+    bin_str = base64.b64encode(data).decode()
+    href = f'<a href="data:application/octet-stream;base64,{bin_str}" download="{os.path.basename(bin_file)}" style="text-decoration: none;">'
+    href += f'<button style="background-color: #4CAF50; border: none; color: white; padding: 15px 32px; text-align: center; text-decoration: none; display: inline-block; font-size: 16px; margin: 4px 2px; cursor: pointer; border-radius: 12px;">📥 {file_label} LETÖLTÉSE</button>'
+    href += '</a>'
+    return href
 
 # --- FŐ CÍM ---
 st.title("🌕 Lunar Lander AI")
 st.markdown("**Deep Q-Network (DQN)** algoritmus demonstrációja.")
 
-
-# --- 1. SZEKCIÓ: AI DEMO (Visszajátszás) ---
+# --- OLDALSÁV (MODOLL BETÖLTÉS) ---
 @st.cache_resource
 def load_model():
     if os.path.exists("lunar_lander_model.zip"):
@@ -23,42 +33,64 @@ def load_model():
     else:
         return None
 
-
 model = load_model()
 
-if model is None:
-    st.error("⚠️ Hiányzik a 'lunar_lander_model.zip' fájl!")
-else:
-    # Konténer a videónak
-    demo_container = st.container()
-
-    with demo_container:
-        st.subheader("🤖 Így csinálja a gép:")
-        start_button = st.button("AI Indítása / Végtelen ciklus ▶️")
+# --- FŐHASÁBOK ---
+col_game, col_download = st.columns([2, 1])
 
-        if start_button:
-            # Helyőrzők
-            status_text = st.empty()
-            video_placeholder = st.empty()
+# --- JOBB OLDAL: LETÖLTÉS (MINDIG LÁTHATÓ ÉS MŰKÖDIK) ---
+with col_download:
+    st.info("🎮 **Próbáld ki te is!**")
+    st.write("Az AI már megtanulta... de vajon neked sikerülne? Töltsd le a játékot a saját gépedre!")
+    
+    zip_filename = "lunar_lander_challenge.zip"
+    if os.path.exists(zip_filename):
+        # HTML Linket használunk a gomb helyett, így a loop nem blokkolja!
+        download_html = get_binary_file_downloader_html(zip_filename, "JÁTÉK CSOMAG")
+        st.markdown(download_html, unsafe_allow_html=True)
+        st.caption("Windows .zip | Tartalmazza: Python kód + Automata indító")
+    else:
+        st.warning("A letölthető fájl feltöltés alatt...")
 
+# --- BAL OLDAL: AI DEMO ---
+with col_game:
+    if model is None:
+        st.error("⚠️ Hiányzik a 'lunar_lander_model.zip'!")
+    else:
+        st.subheader("🤖 AI Visszajátszás")
+        
+        # Helyőrzők a UI elemeknek
+        start_button_placeholder = st.empty()
+        status_text = st.empty()
+        video_placeholder = st.empty()
+
+        # A gomb a placeholderben jön létre
+        start_clicked = start_button_placeholder.button("▶️ Szimuláció Indítása", type="primary")
+
+        if start_clicked:
+            # 1. ELTÜNTETJÜK A GOMBOT
+            start_button_placeholder.empty()
+            
+            # Inicializálás
             env = gym.make("LunarLander-v3", render_mode="rgb_array")
             episode_count = 0
             fps = 30
-
+            
+            # VÉGTELEN CIKLUS
             while True:
                 episode_count += 1
-
-                # Takarítás
-                video_placeholder.empty()
+                
+                # Takarítás az új kör előtt
+                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 = []
-
-                # Szimuláció futtatása
+                frames = [] 
+                
+                # Szimuláció ("láthatatlanul" és gyorsan)
                 while not done and not truncated:
                     action, _states = model.predict(obs, deterministic=True)
                     obs, reward, done, truncated, info = env.step(action)
@@ -69,14 +101,14 @@ else:
                 gif_path = "replay.gif"
                 imageio.mimsave(gif_path, frames, fps=fps)
                 video_duration = len(frames) / fps
-
+                
                 # Lejátszás
-                status_text.info(f"🎬 {episode_count}. kör visszajátszása...")
+                status_text.success(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 videó tart
+                
+                # Várakozás
                 time.sleep(video_duration)
-
+                
                 # Eredmény kiírása
                 if total_reward >= 200:
                     msg = f"✅ SIKER! Profi munka. Pontszám: {total_reward:.2f}"
@@ -85,41 +117,6 @@ else:
                     msg = f"⚠️ OK. Biztonságos. Pontszám: {total_reward:.2f}"
                 else:
                     msg = f"💥 ZUHANÁS! Pontszám: {total_reward:.2f}"
-
+                
                 status_text.markdown(f"### {msg}")
-                time.sleep(2)
-
-# --- 2. SZEKCIÓ: EMBERI KIHÍVÁS (LETÖLTÉS) ---
-st.markdown("---")  # Elválasztó vonal
-st.header("🎮 Te le tudod tenni?")
-
-col1, col2 = st.columns([2, 1])
-
-with col1:
-    st.write("""
-    Az AI már megtanulta... de vajon neked sikerülne? 
-    Töltsd le a játékot, és próbáld ki a saját gépeden az eredeti, nehéz fizikával!
-
-    **A csomag tartalma:**
-    * 🐍 `lunar_lander.py` (A játék kódja)
-    * ⚙️ `INDITAS.bat` (Automatikus telepítő és indító)
-    """)
-
-with col2:
-    # Letöltés gomb
-    zip_filename = "lunar_lander_challenge.zip"
-
-    if os.path.exists(zip_filename):
-        with open(zip_filename, "rb") as fp:
-            st.download_button(
-                label="📥 JÁTÉK LETÖLTÉSE\n(Windows .zip)",
-                data=fp,
-                file_name=zip_filename,
-                mime="application/zip",
-                help="Kattints a letöltéshez, csomagold ki, és indítsd el a .bat fájlt!"
-            )
-    else:
-        st.warning("A letölthető fájl (lunar_lander_challenge.zip) még nincs feltöltve.")
-
-st.markdown("---")
-st.caption("Készítette: Gemini & Faragó Csaba | Powered by Stable Baselines3 & Streamlit")
+                time.sleep(2)