Update app.py

app.py

@@ -9,11 +9,62 @@ import time
 # --- CONFIGURAÇÕES ---
 BOARD_SIZE = 8
 DEVICE = torch.device("cpu")
-MODEL_PATH = "checkers_master_final.pth" # Certifique-se de que este arquivo está no Space!
-
-# --- DEFINIÇÃO DAS CLASSES (Rede Neural e Jogo) ---
-# A Berta copiou a lógica exata do seu script para garantir que funcione igual.
-
+MODEL_PATH = "checkers_master_final.pth" 
+
+st.set_page_config(page_title="AlphaCheckerZero", page_icon="♟️", layout="wide")
+
+# --- ESTILOS CSS PERSONALIZADOS (A MÁGICA VISUAL) ---
+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 */
+        gap: 2px;
+        background-color: #444;
+        padding: 10px;
+        border-radius: 10px;
+        width: fit-content;
+        margin: 0 auto;
+    }
+    .header-cell {
+        display: flex;
+        align-items: center;
+        justify-content: center;
+        color: white;
+        font-weight: bold;
+        font-family: monospace;
+    }
+    .square {
+        width: 60px;
+        height: 60px;
+        display: flex;
+        align-items: center;
+        justify-content: center;
+        font-size: 40px;
+        cursor: default;
+    }
+    .white-square { background-color: #f0d9b5; } /* Cor clara (madeira) */
+    .black-square { background-color: #b58863; } /* Cor escura (madeira) */
+    
+    .piece-white { 
+        color: #fff; 
+        text-shadow: 0 0 5px rgba(0,0,0,0.5);
+        transform: scale(1.2);
+    }
+    .piece-black { 
+        color: #333; 
+        text-shadow: 0 0 2px rgba(255,255,255,0.3);
+        transform: scale(1.2);
+    }
+    .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) ---
 class Checkers:
     def get_initial_board(self):
         board = np.zeros((BOARD_SIZE, BOARD_SIZE), dtype=np.int8)
@@ -165,14 +216,8 @@ class MCTS:
                 node.children[key] = MCTSNode(parent=node, prior=1.0 / len(valid_moves))
         return value
 
-# --- INTERFACE DO STREAMLIT ---
+# --- INTERFACE GRÁFICA MELHORADA ---
 
-st.set_page_config(page_title="AlphaCheckerZero", page_icon="♟️")
-
-st.title("♟️ AlphaCheckerZero Arena")
-st.write("Gabriel Yogi's Neural Network AI")
-
-# 1. Carregar o Modelo (com Cache para ser rápido)
 @st.cache_resource
 def load_model():
     if not os.path.exists(MODEL_PATH):
@@ -185,102 +230,115 @@ def load_model():
 model = load_model()
 
 if model is None:
-    st.error(f"Arquivo '{MODEL_PATH}' não encontrado. Por favor, faça upload do arquivo .pth para o Space.")
+    st.error(f"Arquivo '{MODEL_PATH}' não encontrado!")
     st.stop()
 
-# 2. Inicializar o Estado do Jogo
 if "board" not in st.session_state:
     game = Checkers()
     st.session_state.board = game.get_initial_board()
-    st.session_state.player = 1  # Humano começa (1)
+    st.session_state.player = 1
     st.session_state.game_over = False
-    st.session_state.message = "Sua vez! Você joga com as Brancas (x)."
+    st.session_state.message = "Bem-vindo à Arena! Você joga com as BRANCAS (⚪)."
 
 game = Checkers()
-mcts = MCTS(game, model, sims=150) # Sims ajustado para performance na web
+mcts = MCTS(game, model, sims=150)
 
-# Função para desenhar o tabuleiro como texto (simples e funcional)
-def render_board(board):
-    chars = {1: 'x', 2: 'X', -1: 'o', -2: 'O', 0: '.'}
-    board_str = "   0 1 2 3 4 5 6 7\n"
-    board_str += "  -----------------\n"
-    for r_idx, row in enumerate(board):
-        board_str += f"{r_idx} | {' '.join(chars[val] for val in row)} |\n"
-    board_str += "  -----------------"
-    return board_str
+def format_move_for_human(move):
+    """Transforma ((2,1), (3,2)) em algo mais legível"""
+    if isinstance(move, list): # Pulo múltiplo
+        path = " -> ".join([f"({r},{c})" for (r,c), _ in move] + [str(move[-1][1])])
+        return f"Salto Múltiplo: {path}"
+    else:
+        (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>'
+        
+    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 == -1: content = '<span class="piece-black">⚫</span>'
+            elif piece == -2: content = '<span class="piece-black king">👑</span>' # Dama preta (coroa dourada no CSS)
+            
+            html += f'<div class="square {color_class}">{content}</div>'
+            
+    html += '</div>'
+    return html
+
+st.title("♟️ AlphaCheckerZero Arena")
 
-# Layout principal
-col1, col2 = st.columns([2, 1])
+col1, col2 = st.columns([1.5, 1])
 
 with col1:
-    st.text_area("Tabuleiro", render_board(st.session_state.board), height=250, disabled=True, key="board_display")
+    st.markdown(render_board_html(st.session_state.board), unsafe_allow_html=True)
 
 with col2:
-    st.write("### Status")
-    st.info(st.session_state.message)
-
-    if st.button("Reiniciar Jogo"):
-        st.session_state.board = game.get_initial_board()
-        st.session_state.player = 1
-        st.session_state.game_over = False
-        st.session_state.message = "Jogo reiniciado. Sua vez!"
-        st.rerun()
-
-# Lógica do Jogo
-if not st.session_state.game_over:
-    # Verificar fim de jogo antes de qualquer coisa
-    result = game.check_game_over(st.session_state.board, st.session_state.player)
-    if result is not None:
-        st.session_state.game_over = True
-        if result == 1: st.session_state.message = "VOCÊ GANHOU! Parabéns Gabriel!"
-        elif result == -1: st.session_state.message = "A IA GANHOU. Mais sorte na próxima."
-        else: st.session_state.message = "EMPATE."
-        st.rerun()
-
-    # VEZ DO HUMANO (Player 1)
-    if st.session_state.player == 1:
-        valid_moves = game.get_valid_moves(st.session_state.board, 1)
-        
-        if not valid_moves:
-             # Se não tem movimentos e não deu game over acima, algo estranho aconteceu, mas tratamos como derrota
-             st.session_state.game_over = True
-             st.session_state.message = "Sem movimentos válidos. Você perdeu."
-             st.rerun()
-
-        # Criar lista de strings para o Selectbox
-        move_options = [str(m) for m in valid_moves]
-        selected_move_str = st.selectbox("Escolha sua jogada:", move_options)
-        
-        if st.button("Jogar"):
-            # Encontrar o movimento original baseado na string
-            move_idx = move_options.index(selected_move_str)
-            move = valid_moves[move_idx]
-            
-            # Aplicar movimento
-            st.session_state.board = game.apply_move(st.session_state.board, move)
-            st.session_state.player = -1 # Passa a vez para IA
-            st.session_state.message = "A IA está pensando..."
+    st.write("### 🕹️ Painel de Controle")
+    
+    if st.session_state.game_over:
+        st.warning(st.session_state.message)
+        if st.button("🔄 Jogar Novamente", use_container_width=True):
+            st.session_state.board = game.get_initial_board()
+            st.session_state.player = 1
+            st.session_state.game_over = False
+            st.session_state.message = "Novo jogo iniciado!"
             st.rerun()
-
-    # VEZ DA IA (Player -1)
     else:
-        with st.spinner("A AlphaCheckerZero está pensando..."):
-            # Pequeno delay para a interface atualizar e mostrar a mensagem
-            time.sleep(0.5) 
-            
-            valid_moves, policy = mcts.run(np.copy(st.session_state.board), -1)
+        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 = "A IA não tem movimentos. Você venceu!"
+                 st.session_state.message = "Sem movimentos válidos. Você perdeu. 😔"
                  st.rerun()
 
-            move = valid_moves[np.argmax(policy)]
+            # 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()))
             
-            st.session_state.board = game.apply_move(st.session_state.board, move)
-            st.session_state.player = 1 # Devolve a vez para Humano
-            st.session_state.message = f"IA jogou: {move}. Sua vez!"
-            st.rerun()
+            if st.button("✅ Confirmar Jogada", type="primary", use_container_width=True):
+                move = move_map[selected_desc]
+                st.session_state.board = game.apply_move(st.session_state.board, move)
+                st.session_state.player = -1
+                st.session_state.message = "A IA está calculando..."
+                st.rerun()
 
-else:
-    st.success(st.session_state.message)
+        else:
+            # Vez da IA
+            with st.spinner("A AlphaCheckerZero está pensando..."):
+                time.sleep(0.2) # UX Feel
+                
+                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)]
+                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.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.")