Update app.py

app.py

@@ -1,51 +1,62 @@
 import gradio as gr
 import chess
 import chess.svg
-import io
-import random
-from PIL import Image
-import cairosvg
+import os
+import google.generativeai as genai
+
+# Configure the Google Generative AI client
+genai.configure(api_key=os.environ["API_KEY"])
+
+board = chess.Board()
+
+def get_ai_move(fen):
+    """
+    Gets the AI's next move using Google Gemini via google.generativeai.
+
+    Args:
+        fen (str): The current board position in FEN notation.
+
+    Returns:
+        str or None: The AI's move in UCI format, or None if an error occurs.
+    """
+    # Prepare the prompt for the AI
+    prompt = f"""You are a chess engine. Given the following board position in FEN notation:
+{fen}
+It's your turn to move. Provide the best move in UCI format (e.g., 'e2e4'). Only provide the move, without any additional text."""
 
-def make_move(move_san, board_fen):
-    board = chess.Board(fen=board_fen)
     try:
-        move = board.parse_san(move_san)
-        if move in board.legal_moves:
-            board.push(move)
-            if board.is_game_over():
-                message = "Game over. You win!"
-            else:
-                # Computer makes a random move
-                computer_move = random.choice(list(board.legal_moves))
-                board.push(computer_move)
-                message = f"Computer played: {board.san(computer_move)}"
-                if board.is_game_over():
-                    message += "\nGame over. Computer wins."
-            # Generate board image
-            svg_data = chess.svg.board(board=board, size=400)
-            png_data = cairosvg.svg2png(bytestring=svg_data)
-            image = Image.open(io.BytesIO(png_data))
-            return image, board.fen(), message
+        # Generate the response from the AI model
+        response = genai.generate_text(
+            model="gemini-1.5-flash",  # Use the Gemini model
+            prompt=prompt,
+            temperature=0,  # To make the output more deterministic
+            max_output_tokens=5
+        )
+        ai_move = response.result.strip()
+
+        # Validate the move
+        legal_moves = [move.uci() for move in board.legal_moves]
+        if ai_move in legal_moves:
+            return ai_move
         else:
-            return gr.update(), board_fen, "Illegal move."
+            print(f"Invalid move received from the AI: {ai_move}")
+            return None
     except Exception as e:
-        return gr.update(), board_fen, f"Error: {str(e)}"
-
-def get_initial_board():
-    board = chess.Board()
-    svg_data = chess.svg.board(board=board, size=400)
-    png_data = cairosvg.svg2png(bytestring=svg_data)
-    image = Image.open(io.BytesIO(png_data))
-    return image, board.fen()
-
-with gr.Blocks() as demo:
-    initial_image, initial_fen = get_initial_board()
-    img = gr.Image(value=initial_image, interactive=False)
-    move_input = gr.Textbox(label="Enter your move in standard notation (e.g., e4, Nf3, Bb5)")
-    output = gr.Textbox(label="Message")
-    board_state = gr.State(value=initial_fen)
-    submit_button = gr.Button("Submit Move")
-    
-    submit_button.click(fn=make_move, inputs=[move_input, board_state], outputs=[img, board_state, output])
-
-demo.launch()
+        print(f"Error occurred: {e}")
+        return None
+
+def board_to_svg(board):
+    return chess.svg.board(board=board)
+
+def make_move(user_move):
+    global board
+    try:
+        # Player's move
+        board.push_san(user_move)
+    except ValueError:
+        return gr.update(value=board_to_svg(board)), "Invalid move. Try again."
+
+    # Check for game over
+    if board.is_game_over():
+        result = board.result()
+        board_svg = board_to_svg(b