Improved the bot

Author: shashankp28

notebooks/benchmark.ipynb

@@ -1,13 +1,17 @@
 use axum::{ extract::{ State, Json }, response::IntoResponse, http::StatusCode };
 use serde::{ Deserialize, Serialize };
-use std::{ collections::HashMap, str::FromStr, sync::{ Arc, Mutex } };
+use std::{ str::FromStr, sync::Arc};
 use chess::Board;
 use crate::bot::{
     algorithm::root::search,
-    include::types::{ EngineState, ServerState, TT_TABLE_SIZE },
+    include::types::{
+        EngineState,
+        RepetitionHistory,
+        ServerState,
+        TranspositionTable,
+        TT_TABLE_SIZE,
+    },
 };
-use lru::LruCache;
-use std::num::NonZeroUsize;
 
 #[derive(Debug, Deserialize)]
 pub struct EvalRequest {
@@ -46,27 +50,26 @@ pub async fn eval_position_handler(
         }
     };
 
-    let mut history: HashMap<u64, u32> = HashMap::new();
+    let mut history = RepetitionHistory::new();
     for fen in &payload.history {
         if let Ok(board) = Board::from_str(fen) {
             let hash = board.get_hash();
-            *history.entry(hash).or_insert(0) += 1;
+            history.increment(hash);
         }
     }
 
-    let capacity = NonZeroUsize::new(TT_TABLE_SIZE).unwrap();
-    let transposition_table = Arc::new(Mutex::new(LruCache::new(capacity)));
+    let transposition_table = TranspositionTable::new(TT_TABLE_SIZE);
+
     let mut engine = EngineState {
         game_id: "eval_temp".to_string(),
         current_board,
         history,
-        statistics: HashMap::new(),
+        statistics: Default::default(),
         global_map: Arc::clone(&state.global_map),
         transposition_table,
     };
 
     let board = engine.current_board.clone();
-    // Default time values — can tweak or make them params if needed
     let (best_move, nodes, time_taken_ms, eval, depth) = search(
         payload.time_left_ms,
         payload.time_limit_ms,

src/api/get/get_eval.rs

@@ -1,8 +1,13 @@
 use axum::{ extract::State, response::IntoResponse, Json, http::StatusCode };
-use lru::LruCache;
 use serde::{ Deserialize, Serialize };
-use std::{ collections::HashMap, num::NonZeroUsize, str::FromStr, sync::{ Arc, Mutex } };
-use crate::bot::include::types::{ EngineState, ServerState, TT_TABLE_SIZE };
+use std::{ collections::HashMap, str::FromStr, sync::Arc };
+use crate::bot::include::types::{
+    EngineState,
+    RepetitionHistory,
+    ServerState,
+    TranspositionTable,
+    TT_TABLE_SIZE,
+};
 use chess::Board;
 
 #[derive(Debug, Deserialize)]
@@ -43,16 +48,16 @@ pub async fn new_game_handler(
         }
     };
 
-    let mut history: HashMap<u64, u32> = HashMap::new();
+    // ✅ Use the new RepetitionHistory wrapper
+    let mut history = RepetitionHistory::new();
     for fen in &payload.history {
         if let Ok(board) = Board::from_str(fen) {
             let hash = board.get_hash();
-            *history.entry(hash).or_insert(0) += 1;
+            history.increment(hash);
         }
     }
 
-    let capacity = NonZeroUsize::new(TT_TABLE_SIZE).unwrap();
-    let transposition_table = Arc::new(Mutex::new(LruCache::new(capacity)));
+    let transposition_table = TranspositionTable::new(TT_TABLE_SIZE);
     let engine = EngineState {
         game_id: payload.game_id.clone(),
         current_board,

src/api/post/add_game.rs

@@ -55,7 +55,7 @@ pub async fn make_move_handler(
 
     engine.current_board = engine.current_board.make_move_new(chess_move);
     let hash = engine.current_board.get_hash();
-    *engine.history.entry(hash).or_insert(0) += 1;
+    engine.history.increment(hash);
 
     let new_fen = engine.current_board.to_string();