File size: 7,159 Bytes
11561b8 00a1c64 11561b8 00a1c64 11561b8 00a1c64 11561b8 865a197 a495f34 11561b8 00a1c64 11561b8 00a1c64 a495f34 00a1c64 a495f34 00a1c64 11561b8 00a1c64 11561b8 00a1c64 11561b8 00a1c64 11561b8 00a1c64 11561b8 00a1c64 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 167 168 169 170 171 172 173 174 175 |
import graphviz
import json
from tempfile import NamedTemporaryFile
import os
from graph_generator_utils import add_nodes_and_edges # Keeping this import for consistency, though WBS will use its own recursive logic.
def generate_wbs_diagram(json_input: str) -> str:
"""
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",
"subtasks": [
{"id": "subtask_1_1_1", "label": "1.1.1. Problem Definition",
"sub_subtasks": [
{"id": "ss_task_1_1_1_1", "label": "1.1.1.1. Req. Analysis",
"sub_sub_subtasks": [
{"id": "sss_task_1_1_1_1_1", "label": "1.1.1.1.1. User Stories",
"final_level_tasks": [
{"id": "ft_1_1_1_1_1_1", "label": "1.1.1.1.1.1. Interview Users"}
]
}
]
}
]
}
]
}
]
}
]
}
"""
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,
fontcolor='white',
fontsize='18'
)
# Helper for color and font based on depth for WBS
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):
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
luminance = (0.2126 * current_r + 0.7152 * current_g + 0.0722 * current_b) / 255
return 'white' if luminance < 0.5 else 'black'
def _add_wbs_nodes_recursive(parent_id, current_level_tasks, current_depth):
for task_data in current_level_tasks:
task_id = task_data.get('id')
task_label = task_data.get('label')
if not all([task_id, task_label]):
raise ValueError(f"Invalid task data at depth {current_depth}: {task_data}")
node_fill_color = get_gradient_color(current_depth, base_color)
node_font_color = get_font_color_for_background(current_depth, base_color)
font_size = max(9, 14 - (current_depth * 2))
dot.node(
task_id,
task_label,
shape='box',
style='filled,rounded',
fillcolor=node_fill_color,
fontcolor=node_font_color,
fontsize=str(font_size)
)
dot.edge(parent_id, task_id, color='#4a4a4a', arrowhead='none')
# Recursively call for next level of tasks (subtasks, sub_subtasks, etc.)
# This handles arbitrary nested keys like 'subtasks', 'sub_subtasks', 'final_level_tasks'
next_level_keys = ['tasks', 'subtasks', 'sub_subtasks', 'sub_sub_subtasks', 'final_level_tasks']
for key_idx, key in enumerate(next_level_keys):
if key in task_data and isinstance(task_data[key], list):
_add_wbs_nodes_recursive(task_id, task_data[key], current_depth + 1)
break # Only process the first found sub-level key
# Process phases (level 1 from project_root)
phase_depth = 1
for phase in data['phases']:
phase_id = phase.get('id')
phase_label = phase.get('label')
if not all([phase_id, phase_label]):
raise ValueError(f"Invalid phase data: {phase}")
phase_fill_color = get_gradient_color(phase_depth, base_color)
phase_font_color = get_font_color_for_background(phase_depth, base_color)
font_size_phase = max(9, 14 - (phase_depth * 2))
dot.node(
phase_id,
phase_label,
shape='box',
style='filled,rounded',
fillcolor=phase_fill_color,
fontcolor=phase_font_color,
fontsize=str(font_size_phase)
)
dot.edge('project_root', phase_id, color='#4a4a4a', arrowhead='none')
# Start recursion for tasks under this phase
if 'tasks' in phase and isinstance(phase['tasks'], list):
_add_wbs_nodes_recursive(phase_id, phase['tasks'], phase_depth + 1)
# 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)}"
|