Update app.py

app.py

@@ -5,7 +5,6 @@ import numpy as np
 import os
 import io
 import time
-import uuid
 from collections import OrderedDict
 from PIL import Image, ImageDraw, ImageFont
 from RL_Chess_Alpha import AlphaChessNet, AlphaMCTS
@@ -46,7 +45,7 @@ def load_piece_images():
 piece_images = load_piece_images()
 
 # ==========================================
-# 2. SESSION MANAGEMENT (One per user)
+# 2. SESSION MANAGEMENT (One per browser tab)
 # ==========================================
 class GameSession:
     def __init__(self):
@@ -96,7 +95,9 @@ def render_board(session, status=""):
                 
     return img, status
 
-def handle_click(evt: gr.SelectData, sid, wm, bm):
+def handle_click(evt: gr.SelectData, wm, bm, request: gr.Request):
+    # Grabs the unique ID for this browser tab
+    sid = request.session_hash 
     session = get_session(sid)
     x, y = evt.index
     file, rank = x // 80, 7 - (y // 80)
@@ -149,22 +150,20 @@ def ai_turn(sid, sims):
     
     return render_board(session, f"AI played {san}")
 
-def reset_all(sid):
+def reset_all(request: gr.Request):
+    sid = request.session_hash
     session = get_session(sid)
     session.board.reset()
     session.selected_sq = None
-    # Reset MCTS to flush the search tree cache for the new game
     session.mcts = AlphaMCTS(model, DEVICE)
     return render_board(session, "Game Reset")
 
 # ==========================================
 # 4. GRADIO UI
 # ==========================================
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+# Note: Theme is now passed in launch() in Gradio 6.0
+with gr.Blocks() as demo:
     gr.Markdown("# ♟️ AlphaZero Tactical Engine - Skill level 3")
-    
-    # Generate a unique Session ID for the specific browser tab
-    session_id = gr.State(lambda: str(uuid.uuid4()))
 
     with gr.Row():
         with gr.Column(scale=2):
@@ -181,7 +180,8 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             auto_go = gr.Checkbox(label="Enable Autoplay", value=True)
             reset_btn = gr.Button("Reset Game")
 
-    def main_loop(is_auto, wm, bm, s, sid):
+    def main_loop(is_auto, wm, bm, s, request: gr.Request):
+        sid = request.session_hash
         session = get_session(sid)
         if not is_auto or session.board.is_game_over(): 
             return gr.update(), gr.update()
@@ -192,25 +192,25 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             
         return gr.update(), gr.update()
 
+    # Pass only the UI components to inputs, request object is injected automatically
     board_img.select(
         handle_click, 
-        inputs=[session_id, w_mode, b_mode], 
+        inputs=[w_mode, b_mode], 
         outputs=[board_img, status_txt]
     )
     
     reset_btn.click(
         reset_all, 
-        inputs=[session_id], 
         outputs=[board_img, status_txt]
     )
     
-    # 1-second timer to drive the AI turn
     timer = gr.Timer(1.0)
     timer.tick(
         main_loop, 
-        inputs=[auto_go, w_mode, b_mode, sims_val, session_id], 
+        inputs=[auto_go, w_mode, b_mode, sims_val], 
         outputs=[board_img, status_txt]
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    # Updated launch args for Gradio 6.0 and HF Spaces
+    demo.launch(theme=gr.themes.Soft())