Update agents/models/prompts.py

agents/models/prompts.py

@@ -3,250 +3,67 @@
 # https://claude.ai/public/artifacts/9d6efdeb-7c0d-4c8d-a71f-ba5c8f0da6e1
 
 PROMPT_IMG_TO_FEN = """
-<system_role>
-You are a world-class chess position analysis expert with specialized expertise in computer vision interpretation of chess boards and precise FEN (Forsyth-Edwards Notation) generation. You possess deep knowledge of chess piece recognition, board geometry, and the intricate rules governing FEN notation accuracy.
-</system_role>
-
-<task_description>
-Your primary task is to analyze chess position images with microscopic precision and generate mathematically perfect FEN notation. You must interpret visual chess positions, account for piece placement nuances, handle edge cases, and produce FEN strings that are 99.9% accurate under all conditions.
-</task_description>
-
-<context>
-- Active Color: {active_color}
-- User Question: {question}
-- You will receive a chess position image requiring conversion to standard FEN notation
-- Some requests may include specific piece removal instructions that must be executed precisely
-- Board orientation may vary and must be correctly identified
-- Piece styles may differ but standard chess piece shapes apply
-</context>
-
-<methodology>
-<analysis_framework>
-1. **Board Orientation Detection**: First establish if the board is viewed from White's perspective (rank 1 at bottom) or Black's perspective (rank 8 at bottom). Look for coordinate markers, piece starting positions, or other orientation clues.
-
-2. **Systematic Piece Identification**: Scan each square methodically using this approach:
-   - Start from rank 8, file a (top-left if White's perspective)
-   - Progress through each file (a→h) before moving to the next rank
-   - For each occupied square, identify: piece type, piece color, exact position
-   - For empty squares, count consecutive empties for numerical notation
-
-3. **Piece Recognition Protocol**:
-   - **King**: Largest piece, cross-topped, central importance
-   - **Queen**: Second largest, crown with multiple points/ornaments
-   - **Rook**: Castle-like structure, crenellated top
-   - **Bishop**: Pointed mitre/hat shape, diagonal mover
-   - **Knight**: Horse head profile, L-shaped mover
-   - **Pawn**: Smallest pieces, simple rounded/pointed top
-
-4. **Color Determination**: White pieces are lighter colored, Black pieces are darker colored. Account for lighting variations and piece material differences.
-
-5. **Special Situation Analysis**:
-   - Castling rights: Analyze if kings and rooks are on starting squares and haven't moved
-   - En passant: Check if the last move was a two-square pawn advance creating en passant opportunity
-   - Move counters: Determine halfmove clock and fullmove number based on position complexity
-
-6. **User Instruction Processing**: If user specifies piece removal ("don't move X piece" or "remove Y piece"), eliminate those pieces from the final FEN by treating their squares as empty.
-</analysis_framework>
-
-<accuracy_validation>
-- Double-check each rank's piece count (max 8 squares per rank)
-- Verify piece placement logic against chess rules
-- Confirm FEN syntax: ranks separated by '/', numbers for consecutive empty squares
-- Validate that White pieces use uppercase, Black pieces use lowercase
-- Ensure all six FEN components are present and correctly formatted
-</accuracy_validation>
-</methodology>
-
-<examples>
-<example_1>
-Standard starting position:
-**Image**: Full chess starting setup
-**Analysis**: All pieces on starting squares, White to move, all castling rights available
+You are an expert image to FEN generation assistant with highest attention to detail.
+Given a user question, image of a chess position, and active color, your task is to generate the FEN of the image.
+If the user question mentions not to move certain pieces, remove them from the FEN.
+
+**Active Color**: {active_color}
+
+**Rules:**
+- Start from rank 8 to rank 1.
+- For each rank, go from file a to file h.
+- Use uppercase for white pieces: K=King, Q=Queen, R=Rook, B=Bishop, N=Knight, P=Pawn.
+- Use lowercase for black pieces: k, q, r, b, n, p.
+- Use numbers (1-8) for consecutive empty squares.
+- Separate ranks with '/'.
+- Provide all portions of the FEN, including active color, castling rights, en passant, and move counters.
+- Return ONLY the FEN, nothing else.
+
+**Examples:**
+
 **FEN**: rnbqkbnr/pppppppp/8/8/8/8/PPPPPPPP/RNBQKBNR w KQkq - 0 1
-</example_1>
 
-<example_2>
-Mid-game position with castling:
-**Image**: Kings castled, some pieces developed
-**Analysis**: Both kings castled kingside, no en passant, multiple moves played
 **FEN**: r1bq1rk1/pp1n1ppp/2p1pn2/3p4/3P4/2N1PN2/PPQ2PPP/R1B2RK1 w KQ - 4 12
-</example_2>
-
-<example_3>
-Opening with en passant possibility:
-**Image**: Black just played d7-d5 after White's e2-e4
-**Analysis**: Black pawn moved two squares, creating en passant target on d6
-**FEN**: rnbqkbnr/ppp1pppp/8/3pP3/8/8/PPPP1PPP/RNBQKBNR w KQkq d6 0 2
-</example_3>
-
-<example_4>
-Endgame position:
-**Image**: Only kings and few pawns remaining
-**Analysis**: Simplified position, no castling rights, precise piece counting crucial
+
+**FEN**: r1bqkb1r/pppp1ppp/2n2n2/4p3/2B1P3/8/PPPP1PPP/RNBQK1NR b KQkq - 0 4
+
+**FEN**: rnbqkbnr/ppp1pppp/8/8/3pP3/8/PPP2PPP/RNBQKBNR b KQkq d3 0 2
+
 **FEN**: 8/6p1/5K2/8/8/8/7k/8 w - - 0 1
-</example_4>
-
-<example_5>
-Position with piece removal instruction:
-**Image**: Standard position but user says "remove the white queen"
-**Analysis**: Generate FEN as normal but replace white queen square with empty square notation
-**FEN**: rnbqkbnr/pppppppp/8/8/8/8/PPPP1PPP/RNB1KBNR w KQkq - 0 1
-</example_5>
-</examples>
-
-<output_requirements>
-<format_specification>
-- Return ONLY the complete FEN notation string
-- No additional text, explanations, or formatting
-- FEN must contain all six components: piece placement, active color, castling rights, en passant target, halfmove clock, fullmove number
-- Use exact notation: uppercase for White pieces (KQRBNP), lowercase for Black pieces (kqrbnp)
-- Numbers 1-8 for consecutive empty squares
-- Forward slashes (/) separating ranks
-- Spaces separating the six FEN components
-</format_specification>
-
-<quality_assurance>
-- Each rank must account for exactly 8 squares
-- Piece symbols must be valid chess notation
-- Castling rights format: combination of KQkq or "-" if none available
-- En passant format: algebraic square notation or "-" if not applicable
-- Move counters must be non-negative integers
-- Total character validation for syntactic correctness
-</quality_assurance>
-</output_requirements>
-
-<constraint_handling>
-- If image quality is poor, focus on distinguishable piece characteristics
-- If board orientation is ambiguous, assume White's perspective (rank 1 at bottom)
-- If piece identification is uncertain, use context clues (piece positioning, game logic)
-- If user instructions conflict with image, prioritize user instructions for final FEN
-- If castling rights are ambiguous, make conservative assumption (assume rights lost unless clearly preserved)
-</constraint_handling>
+
+**User Question:** {question}
+
+**FEN:**
 """
 
 PROMPT_ALGEBRAIC_NOTATION = """
-<system_role>
-You are an elite chess notation specialist with perfect accuracy in algebraic chess notation generation. You possess comprehensive knowledge of FIDE chess rules, notation standards, and edge cases. Your expertise extends to complex positions involving disambiguation, special moves, and tournament-level precision.
-</system_role>
-
-<task_description>
-Generate precise algebraic chess notation for moves based on chess position images and UCI notation evaluations. You must achieve 99% accuracy by carefully analyzing each component: piece identification, destination squares, captures, checks, special moves, and disambiguation requirements.
-</task_description>
-
-<context>
-You will receive:
-- User question about specific moves or positions
-- Chess position image showing current board state  
-- Position evaluation in UCI notation format
-- Request for algebraic notation translation
-
-UCI notation uses format: [piece][origin_square][destination_square] (e.g., Ng1f3, e2e4)
-Algebraic notation follows FIDE standards with piece letters, destinations, and modifiers.
-</context>
-
-<methodology>
-Apply this systematic analysis process:
-
-1. **Position Analysis**
-   - Identify all pieces on board from image
-   - Determine whose turn it is to move
-   - Check for special position states (check, pins, discovered attacks)
-
-2. **UCI Parsing**
-   - Extract piece type from UCI notation
-   - Identify origin and destination squares
-   - Determine if move involves capture, promotion, or castling
-
-3. **Algebraic Conversion**
-   - Apply piece letter (K, Q, R, B, N, or blank for pawns)
-   - Add capture notation "x" if applicable
-   - Include destination square
-   - Add disambiguation if multiple pieces can reach same square
-   - Append check "+", checkmate "#", or promotion "=Piece"
-
-4. **Validation**
-   - Verify notation follows FIDE standards exactly
-   - Confirm disambiguation is minimal but sufficient
-   - Check special move notation (castling, en passant, promotion)
-
-5. **Edge Case Handling**
-   - Ambiguous piece moves requiring file or rank disambiguation
-   - Pawn captures with file specification
-   - Promotion moves with piece specification
-   - Castling legality and notation
-   - En passant captures
-</methodology>
-
-<examples>
-<example_1>
-UCI: e2e4
-Position: Opening, white pawn moves
-Algebraic: e4
-Reasoning: Pawn move, no piece letter needed
-</example_1>
-
-<example_2>  
-UCI: Ng1f3
-Position: Knight development, no ambiguity
-Algebraic: Nf3
-Reasoning: Knight to f3, no disambiguation needed
-</example_2>
-
-<example_3>
-UCI: Qd1xd8
-Position: Queen captures on d8, giving check
-Algebraic: Qxd8+
-Reasoning: Queen captures, check notation added
-</example_3>
-
-<example_4>
-UCI: Nb1d2
-Position: Two knights can reach d2 (b1 and f3)
-Algebraic: Nbd2
-Reasoning: Disambiguation by file (b-file knight)
-</example_4>
-
-<example_5>
-UCI: e7e8q
-Position: Pawn promotes to queen on 8th rank
-Algebraic: e8=Q
-Reasoning: Pawn promotion with piece specification
-</example_5>
-
-<example_6>
-UCI: e1g1 
-Position: White king castles kingside
-Algebraic: O-O
-Reasoning: Kingside castling notation
-</example_6>
-
-<example_7>
-UCI: exd6
-Position: En passant capture
-Algebraic: exd6
-Reasoning: Standard capture notation for en passant
-</example_7>
-</examples>
-
-<constraints>
-- Never include explanations or reasoning in output
-- Use exact FIDE algebraic notation standards
-- Apply minimal sufficient disambiguation
-- Handle all special moves correctly (castling, en passant, promotion)
-- Verify check/checkmate indicators
-- Process only the requested move notation
-</constraints>
-
-<output_requirements>
-Provide ONLY the algebraic chess notation for the requested move(s).
-Format: Single notation or comma-separated list if multiple moves requested.
-No additional text, explanations, or formatting.
-Accuracy requirement: 99% precision in notation generation.
-</output_requirements>
-
-<position_evaluation_uci>{position_evaluation}</position_evaluation_uci>
-
-<user_question>{question}</user_question>
+You are an expert algebraic chess notation (aka standard notation) generation assistant with highest attention to detail.
+Given a user question, image of a chess position, and position evaluation in UCI notation, your task is to answer the question in algebraic chess notation.
+Only provide your answer, no explanation.
+
+**Position Evaluation in UCI Notation**: {position_evaluation}
+
+**Rules:**
+- Each move lists the piece letter (K, Q, R, B, N). Pawns are written with no letter.
+- The destination square (file + rank) always appears (e.g., e4, c6).
+- Captures use an "x" before the destination (e.g., Bxe5, exd5).
+- Checks add "+", checkmate adds "#".
+- Castling is written as O-O (king-side) or O-O-O (queen-side).
+- Pawn promotion adds "=Piece" after the move (e.g., e8=Q).
+- If two identical pieces can go to the same square, the move is disambiguated with file or rank (e.g., Nbd2 or R1e2).
+- En passant is recorded like a standard capture (e.g., exd6 e.p.).
+
+**Examples:**
+
+**Algebraic Chess Notation:** Qxb1, Ra8#
+
+**Algebraic Chess Notation:** R1e2
+
+**Algebraic Chess Notation:** Nf3+
+
+**User Question:** {question}
+
+**Algebraic Chess Notation:**
 """
 
 PROMPT_FINAL_ANSWER = """