Update app.py

app.py

@@ -13,13 +13,13 @@ MODEL_PATH = "checkers_master_final.pth"
 
 st.set_page_config(page_title="AlphaCheckerZero", page_icon="♟️", layout="wide")
 
-# --- ESTILOS CSS PERSONALIZADOS (A MÁGICA VISUAL) ---
+# --- ESTILOS CSS PERSONALIZADOS ---
 st.markdown("""
 <style>
     .board-container {
         display: grid;
-        grid-template-columns: 30px repeat(8, 60px); /* Coluna guia + 8 colunas do jogo */
-        grid-template-rows: 30px repeat(8, 60px);    /* Linha guia + 8 linhas do jogo */
+        grid-template-columns: 30px repeat(8, 60px);
+        grid-template-rows: 30px repeat(8, 60px);
         gap: 2px;
         background-color: #444;
         padding: 10px;
@@ -44,8 +44,8 @@ st.markdown("""
         font-size: 40px;
         cursor: default;
     }
-    .white-square { background-color: #f0d9b5; } /* Cor clara (madeira) */
-    .black-square { background-color: #b58863; } /* Cor escura (madeira) */
+    .white-square { background-color: #f0d9b5; }
+    .black-square { background-color: #b58863; }
     
     .piece-white { 
         color: #fff; 
@@ -59,12 +59,11 @@ st.markdown("""
     }
     .king { border: 2px solid gold; border-radius: 50%; padding: 2px; box-shadow: 0 0 10px gold;}
     
-    /* Ajuste para deixar o selectbox mais bonito */
     .stSelectbox label { font-size: 1.2rem; font-weight: bold; }
 </style>
 """, unsafe_allow_html=True)
 
-# --- LÓGICA DO JOGO (Inalterada) ---
+# --- LÓGICA DO JOGO ---
 class Checkers:
     def get_initial_board(self):
         board = np.zeros((BOARD_SIZE, BOARD_SIZE), dtype=np.int8)
@@ -124,7 +123,13 @@ class Checkers:
 
     def apply_move(self, board, move):
         b_ = np.copy(board)
-        is_jump_chain = isinstance(move, list) or (isinstance(move, tuple) and isinstance(move[0], tuple) and isinstance(move[0][0], tuple))
+        # AQUI TAMBÉM PODERIA DAR ERRO, ENTÃO GARANTIMOS O FORMATO
+        is_jump_chain = False
+        if isinstance(move, list):
+            is_jump_chain = True
+        elif isinstance(move, tuple) and len(move) > 0 and isinstance(move[0], tuple) and len(move[0]) > 0 and isinstance(move[0][0], tuple):
+            is_jump_chain = True
+
         sub_moves = move if is_jump_chain else [move]
         for (r1, c1), (r2, c2) in sub_moves:
             piece = b_[r1, c1]
@@ -206,7 +211,10 @@ class MCTS:
             else: start_pos_tuple = move[0]
             start_pos_idx = start_pos_tuple[0] * BOARD_SIZE + start_pos_tuple[1]
             prior = policy_probs[start_pos_idx]
+            
+            # IMPORTANTE: MCTS converte lista para tupla aqui para usar como chave de dicionário
             key = tuple(move) if isinstance(move, list) else move
+            
             move_priors[key] = prior; total_prior += prior
         if total_prior > 0:
             for move_key, prior in move_priors.items(): node.children[move_key] = MCTSNode(parent=node, prior=prior / total_prior)
@@ -216,7 +224,7 @@ class MCTS:
                 node.children[key] = MCTSNode(parent=node, prior=1.0 / len(valid_moves))
         return value
 
-# --- INTERFACE GRÁFICA MELHORADA ---
+# --- INTERFACE GRÁFICA ---
 
 @st.cache_resource
 def load_model():
@@ -244,38 +252,48 @@ game = Checkers()
 mcts = MCTS(game, model, sims=150)
 
 def format_move_for_human(move):
-    """Transforma ((2,1), (3,2)) em algo mais legível"""
-    if isinstance(move, list): # Pulo múltiplo
+    """
+    Formata o movimento para texto legível.
+    Lida tanto com listas (pulos originais) quanto tuplas aninhadas (pulos convertidos pelo MCTS).
+    """
+    is_jump = False
+    
+    # 1. Se for lista, é um pulo
+    if isinstance(move, list):
+        is_jump = True
+        
+    # 2. Se for tupla, precisamos ver o conteúdo para saber se é pulo ou movimento simples
+    elif isinstance(move, tuple):
+        # Se o primeiro item da tupla for OUTRA tupla (ex: ((r,c), (r,c))), então é um pulo que foi convertido
+        # Um movimento simples teria um int como primeiro sub-item: ((2,3), (3,4)) -> move[0] é (2,3), move[0][0] é 2 (int).
+        if len(move) > 0 and isinstance(move[0], tuple) and len(move[0]) > 0 and isinstance(move[0][0], tuple):
+            is_jump = True
+
+    if is_jump:
+        # Pulo múltiplo ou captura simples
         path = " -> ".join([f"({r},{c})" for (r,c), _ in move] + [str(move[-1][1])])
-        return f"Salto Múltiplo: {path}"
+        return f"Salto/Captura: {path}"
     else:
+        # Movimento simples
         (r1, c1), (r2, c2) = move
         return f"Mover de ({r1}, {c1}) para ({r2}, {c2})"
 
 def render_board_html(board):
     html = '<div class="board-container">'
-    
-    # Cabeçalho das colunas (0-7)
-    html += '<div class="header-cell"></div>' # Canto vazio
-    for c in range(8):
-        html += f'<div class="header-cell">{c}</div>'
+    html += '<div class="header-cell"></div>'
+    for c in range(8): html += f'<div class="header-cell">{c}</div>'
         
     for r in range(8):
-        # Cabeçalho da linha (0-7)
         html += f'<div class="header-cell">{r}</div>'
-        
         for c in range(8):
             color_class = "black-square" if (r + c) % 2 == 1 else "white-square"
             piece = board[r, c]
-            
             content = ""
             if piece == 1: content = '<span class="piece-white">⚪</span>'
-            elif piece == 2: content = '<span class="piece-white king">👑</span>' # Dama branca
+            elif piece == 2: content = '<span class="piece-white king">👑</span>'
             elif piece == -1: content = '<span class="piece-black">⚫</span>'
-            elif piece == -2: content = '<span class="piece-black king">👑</span>' # Dama preta (coroa dourada no CSS)
-            
+            elif piece == -2: content = '<span class="piece-black king">👑</span>'
             html += f'<div class="square {color_class}">{content}</div>'
-            
     html += '</div>'
     return html
 
@@ -300,16 +318,13 @@ with col2:
     else:
         st.info(st.session_state.message)
         
-        # Lógica de Turno
         if st.session_state.player == 1:
             valid_moves = game.get_valid_moves(st.session_state.board, 1)
-            
             if not valid_moves:
                  st.session_state.game_over = True
                  st.session_state.message = "Sem movimentos válidos. Você perdeu. 😔"
                  st.rerun()
 
-            # Dicionário para mapear a descrição bonita para o objeto de movimento real
             move_map = {format_move_for_human(m): m for m in valid_moves}
             selected_desc = st.selectbox("Sua vez! Escolha o movimento:", list(move_map.keys()))
             
@@ -321,24 +336,26 @@ with col2:
                 st.rerun()
 
         else:
-            # Vez da IA
             with st.spinner("A AlphaCheckerZero está pensando..."):
-                time.sleep(0.2) # UX Feel
-                
+                time.sleep(0.2)
                 valid_moves, policy = mcts.run(np.copy(st.session_state.board), -1)
-                
                 if not valid_moves:
                      st.session_state.game_over = True
                      st.session_state.message = "A IA travou! VOCÊ VENCEU! 🎉"
                      st.rerun()
 
                 move = valid_moves[np.argmax(policy)]
+                
+                # AQUI É ONDE OCORRIA O ERRO ANTES:
+                # A IA retorna o movimento (que pode ser uma tupla de pulo)
+                # E agora a função 'format_move_for_human' sabe lidar com isso.
+                move_text = format_move_for_human(move)
+                
                 st.session_state.board = game.apply_move(st.session_state.board, move)
                 st.session_state.player = 1
-                st.session_state.message = f"IA moveu: {format_move_for_human(move)}. Sua vez!"
+                st.session_state.message = f"IA moveu: {move_text}. Sua vez!"
                 st.rerun()
 
-# Rodapé explicativo
 st.markdown("---")
 st.caption("**Legenda:** ⚪ Suas Peças | ⚫ Peças da IA | 👑 Dama")
 st.caption("Desenvolvido por Gabriel Yogi com auxílio de Berta.")