File size: 6,529 Bytes
11561b8 865a197 11561b8 865a197 11561b8 865a197 a495f34 11561b8 a495f34 865a197 a495f34 11561b8 a495f34 11561b8 a495f34 11561b8 a495f34 11561b8 a495f34 11561b8 a495f34 11561b8 a495f34 11561b8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 |
import graphviz
import json
from tempfile import NamedTemporaryFile
import os
from graph_generator_utils import add_nodes_and_edges
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.
Returns:
str: The filepath to the generated PNG image file.
Expected JSON Format Example:
{
"project_title": "Software Development Project",
"phases": [
{
"id": "phase_prep",
"label": "1. Preparation",
"tasks": [
{"id": "task_vision", "label": "1.1. Identify Vision"},
{"id": "task_design", "label": "1.2. Design & Staffing"}
]
},
{
"id": "phase_plan",
"label": "2. Planning",
"tasks": [
{"id": "task_cost", "label": "2.1. Cost Analysis"},
{"id": "task_benefit", "label": "2.2. Benefit Analysis"},
{"id": "task_risk", "label": "2.3. Risk Assessment"}
]
}
]
}
"""
try:
if not json_input.strip():
return "Error: Empty input"
data = json.loads(json_input)
if 'project_title' not in data or 'phases' not in data:
raise ValueError("Missing required fields: project_title or phases")
dot = graphviz.Digraph(
name='WBSDiagram',
format='png',
graph_attr={
'rankdir': 'TB', # Top-to-Bottom hierarchy
'splines': 'ortho', # Straight lines
'bgcolor': 'white', # White background
'pad': '0.5', # Padding
'ranksep': '0.8', # Adjust vertical separation between ranks
'nodesep': '0.5' # Adjust horizontal separation between nodes
}
)
base_color = '#19191a' # Hardcoded base color
# Project Title node (main node)
dot.node(
'project_root',
data['project_title'],
shape='box',
style='filled,rounded',
fillcolor=base_color, # Use the selected base color
fontcolor='white',
fontsize='18'
)
# Logic for color gradient within WBS specific nodes (Phases and Tasks)
# 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)
base_b = int(base_hex_color[5:7], 16)
current_r = base_r + int((255 - base_r) * depth * lightening_factor)
current_g = base_g + int((255 - base_g) * depth * lightening_factor)
current_b = base_b + int((255 - base_b) * depth * lightening_factor)
return f'#{min(255, current_r):02x}{min(255, current_g):02x}{min(255, current_b):02x}'
def get_font_color_for_background(depth, base_hex_color, lightening_factor=0.12):
# Calculate brightness/lightness of the node color at this depth
# and return black/white for text accordingly
base_r = int(base_hex_color[1:3], 16)
base_g = int(base_hex_color[3:5], 16)
base_b = int(base_hex_color[5:7], 16)
current_r = base_r + (255 - base_r) * depth * lightening_factor
current_g = base_g + (255 - base_g) * depth * lightening_factor
current_b = base_b + (255 - base_b) * depth * lightening_factor
# Simple luminance check (ITU-R BT.709 coefficients)
luminance = (0.2126 * current_r + 0.7152 * current_g + 0.0722 * current_b) / 255
return 'white' if luminance < 0.5 else 'black'
current_depth = 1 # Depth for phases
for phase in data['phases']:
phase_id = phase.get('id')
phase_label = phase.get('label')
tasks = phase.get('tasks', [])
if not all([phase_id, phase_label]):
raise ValueError(f"Invalid phase: {phase}")
phase_fill_color = get_gradient_color(current_depth, base_color)
phase_font_color = get_font_color_for_background(current_depth, base_color)
dot.node(
phase_id,
phase_label,
shape='box',
style='filled,rounded',
fillcolor=phase_fill_color,
fontcolor=phase_font_color,
fontsize='14'
)
dot.edge('project_root', phase_id, color='#4a4a4a', arrowhead='none') # Connect to root
task_depth = current_depth + 1 # Depth for tasks
task_font_size = max(9, 14 - (task_depth * 2))
task_nodes_in_phase = []
for task in tasks:
task_id = task.get('id')
task_label = task.get('label')
if not all([task_id, task_label]):
raise ValueError(f"Invalid task: {task}")
task_fill_color = get_gradient_color(task_depth, base_color)
task_font_color = get_font_color_for_background(task_depth, base_color)
dot.node(
task_id,
task_label,
shape='box',
style='filled,rounded',
fillcolor=task_fill_color,
fontcolor=task_font_color,
fontsize=str(task_font_size)
)
dot.edge(phase_id, task_id, color='#4a4a4a', arrowhead='none') # Connect task to phase
task_nodes_in_phase.append(task_id)
# 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}')
# Save to temporary file
with NamedTemporaryFile(delete=False, suffix='.png') as tmp:
dot.render(tmp.name, format='png', cleanup=True)
return tmp.name + '.png'
except json.JSONDecodeError:
return "Error: Invalid JSON format"
except Exception as e:
return f"Error: {str(e)}"
|