Update wbs_diagram_generator.py

wbs_diagram_generator.py

@@ -2,17 +2,15 @@ import graphviz
 import json
 from tempfile import NamedTemporaryFile
 import os
-from graph_generator_utils import add_nodes_and_edges # Reusing common utility
+from graph_generator_utils import add_nodes_and_edges
 
-def generate_wbs_diagram(json_input: str, base_color: str) -> str: # base_color is now correctly used
+def generate_wbs_diagram(json_input: str) -> str: # Removed base_color parameter
     """
     Generates a Work Breakdown Structure (WBS) Diagram from JSON input.
 
     Args:
         json_input (str): A JSON string describing the WBS structure.
                           It must follow the Expected JSON Format Example below.
-        base_color (str): The hexadecimal color string (e.g., '#19191a') for the base
-                          color of the nodes, from which a gradient will be generated.
 
     Returns:
         str: The filepath to the generated PNG image file.
@@ -63,7 +61,7 @@ def generate_wbs_diagram(json_input: str, base_color: str) -> str: # base_color
             }
         )
         
-        # This line was REMOVED to ensure the passed base_color is used: base_color = '#19191a' 
+        base_color = '#19191a' # Hardcoded base color
 
         # Project Title node (main node)
         dot.node(
@@ -77,7 +75,7 @@ def generate_wbs_diagram(json_input: str, base_color: str) -> str: # base_color
         )
 
         # Logic for color gradient within WBS specific nodes (Phases and Tasks)
-        # This ensures the gradient works correctly with the passed base_color
+        # This ensures the gradient works correctly with the hardcoded base_color
         def get_gradient_color(depth, base_hex_color, lightening_factor=0.12):
             base_r = int(base_hex_color[1:3], 16)
             base_g = int(base_hex_color[3:5], 16)
@@ -155,15 +153,6 @@ def generate_wbs_diagram(json_input: str, base_color: str) -> str: # base_color
             # Use subgraph to enforce vertical alignment for tasks within a phase
             if task_nodes_in_phase: # Only create subgraph if there are tasks
                 dot.subgraph(name=f'cluster_{phase_id}')
-                # You would typically add nodes to the subgraph if you want
-                # specific layout within it. For WBS columnar, just using rank attribute might suffice
-                # or ensuring proper `ranksep` and `nodesep` at graph level.
-                # The crucial part for columnar WBS is often setting nodes to the same rank
-                # or controlling the order. This is implicitly handled by `rankdir=TB`.
-                # The subgraphs themselves mostly define visual grouping (border) in Graphviz.
-                # Let's try to ensure vertical flow is emphasized without complex subgraphs that
-                # might interfere with the main layout engine. The primary connection handles rank.
-
         # Save to temporary file
         with NamedTemporaryFile(delete=False, suffix='.png') as tmp:
             dot.render(tmp.name, format='png', cleanup=True)