diff --git a/src/blunderboard/blunderevaluator.py b/src/blunderboard/blunderevaluator.py index 0b8af1e..766d6a4 100644 --- a/src/blunderboard/blunderevaluator.py +++ b/src/blunderboard/blunderevaluator.py @@ -46,9 +46,11 @@ class BlunderEvaluator: self.current_wdl = self.api_wdl() self.wdls: list = [] self.current_fen = self.engine.get_fen_position() + self.white_to_move = True def reset(self): self.engine.set_position() + self.white_to_move = True def move(self, move) -> None: """ @@ -69,6 +71,10 @@ class BlunderEvaluator: # path self.play_sound("blunder") print("Blunder!") + if self.white_to_move: + self.white_to_move = False + elif not self.white_to_move: + self.white_to_move = True else: print("Invalid move") self.play_sound("illegal") @@ -112,9 +118,7 @@ class BlunderEvaluator: """ if len(self.wdls) > 1: # Don't check for blunders on the first move previous_wdl = self.wdls[len(self.evaluations) - 2] - if abs(previous_wdl[0] - self.current_wdl[0]) > 300: - return True - elif abs(previous_wdl[2] - self.current_wdl[2]) > 300: + if abs(previous_wdl[0] - self.current_wdl[2]) > 300: return True else: return False