Update app.py

app.py

@@ -18,9 +18,11 @@ def get_ai_move(fen):
         str or None: The AI's move in UCI format, or None if an error occurs.
     """
     # Prepare the prompt for the AI
-    prompt = f"""As a chess engine, analyze the following board position given in FEN notation:
+    prompt = f"""You are playing a chess game, and the current board position is given 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."""
+It's your turn to move. You need to provide the best legal chess move following the rules of chess. 
+The move should be in UCI format, which represents the starting and ending squares (for example, 'e2e4'). 
+Make sure the move you provide is legal and valid according to the current board position."""
 
     try:
         # Generate the response from the AI model
@@ -28,7 +30,7 @@ It's your turn to move. Provide the best move in UCI format (e.g., e2e4). Only p
             model="gemini-1.5-flash",
             prompt=prompt,
             temperature=0,  # To make the output more deterministic
-            max_output_tokens=5
+            max_output_tokens=10
         )
         ai_move = response.result.strip()
         print(f"AI move: {ai_move}")