Update binary_tree_generator.py

binary_tree_generator.py

@@ -59,7 +59,7 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
             name='BinaryTree',
             format='png',
             graph_attr={
-                'rankdir': 'TB',        # Top-to-Bottom layout (vertical hierarchy)
+                'rankdir': 'TB',        # Top-to-Bottom layout 
                 'splines': 'line',      # Straight lines
                 'bgcolor': 'white',     # White background
                 'pad': '0.5',           # Padding around the graph
@@ -68,7 +68,7 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
             }
         )
         
-        base_color = '#19191a' # Hardcoded base color
+        base_color = '#19191a' 
         
         def add_binary_tree_nodes(node, current_depth=0):
             """
@@ -80,12 +80,12 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
             node_id = node.get('id', f'node_{current_depth}')
             node_label = node.get('label', 'Node')
             
-            # Calculate color for current depth using same logic as graph_generator_utils
+            
             lightening_factor = 0.12 
             
-            # Convert base_color hex to RGB for interpolation
+            
             if not isinstance(base_color, str) or not base_color.startswith('#') or len(base_color) != 7:
-                base_color_safe = '#19191a'  # Fallback to default dark if invalid
+                base_color_safe = '#19191a'  
             else:
                 base_color_safe = base_color
                 
@@ -93,25 +93,23 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
             base_g = int(base_color_safe[3:5], 16)
             base_b = int(base_color_safe[5:7], 16)
             
-            # Calculate current node color by blending towards white
+            
             current_r = base_r + int((255 - base_r) * current_depth * lightening_factor)
             current_g = base_g + int((255 - base_g) * current_depth * lightening_factor)
             current_b = base_b + int((255 - base_b) * current_depth * lightening_factor)
             
-            # Clamp values to 255 to stay within valid RGB range
+            
             current_r = min(255, current_r)
             current_g = min(255, current_g)
             current_b = min(255, current_b)
             
             node_color = f'#{current_r:02x}{current_g:02x}{current_b:02x}'
             
-            # Font color: white for dark nodes, black for very light nodes for readability
             font_color = 'white' if current_depth * lightening_factor < 0.6 else 'black'
             
-            # Font size adjusts based on depth, ensuring a minimum size
-            font_size = max(9, 14 - (current_depth * 2))
+            font_size = max(9, 14 - current_depth)
+            
             
-            # Add the current node
             dot.node(
                 node_id,
                 node_label,
@@ -124,7 +122,6 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
                 height='0.8'
             )
             
-            # Process left child
             left_child = node.get('left')
             if left_child:
                 add_binary_tree_nodes(left_child, current_depth + 1)
@@ -132,11 +129,10 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
                 dot.edge(
                     node_id,
                     left_id,
-                    color='#4a4a4a',  # Dark gray for lines
+                    color='#4a4a4a',  
                     arrowsize='0.8'
                 )
             
-            # Process right child
             right_child = node.get('right')
             if right_child:
                 add_binary_tree_nodes(right_child, current_depth + 1)
@@ -144,11 +140,10 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
                 dot.edge(
                     node_id,
                     right_id,
-                    color='#4a4a4a',  # Dark gray for lines
+                    color='#4a4a4a',  
                     arrowsize='0.8'
                 )
         
-        # Add binary tree nodes and edges recursively
         add_binary_tree_nodes(data['root'], current_depth=0)
 
         with NamedTemporaryFile(delete=False, suffix=f'.{output_format}') as tmp: