Update app.py

app.py

@@ -5,14 +5,43 @@ import os
 import git
 import sys
 import random
-import base64
-from streamlit.components.v1 import html
 
 # 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
+        
+    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
+    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)
+        
+        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
+            
+            # Tour de l'IA
+            game_str = get_game_string()
+            ai_move = get_ai_move(game_str)
+            if ai_move:
+                try_move(ai_move)
+        
+        st.session_state.selected_square = None
+
 def render_chessboard():
-    """Render un échiquier interactif en SVG"""
+    """Render un échiquier interactif en SVG avec des boutons pour chaque case"""
     board_svg = chess.svg.board(
         board=st.session_state.board,
         size=400,
@@ -20,35 +49,33 @@ def render_chessboard():
         squares=get_legal_moves_squares() if st.session_state.show_hints else None,
     )
     
-    # Style pour centrer l'échiquier et ajouter des styles supplémentaires
-    html_content = f"""
-    <div style="display: flex; justify-content: center; margin-bottom: 20px;">
-        <style>
-            .chess-board {{
-                cursor: pointer;
-            }}
-            .square.legal-move {{
-                fill: #afa !important;
-                opacity: 0.5;
-            }}
-            .square.selected {{
-                fill: #0af !important;
-                opacity: 0.5;
-            }}
-            .highlight {{
-                fill: #ff0 !important;
-                opacity: 0.5;
-            }}
-        </style>
-        {board_svg}
-    </div>
-    """
-    html(html_content, height=450)
+    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 st.session_state.selected_square:
+    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:
@@ -56,18 +83,10 @@ def get_legal_moves_squares():
                 squares.append(chess.square_name(move.to_square))
     return squares
 
-# Initialisation des variables de session
-if 'board' not in st.session_state:
-    st.session_state.board = chess.Board()
-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 calculate_max_length(move_count):
-    """Calcule le max_length optimal"""
+    """
+    Calcule le max_length optimal basé sur le nombre de coups joués
+    """
     base_length = 10
     increment = 1
     max_length = 100
@@ -89,12 +108,23 @@ def setup_inference():
     )
     return ChessGenerator(config)
 
+# Initialisation
 try:
     generator = setup_inference()
 except Exception as e:
     st.error("Erreur lors de l'initialisation.")
     st.stop()
 
+# Initialisation de l'état
+if 'board' not in st.session_state:
+    st.session_state.board = chess.Board()
+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"""
     moves_count = len(st.session_state.moves)
@@ -102,15 +132,15 @@ def get_ai_move(prompt):
     generator.config.max_length = dynamic_max_length
     
     try:
-        if not prompt:
+        if not prompt:  # Premier coup
             response = generator.generate("1.")
         else:
             moves = prompt.split()
             last_moves = " ".join(moves[-4:])
-            if len(moves) % 2 == 0:
+            if len(moves) % 2 == 0:  # Si on vient de finir un coup noir
                 next_move_num = f"{(len(moves)//2 + 1)}."
                 response = generator.generate(f"{last_moves} {next_move_num}")
-            else:
+            else:  # Si on vient de finir un coup blanc
                 response = generator.generate(f"{last_moves}")
         
         moves = response[0].split() if isinstance(response, list) else response.split()
@@ -125,6 +155,7 @@ def get_ai_move(prompt):
         except ValueError:
             pass
         
+        # En cas d'échec, jouer un coup légal aléatoire
         legal_moves = list(st.session_state.board.legal_moves)
         if legal_moves:
             random_move = random.choice(legal_moves)
@@ -161,28 +192,22 @@ def get_game_string():
 # Interface utilisateur
 st.title("♟️ Échecs contre IA")
 
+# Mise en page principale
 col1, col2 = st.columns([2, 1])
 
 with col1:
-    # Affichage de l'échiquier
+    # Échiquier interactif
     render_chessboard()
     
     # Input texte comme méthode alternative
     move = st.text_input(
-        "Saisissez votre coup",
+        "Ou saisissez votre coup ici",
         key=f"move_input_{len(st.session_state.moves)}",
         placeholder="ex: e4, Nf3, O-O"
     )
 
-    # Option pour afficher les coups légaux
-    st.checkbox("Afficher les coups possibles", value=True, key="show_hints")
-
 with col2:
-    # Informations sur la partie
-    current_max_length = calculate_max_length(len(st.session_state.moves))
-    st.info(f"Longueur de génération actuelle: {current_max_length} tokens")
-    
-    # Historique
+    # Informations de la partie
     st.subheader("Partie en cours")
     game_str = get_game_string()
     if game_str:
@@ -202,6 +227,9 @@ 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()
@@ -209,42 +237,6 @@ with col2:
         st.session_state.selected_square = None
         st.rerun()
 
-# Gestion des clics sur l'échiquier (à implémenter avec st.experimental_get_query_params)
-params = st.experimental_get_query_params()
-if 'square' in params:
-    clicked_square = params['square'][0]
-    if st.session_state.selected_square is None:
-        # Premier clic - sélection de la pièce
-        piece = st.session_state.board.piece_at(chess.parse_square(clicked_square))
-        if piece and piece.color == st.session_state.board.turn:
-            st.session_state.selected_square = clicked_square
-    else:
-        # Deuxième clic - déplacement
-        from_square = chess.parse_square(st.session_state.selected_square)
-        to_square = chess.parse_square(clicked_square)
-        move = chess.Move(from_square, to_square)
-        
-        if move in st.session_state.board.legal_moves:
-            # Appliquer le coup
-            st.session_state.board.push(move)
-            st.session_state.moves.append(st.session_state.board.san(move))
-            st.session_state.selected_square = None
-            
-            # Tour de l'IA
-            with st.spinner("L'IA réfléchit..."):
-                ai_move = get_ai_move(get_game_string())
-                if ai_move and try_move(ai_move):
-                    if st.session_state.board.is_checkmate():
-                        st.success("Échec et mat!")
-                    elif st.session_state.board.is_game_over():
-                        st.info("Partie terminée!")
-                else:
-                    st.error("Problème avec le coup de l'IA")
-            st.rerun()
-        else:
-            st.session_state.selected_square = None
-    st.rerun()
-
 # Gestion des coups via input texte
 if move:
     if try_move(move):