Spaces:

Zual
/

humanchess

Sleeping

App Files Files Community

Zual commited on Oct 26, 2024

Commit

215ec13

verified ·

1 Parent(s): f92c57e

Update app.py

Browse files

Files changed (1) hide show

app.py +117 -79

app.py CHANGED Viewed

@@ -1,87 +1,112 @@
 import streamlit as st
 import chess
-import chess.svg
 import os
 import git
 import sys
 import random
 # Configuration de la page
 st.set_page_config(page_title="Échecs contre IA", layout="wide")
 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:
-        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:
-        from_square = chess.parse_square(st.session_state.selected_square)
-        move = chess.Move(from_square, square)
-        if move in st.session_state.board.legal_moves:
-            st.session_state.board.push(move)
-            st.session_state.moves.append(st.session_state.board.san(move))
-            # 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
-        st.rerun()
 def calculate_max_length(move_count):
     """Calcule le max_length optimal"""
@@ -106,11 +131,11 @@ 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
@@ -118,8 +143,6 @@ 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
 def get_ai_move(prompt):
     """Obtient le coup de l'IA"""
@@ -133,10 +156,10 @@ def get_ai_move(prompt):
         else:
             moves = prompt.split()
             last_moves = " ".join(moves[-4:])
-            if len(moves) % 2 == 0:
                 next_move_num = f"{(len(moves)//2 + 1)}."
                 response = generator.generate(f"{last_moves} {next_move_num}")
-            else:
                 response = generator.generate(f"{last_moves}")
         moves = response[0].split() if isinstance(response, list) else response.split()
@@ -188,12 +211,11 @@ def get_game_string():
 # Interface utilisateur
 st.title("♟️ Échecs contre IA")
-# Layout principal
 col1, col2 = st.columns([2, 1])
 with col1:
     # Échiquier interactif
-    display_interactive_board()
     # Input texte comme méthode alternative
     move = st.text_input(
@@ -212,7 +234,7 @@ with col2:
     # Instructions
     st.markdown("""
     **Comment jouer:**
-    - Cliquez sur une pièce puis sur la case de destination
     - Ou utilisez la notation algébrique :
         - Pion: e4, d5
         - Cavalier: Nf3, Nc6
@@ -227,7 +249,23 @@ with col2:
     if st.button("Nouvelle partie"):
         st.session_state.board = chess.Board()
         st.session_state.moves = []
-        st.session_state.selected_square = None
         st.rerun()
 # Gestion des coups via input texte

 import streamlit as st
 import chess
 import os
 import git
 import sys
 import random
+from streamlit.components.v1 import html
 # Configuration de la page
 st.set_page_config(page_title="Échecs contre IA", layout="wide")
 def render_chessboard():
+    """Render un échiquier interactif avec chessboard.js"""
+    # Obtenir la position FEN actuelle
+    fen = st.session_state.board.fen()
+    # Code JavaScript et HTML pour l'échiquier interactif
+    html_content = f"""
+    <!DOCTYPE html>
+    <html>
+    <head>
+        <meta charset="UTF-8">
+        <link
+            rel="stylesheet"
+            href="https://unpkg.com/@chrisoakman/[email protected]/dist/chessboard-1.0.0.min.css"
+            integrity="sha384-q94+BZtLrkL1/ohfjR8c6L+A6qzNH9R2hBLwyoAfu3i/WCvQjzL2RQJ3uNHDISdU"
+            crossorigin="anonymous">
+        <script
+            src="https://code.jquery.com/jquery-3.5.1.min.js"
+            integrity="sha384-ZvpUoO/+PpLXR1lu4jmpXWu80pZlYUAfxl5NsBMWOEPSjUn/6Z/hRTt8+pR6L4N2"
+            crossorigin="anonymous">
+        </script>
+        <script
+            src="https://unpkg.com/@chrisoakman/[email protected]/dist/chessboard-1.0.0.min.js"
+            integrity="sha384-8Vi8VHwn3vjQ9eUHUxex3JSN/NFqUg3QbPyX8kWyb93+8AC/pPWTzj+nHtbC5bxD"
+            crossorigin="anonymous">
+        </script>
+        <script src="https://cdnjs.cloudflare.com/ajax/libs/chess.js/0.10.3/chess.min.js"></script>
+    </head>
+    <body>
+        <div id="board" style="width: 400px; margin: auto;"></div>
+        <script>
+            // Initialisation du jeu
+            var board = null;
+            var game = new Chess('{fen}');
+            var $status = $('#status');
+            function onDragStart (source, piece, position, orientation) {{
+                // Autoriser uniquement les pièces blanches à être déplacées
+                if (game.turn() === 'b' ||
+                    game.game_over() ||
+                    (game.turn() === 'w' && piece.search(/^b/) !== -1)) {{
+                    return false;
+                }}
+            }}
+            function onDrop (source, target) {{
+                // Vérifier si le coup est légal
+                var move = game.move({{
+                    from: source,
+                    to: target,
+                    promotion: 'q'
+                }});
+                // Coup illégal
+                if (move === null) return 'snapback';
+                // Envoyer le coup à Streamlit
+                window.parent.postMessage({{
+                    type: 'move',
+                    move: move.san
+                }}, '*');
+            }}
+            function onSnapEnd () {{
+                board.position(game.fen());
+            }}
+            // Configuration de l'échiquier
+            var config = {{
+                position: '{fen}',
+                draggable: true,
+                onDragStart: onDragStart,
+                onDrop: onDrop,
+                onSnapEnd: onSnapEnd,
+                pieceTheme: 'https://unpkg.com/@chrisoakman/[email protected]/dist/img/chesspieces/wikipedia/{{piece}}.png'
+            }};
+            // Initialisation de l'échiquier
+            board = Chessboard('board', config);
+            // Gestion des messages de Streamlit
+            window.addEventListener('message', function(e) {{
+                if (e.data.type === 'update') {{
+                    board.position(e.data.fen);
+                    game.load(e.data.fen);
+                }}
+            }});
+        </script>
+        <style>
+            .highlight-square {{
+                box-shadow: inset 0 0 3px 3px yellow;
+            }}
+        </style>
+    </body>
+    </html>
+    """
+    html(html_content, height=450)
 def calculate_max_length(move_count):
     """Calcule le max_length optimal"""
     )
     return ChessGenerator(config)
+# Initialisation du modèle
 try:
     generator = setup_inference()
 except Exception as e:
+    st.error("Erreur lors de l'initialisation du modèle.")
     st.stop()
 # Initialisation de l'état
     st.session_state.board = chess.Board()
 if 'moves' not in st.session_state:
     st.session_state.moves = []
 def get_ai_move(prompt):
     """Obtient le coup de l'IA"""
         else:
             moves = prompt.split()
             last_moves = " ".join(moves[-4:])
+            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:  # 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()
 # Interface utilisateur
 st.title("♟️ Échecs contre IA")
 col1, col2 = st.columns([2, 1])
 with col1:
     # Échiquier interactif
+    render_chessboard()
     # Input texte comme méthode alternative
     move = st.text_input(
     # Instructions
     st.markdown("""
     **Comment jouer:**
+    - Glissez-déposez les pièces blanches
     - Ou utilisez la notation algébrique :
         - Pion: e4, d5
         - Cavalier: Nf3, Nc6
     if st.button("Nouvelle partie"):
         st.session_state.board = chess.Board()
         st.session_state.moves = []
+        st.rerun()
+# Gestion des messages du JavaScript
+if 'js_move' in st.session_state:
+    move = st.session_state.js_move
+    st.session_state.js_move = None
+    if try_move(move):
+        game_str = get_game_string()
+        with st.spinner("L'IA réfléchit..."):
+            ai_move = get_ai_move(game_str)
+            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()
 # Gestion des coups via input texte