AI2GTP

katrain.py

@@ -36,7 +36,10 @@ class KaTrainGui(BoxLayout):
         self.logger = lambda message, level=OUTPUT_INFO: self.log(message, level)
 
         self._load_config()
+
         self.debug_level = self.config("debug/level", OUTPUT_INFO)
+        self.ai_settings = self.config("ai")
+        self.controls.ai_mode_groups["W"].values = self.controls.ai_mode_groups["B"].values = self.ai_settings.keys()
         self.message_queue = Queue()
 
         self._keyboard = Window.request_keyboard(None, self, "")
@@ -91,8 +94,8 @@ class KaTrainGui(BoxLayout):
             self.game.analyze_undo(cn, self.config("trainer"))  # not via message loop
         if (
             cn.analysis_ready
-            and "ai" in self.controls.player_mode(cn.next_player)
-            and not "pause" in self.controls.ai_mode(cn.next_player)
+            and "ai" in self.controls.player_mode(cn.next_player).lower()
+            and not "pause" in self.controls.ai_mode(cn.next_player).lower()
             and not cn.children
             and not self.game.ended
             and not (auto_undo and cn.auto_undo is None)
@@ -141,7 +144,10 @@ class KaTrainGui(BoxLayout):
 
     def _do_ai_move(self, node=None):
         if node is None or self.game.current_node == node:
-            ai_move(self.game, self.controls.ai_mode(self.game.current_node.next_player), self.config("ai"))
+            mode = self.controls.ai_mode(self.game.current_node.next_player)
+            settings = self.ai_settings[mode]
+            if settings:
+                ai_move(self.game, mode, settings)
 
     def _do_undo(self, n_times=1):
         self.game.undo(n_times)