Update app.py

app.py

@@ -234,12 +234,16 @@ def train_ppo(episodes=300, steps_per_update=256, lr=3e-4):
 # ==========================================
 # 4. HF SPACE DEMO
 # ==========================================
+# ==========================================
+# 4. HF SPACE DEMO (ИСПРАВЛЕНО ДЛЯ HF SPACES)
+# ==========================================
 def create_demo():
     print("🏗️ Обучение PPO-агента в бесконечном мире...")
     agent, history = train_ppo(episodes=100)
     agent.eval()
     
-    def run_exploration(n_steps=200):
+    def run_exploration(n_steps: int = 200):
+        """Функция с явной типизацией для избежания бага schema в Gradio"""
         env = InfiniteWorld(seed=random.randint(0, 99999))
         obs = env.reset()
         hidden = None
@@ -247,7 +251,7 @@ def create_demo():
         positions = []
         
         with torch.no_grad():
-            for _ in range(n_steps):
+            for _ in range(min(n_steps, 500)):  # Ограничиваем макс шаги
                 fig, ax = plt.subplots(figsize=(5, 5))
                 ax.imshow(obs)
                 ax.set_title(f"Pos: {env.agent_pos} | Chunks: {len(env.chunks)}")
@@ -262,26 +266,54 @@ def create_demo():
                 if done:
                     break
         
-        # Карта посещённых позиций
-        fig2, ax2 = plt.subplots(figsize=(6, 6))
-        xs, ys = zip(*positions)
-        ax2.scatter(xs, ys, c=range(len(positions)), cmap='viridis', s=1)
-        ax2.set_title("Траектория исследования")
-        ax2.set_aspect('equal')
-        frames.append(fig2)
-        plt.close(fig2)
+        # Карта траектории
+        if positions:
+            fig2, ax2 = plt.subplots(figsize=(6, 6))
+            xs, ys = zip(*positions)
+            ax2.scatter(xs, ys, c=range(len(positions)), cmap='viridis', s=1)
+            ax2.set_title("Траектория исследования")
+            ax2.set_aspect('equal')
+            frames.append(fig2)
+            plt.close(fig2)
         
         return frames
     
+    # Явное указание типов компонентов устраняет баг schema
     with gr.Blocks(title="Infinite Builder Agent") as demo:
         gr.Markdown("# 🌍 Бесконечный мир: PPO-агент с нуля")
         gr.Markdown("Агент видит только локальный патч 16×16, имеет GRU-память и исследует процедурный мир.")
-        btn = gr.Button("▶️ Запустить исследование (200 шагов)")
-        gallery = gr.Gallery(label="Процесс", columns=5, height="auto")
-        btn.click(fn=run_exploration, outputs=gallery)
+        
+        steps_input = gr.Slider(
+            minimum=50, 
+            maximum=500, 
+            step=50, 
+            value=200, 
+            label="Количество шагов исследования"
+        )
+        btn = gr.Button("▶️ Запустить исследование", variant="primary")
+        gallery = gr.Gallery(
+            label="Процесс строительства и исследования", 
+            columns=4, 
+            height="auto",
+            object_fit="contain"
+        )
+        
+        btn.click(
+            fn=run_exploration, 
+            inputs=[steps_input],
+            outputs=[gallery]
+        )
     
     return demo
 
+
 if __name__ == "__main__":
     demo = create_demo()
-    demo.launch()
+    # КРИТИЧНО: Для HF Spaces обязательно server_name="0.0.0.0"
+    # share=False на Spaces, т.к. прокси сам маршрутизирует трафик
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False,
+        show_api=False
+    )