clean up message log and d3 visualization

Author: TataKKKL

visual-tree-search-app/components/MCTSVisual.tsx

@@ -483,14 +483,14 @@ const MCTSVisual: React.FC<SimpleSearchVisualProps> = ({ messages }) => {
             <span className="w-3 h-3 rounded-full inline-block mr-1 bg-blue-500 dark:bg-blue-600"></span>
             <span className="text-gray-700 dark:text-gray-300">Selected</span>
           </div>
-          <div className="flex items-center">
+          {/* <div className="flex items-center">
             <span className="w-3 h-3 rounded-full inline-block mr-1 bg-green-500 dark:bg-green-600"></span>
             <span className="text-gray-700 dark:text-gray-300">Sim Start</span>
           </div>
           <div className="flex items-center">
             <span className="w-3 h-3 rounded-full inline-block mr-1 bg-orange-500 dark:bg-orange-600"></span>
             <span className="text-gray-700 dark:text-gray-300">Simulated</span>
-          </div>
+          </div> */}
         </div>
       </div>
       <div 

visual-tree-search-backend/app/api/lwats/agents_async/SearchAgents/mcts_agent.py

@@ -264,11 +264,12 @@ async def mcts_search(self, websocket=None) -> Optional[LATSNode]:
             print(f"{GREEN}Step 1: Node Selection{RESET}")
             await self.websocket_step_start(step=1, step_name="node_selection", websocket=websocket)
             selected_node = await self.node_selection(self.root_node, websocket)
-            tree_data = self._get_tree_data()
-            if websocket:
-                await self.websocket_tree_update(type="tree_update_node_selection", websocket=websocket, tree_data=tree_data)
-            else:
-                print_entire_tree(self.root_node)
+            # await self.websocket_node_selection(selected_node, websocket=websocket)
+            # tree_data = self._get_tree_data()
+            # if websocket:
+            #     await self.websocket_tree_update(type="tree_update_node_selection", websocket=websocket, tree_data=tree_data)
+            # else:
+            #     print_entire_tree(self.root_node)
 
             if selected_node is None:
                 logger.warning("All paths lead to terminal nodes. Ending search.")
@@ -345,17 +346,17 @@ async def mcts_search(self, websocket=None) -> Optional[LATSNode]:
                 print(f"  Visits: {node.visits}")
                 print(f"  Value: {old_value:.3f} -> {node.value:.3f}")
                 # add websocket information, just use websocket here
-                if websocket:
-                    await websocket.send_json({
-                        "type": "backpropagation",
-                        "node_id": id(node),
-                        "node_parent_id": id(node.parent),
-                        "node_action": node.action,
-                        "node_value": node.value,
-                        "node_visits": node.visits,
-                        "node_old_value": old_value,
-                        "node_description": node.natural_language_description,
-                    })
+                # if websocket:
+                #     await websocket.send_json({
+                #         "type": "backpropagation",
+                #         "node_id": id(node),
+                #         "node_parent_id": id(node.parent),
+                #         "node_action": node.action,
+                #         "node_value": node.value,
+                #         "node_visits": node.visits,
+                #         "node_old_value": old_value,
+                #         "node_description": node.natural_language_description,
+                #     })
 
             tree_data = self._get_tree_data()
             print_entire_tree(self.root_node)