Spaces:

Zual
/

humanchess

Sleeping

App Files Files Community

Zual commited on Oct 26, 2024

Commit

7678641

verified ·

1 Parent(s): 5fcf9d2

Update app.py

Browse files

Files changed (1) hide show

app.py +53 -145

app.py CHANGED Viewed

@@ -1,122 +1,33 @@
 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://cdnjs.cloudflare.com/ajax/libs/chessboard-js/1.0.0/chessboard-1.0.0.min.css"
-            integrity="sha512-TU/clvRaSqKB43MX6dvJPgxm0ytgRrg/6tkbTPCFwOXG6Ej70nGCkcFkatrbcPHqR0mlPCy3WuYNg936BGvNOg=="
-            crossorigin="anonymous"
-        />
-        <script
-            src="https://code.jquery.com/jquery-3.6.0.min.js"
-            integrity="sha256-/xUj+3OJU5yExlq6GSYGSHk7tPXikynS7ogEvDej/m4="
-            crossorigin="anonymous">
-        </script>
-        <script
-            src="https://cdnjs.cloudflare.com/ajax/libs/chessboard-js/1.0.0/chessboard-1.0.0.min.js"
-            integrity="sha512-WfASs5HtTgTL/eZs0Td0JVJZ+8tNxF/MDNU99TzJI8D4jXrZdQZRO9G7EPIVQhvmb7yUEEH+AKY9PfGi4pqrw=="
-            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}');
-            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://cdnjs.cloudflare.com/ajax/libs/chessboard-js/1.0.0/img/chesspieces/wikipedia/{{piece}}.png'
-            }};
-            // Initialisation de l'échiquier
-            board = Chessboard('board', config);
-            // S'assurer que l'échiquier est responsive
-            $(window).resize(board.resize);
-            // 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"""
-    base_length = 10
-    increment = 1
-    max_length = 100
-    return min(base_length + (move_count * increment), max_length)
 # Setup du modèle
 @st.cache_resource
@@ -134,11 +45,11 @@ def setup_inference():
     )
     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
@@ -147,6 +58,13 @@ if 'board' not in st.session_state:
 if 'moves' not in st.session_state:
     st.session_state.moves = []
 def get_ai_move(prompt):
     """Obtient le coup de l'IA"""
     moves_count = len(st.session_state.moves)
@@ -159,10 +77,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
                 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()
@@ -182,9 +100,8 @@ def get_ai_move(prompt):
         if legal_moves:
             random_move = random.choice(legal_moves)
             return st.session_state.board.san(random_move)
     except Exception as e:
-        print(f"Erreur dans get_ai_move: {e}")
         return None
 def try_move(move_str):
@@ -211,24 +128,34 @@ def get_game_string():
             result.append(st.session_state.moves[i+1])
     return " ".join(result)
 # 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(
-        "Ou saisissez votre coup ici",
         key=f"move_input_{len(st.session_state.moves)}",
         placeholder="ex: e4, Nf3, O-O"
     )
 with col2:
-    # Informations de la partie
     st.subheader("Partie en cours")
     game_str = get_game_string()
     if game_str:
@@ -237,15 +164,13 @@ with col2:
     # Instructions
     st.markdown("""
     **Comment jouer:**
-    - Glissez-déposez les pièces blanches
-    - Ou utilisez la notation algébrique :
-        - Pion: e4, d5
-        - Cavalier: Nf3, Nc6
-        - Fou: Bc4, Be7
-        - Tour: Ra1, Rd8
-        - Dame: Qd1, Qh4
-        - Roi: Ke2, Kg8
-        - Roque: O-O ou O-O-O
     """)
     # Nouvelle partie
@@ -254,24 +179,7 @@ with col2:
         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
 if move:
     if try_move(move):
         game_str = get_game_string()

 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_board():
+    """Rendre l'échiquier en SVG avec mise en évidence des coups légaux"""
+    # Déterminer les cases à mettre en évidence
+    squares = None
+    last_move = None
+    if st.session_state.board.move_stack:
+        last_move = st.session_state.board.peek()
+    # Créer le SVG
+    board_svg = chess.svg.board(
+        board=st.session_state.board,
+        size=400,
+        lastmove=last_move,
+        check=st.session_state.board.king(st.session_state.board.turn) if st.session_state.board.is_check() else None
+    )
+    # On utilise unsafe_allow_html=True car on sait que le SVG est sûr (généré par python-chess)
+    st.write(board_svg, unsafe_allow_html=True)
 # Setup du modèle
 @st.cache_resource
     )
     return ChessGenerator(config)
+# Initialisation
 try:
     generator = setup_inference()
 except Exception as e:
+    st.error(f"Erreur lors de l'initialisation: {str(e)}")
     st.stop()
 # Initialisation de l'état
 if 'moves' not in st.session_state:
     st.session_state.moves = []
+def calculate_max_length(move_count):
+    """Calcule le max_length optimal"""
+    base_length = 10
+    increment = 1
+    max_length = 100
+    return min(base_length + (move_count * increment), max_length)
 def get_ai_move(prompt):
     """Obtient le coup de l'IA"""
     moves_count = len(st.session_state.moves)
         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()
         if legal_moves:
             random_move = random.choice(legal_moves)
             return st.session_state.board.san(random_move)
     except Exception as e:
+        st.error(f"Erreur lors de la génération du coup de l'IA: {str(e)}")
         return None
 def try_move(move_str):
             result.append(st.session_state.moves[i+1])
     return " ".join(result)
+def display_legal_moves():
+    """Affiche les coups légaux possibles"""
+    legal_moves = [st.session_state.board.san(move) for move in st.session_state.board.legal_moves]
+    st.text("Coups légaux possibles:")
+    st.code(", ".join(legal_moves))
 # Interface utilisateur
 st.title("♟️ Échecs contre IA")
 col1, col2 = st.columns([2, 1])
 with col1:
+    # Échiquier
+    render_board()
+    # Input du coup
     move = st.text_input(
+        "Votre coup",
         key=f"move_input_{len(st.session_state.moves)}",
         placeholder="ex: e4, Nf3, O-O"
     )
+    # Afficher les coups légaux si demandé
+    if st.checkbox("Afficher les coups possibles"):
+        display_legal_moves()
 with col2:
+    # Historique et état de la partie
     st.subheader("Partie en cours")
     game_str = get_game_string()
     if game_str:
     # Instructions
     st.markdown("""
     **Comment jouer:**
+    - Pion: e4, d5
+    - Cavalier: Nf3, Nc6
+    - Fou: Bc4, Be7
+    - Tour: Ra1, Rd8
+    - Dame: Qd1, Qh4
+    - Roi: Ke2, Kg8
+    - Roque: O-O ou O-O-O
     """)
     # Nouvelle partie
         st.session_state.moves = []
         st.rerun()
+# Logique du jeu
 if move:
     if try_move(move):
         game_str = get_game_string()