Update binary_tree_generator.py

binary_tree_generator.py

@@ -1,4 +1,81 @@
-import graphviz
+def add_binary_tree_nodes(node, current_depth=0):
+            """
+            Add binary tree nodes recursively with proper styling.
+            """
+            if not node:
+                return
+            
+            node_id = node.get('id', f'node_{current_depth}')
+            node_label = node.get('label', 'Node')
+            
+            # Calculate color for current depth using the 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
+            else:
+                base_color_safe = base_color
+                
+            base_r = int(base_color_safe[1:3], 16)
+            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))
+            
+            # Add the current node
+            dot.node(
+                node_id,
+                node_label,
+                shape='circle',
+                style='filled',
+                fillcolor=node_color,
+                fontcolor=font_color,
+                fontsize=str(font_size),
+                width='0.8',
+                height='0.8'
+            )
+            
+            # Edge color
+            edge_color = '#4a4a4a'  # Dark gray for lines
+            
+            # Process left child
+            left_child = node.get('left')
+            if left_child:
+                add_binary_tree_nodes(left_child, current_depth + 1)
+                left_id = left_child.get('id', f'node_{current_depth + 1}_left')
+                dot.edge(
+                    node_id,
+                    left_id,
+                    color=edge_color,
+                    arrowsize='0.8'
+                )
+            
+            # Process right child
+            right_child = node.get('right')
+            if right_child:
+                add_binary_tree_nodes(right_child, current_depth + 1)
+                right_id = right_child.get('id', f'node_{current_depth + 1}_right')
+                dot.edge(
+                    node_id,
+                    right_id,
+                    color=edge_import graphviz
 import json
 from tempfile import NamedTemporaryFile
 import os
@@ -80,20 +157,36 @@ 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 opacity based on depth
-            max_depth = 5  # Assume maximum depth for color calculation
-            if current_depth >= max_depth:
-                opacity_value = 128  # Minimum opacity (50%)
+            # 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
             else:
-                opacity_value = int(255 * (1.0 - (current_depth * 0.6 / max_depth)))
+                base_color_safe = base_color
+                
+            base_r = int(base_color_safe[1:3], 16)
+            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}'
             
-            opacity = format(opacity_value, '02x')
-            node_color = f"{base_color}{opacity}"
+            # 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'
             
-            # Determine font color based on background brightness
-            # If opacity_value is high (dark background), use white text
-            # If opacity_value is low (light background), use black text
-            font_color = 'white' if opacity_value > 180 else 'black'
+            # Font size adjusts based on depth, ensuring a minimum size
+            font_size = max(9, 14 - (current_depth * 2))
             
             # Add the current node
             dot.node(
@@ -103,7 +196,7 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
                 style='filled',
                 fillcolor=node_color,
                 fontcolor=font_color,
-                fontsize='14',
+                fontsize=str(font_size),
                 width='0.8',
                 height='0.8'
             )
@@ -116,7 +209,7 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
                 dot.edge(
                     node_id,
                     left_id,
-                    color='#666666',
+                    color='#4a4a4a',  # Dark gray for lines
                     arrowsize='0.8'
                 )
             
@@ -128,7 +221,7 @@ def generate_binary_tree_diagram(json_input: str, output_format: str) -> str:
                 dot.edge(
                     node_id,
                     right_id,
-                    color='#666666',
+                    color='#4a4a4a',  # Dark gray for lines
                     arrowsize='0.8'
                 )