Update src/streamlit_app.py

src/streamlit_app.py

@@ -252,12 +252,15 @@ if not board.is_game_over() and board.turn == chess.WHITE:
                 st.rerun()
 
 if not board.is_game_over() and board.turn == chess.BLACK:
-    ai_name = "o2" if agent_loaded else "Random AI"
-    st.write(f"### {ai_name}'s Turn (Black)")
-    with st.spinner(f"{ai_name} is thinking..."):
+    st.write("### o2's Turn (Black)")
+    with st.spinner("o2 is thinking..."):
         try:
             if agent_loaded and agent:
-                move = agent.select_move(board, use_mcts=True, simulations=30)
+                # Use temperature sampling for first 10 moves, then greedy
+                if len(history) < 20:  # 10 moves per side
+                    move = agent.select_move(board, use_mcts=True, simulations=30, temperature=1.2)
+                else:
+                    move = agent.select_move(board, use_mcts=True, simulations=30, temperature=0.0)
             else:
                 legal_moves = list(board.legal_moves)
                 move = random.choice(legal_moves) if legal_moves else None
@@ -265,14 +268,12 @@ if not board.is_game_over() and board.turn == chess.BLACK:
                 move_san = board.san(move)
                 board.push(move)
                 history.append(move.uci())
-                render_board()  # Update board immediately
-                render_pgn()    # Update PGN immediately
-                st.success(f"{ai_name} played: {move_san} ({move.uci()})")
+                st.success(f"o2 played: {move_san} ({move.uci()})")
                 st.rerun()
             else:
-                st.error(f"{ai_name} couldn't find a valid move")
+                st.error("o2 couldn't find a valid move")
         except Exception as e:
-            st.error(f"Error during {ai_name} move: {e}")
+            st.error(f"Error during o2 move: {e}")
 
 if board.is_game_over():
     st.write("### Game Over!")