Update app.py

app.py

@@ -9,28 +9,70 @@ import random
 # Configuration de la page
 st.set_page_config(page_title="Échecs contre IA", layout="wide")
 
-def handle_click(square_name):
-    """Gère les clics sur l'échiquier"""
-    if 'selected_square' not in st.session_state:
-        st.session_state.selected_square = None
+def render_chessboard():
+    """Render l'échiquier"""
+    squares = None
+    if st.session_state.selected_square is not None:
+        squares = []
+        from_square = chess.parse_square(st.session_state.selected_square)
+        for move in st.session_state.board.legal_moves:
+            if move.from_square == from_square:
+                squares.append(move.to_square)
+
+    board_svg = chess.svg.board(
+        board=st.session_state.board,
+        size=400,
+        lastmove=st.session_state.board.peek() if st.session_state.board.move_stack else None,
+        squares=chess.SquareSet(squares) if squares else None
+    )
+    
+    return board_svg
+
+def display_interactive_board():
+    """Affiche l'échiquier avec les contrôles interactifs"""
+    board_svg = render_chessboard()
+    
+    col1, col2, col3 = st.columns([1, 2, 1])
+    
+    with col2:
+        # Afficher l'échiquier
+        st.write(f"""
+            <div style="display: flex; justify-content: center; position: relative;">
+                {board_svg}
+                <div style="position: absolute; top: 0; left: 0; right: 0; bottom: 0; 
+                    display: grid; grid-template-columns: repeat(8, 1fr); 
+                    grid-template-rows: repeat(8, 1fr);">
+        """, unsafe_allow_html=True)
+        
+        # Créer les boutons transparents pour chaque case
+        for rank in range(7, -1, -1):
+            for file in range(8):
+                square_name = chess.square_name(chess.square(file, rank))
+                if st.button("", key=f"square_{square_name}", 
+                           use_container_width=True,
+                           help=square_name):
+                    handle_square_click(square_name)
         
+        st.write("</div></div>", unsafe_allow_html=True)
+
+def handle_square_click(square_name):
+    """Gère le clic sur une case"""
+    square = chess.parse_square(square_name)
+    
+    # Si aucune pièce n'est sélectionnée
     if st.session_state.selected_square is None:
-        # Premier clic - sélection de la pièce
-        square = chess.parse_square(square_name)
         piece = st.session_state.board.piece_at(square)
         if piece and piece.color == st.session_state.board.turn:
             st.session_state.selected_square = square_name
+            st.rerun()
+    # Si une pièce est déjà sélectionnée
     else:
-        # Deuxième clic - déplacement
         from_square = chess.parse_square(st.session_state.selected_square)
-        to_square = chess.parse_square(square_name)
-        move = chess.Move(from_square, to_square)
+        move = chess.Move(from_square, square)
         
         if move in st.session_state.board.legal_moves:
-            san_move = st.session_state.board.san(move)
             st.session_state.board.push(move)
-            st.session_state.moves.append(san_move)
-            st.session_state.selected_square = None
+            st.session_state.moves.append(st.session_state.board.san(move))
             
             # Tour de l'IA
             game_str = get_game_string()
@@ -39,54 +81,10 @@ def handle_click(square_name):
                 try_move(ai_move)
         
         st.session_state.selected_square = None
-
-def render_chessboard():
-    """Render un échiquier interactif en SVG avec des boutons pour chaque case"""
-    board_svg = chess.svg.board(
-        board=st.session_state.board,
-        size=400,
-        lastmove=st.session_state.board.peek() if st.session_state.board.move_stack else None,
-        squares=get_legal_moves_squares() if st.session_state.show_hints else None,
-    )
-    
-    st.write(f"""
-        <div style="display: flex; justify-content: center;">
-            <div style="position: relative; width: 400px;">
-                <div style="position: absolute; top: 0; left: 0; right: 0; bottom: 0;">
-                    {board_svg}
-                </div>
-                <div style="position: absolute; top: 0; left: 0; right: 0; bottom: 0; 
-                     display: grid; grid-template-columns: repeat(8, 1fr); 
-                     height: 400px;">
-    """, unsafe_allow_html=True)
-    
-    # Créer une grille de boutons transparents
-    for rank in range(7, -1, -1):
-        for file in range(8):
-            square = chess.square(file, rank)
-            square_name = chess.square_name(square)
-            if st.button("", key=f"square_{square_name}", 
-                        help=square_name,
-                        use_container_width=True):
-                handle_click(square_name)
-    
-    st.write("</div></div></div>", unsafe_allow_html=True)
-
-def get_legal_moves_squares():
-    """Obtenir les cases des coups légaux pour la pièce sélectionnée"""
-    squares = []
-    if hasattr(st.session_state, 'selected_square') and st.session_state.selected_square:
-        legal_moves = st.session_state.board.legal_moves
-        from_square = chess.parse_square(st.session_state.selected_square)
-        for move in legal_moves:
-            if move.from_square == from_square:
-                squares.append(chess.square_name(move.to_square))
-    return squares
+        st.rerun()
 
 def calculate_max_length(move_count):
-    """
-    Calcule le max_length optimal basé sur le nombre de coups joués
-    """
+    """Calcule le max_length optimal"""
     base_length = 10
     increment = 1
     max_length = 100
@@ -122,8 +120,6 @@ if 'moves' not in st.session_state:
     st.session_state.moves = []
 if 'selected_square' not in st.session_state:
     st.session_state.selected_square = None
-if 'show_hints' not in st.session_state:
-    st.session_state.show_hints = True
 
 def get_ai_move(prompt):
     """Obtient le coup de l'IA"""
@@ -137,10 +133,10 @@ def get_ai_move(prompt):
         else:
             moves = prompt.split()
             last_moves = " ".join(moves[-4:])
-            if len(moves) % 2 == 0:  # Si on vient de finir un coup noir
+            if len(moves) % 2 == 0:
                 next_move_num = f"{(len(moves)//2 + 1)}."
                 response = generator.generate(f"{last_moves} {next_move_num}")
-            else:  # Si on vient de finir un coup blanc
+            else:
                 response = generator.generate(f"{last_moves}")
         
         moves = response[0].split() if isinstance(response, list) else response.split()
@@ -192,12 +188,12 @@ def get_game_string():
 # Interface utilisateur
 st.title("♟️ Échecs contre IA")
 
-# Mise en page principale
+# Layout principal
 col1, col2 = st.columns([2, 1])
 
 with col1:
     # Échiquier interactif
-    render_chessboard()
+    display_interactive_board()
     
     # Input texte comme méthode alternative
     move = st.text_input(
@@ -227,9 +223,6 @@ with col2:
         - Roque: O-O ou O-O-O
     """)
     
-    # Option pour afficher les coups possibles
-    st.checkbox("Afficher les coups possibles", key="show_hints")
-    
     # Nouvelle partie
     if st.button("Nouvelle partie"):
         st.session_state.board = chess.Board()