File size: 3,923 Bytes
cf182ce 4dfdfd3 cf182ce d418d0d 5b8a84b d418d0d 5b8a84b d418d0d 5b8a84b d418d0d 5b8a84b d418d0d 5b8a84b d418d0d 5b8a84b d418d0d 5b8a84b d418d0d cf182ce 0ad9f40 cf182ce 4dfdfd3 cf182ce |
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 |
import graphviz
import json
from tempfile import NamedTemporaryFile
import os
from graph_generator_utils import add_nodes_and_edges
def generate_synoptic_chart(json_input: str, base_color: str) -> str: # base_color is now correctly used
"""
Generates a synoptic chart (horizontal flowchart) from JSON input.
Args:
json_input (str): A JSON string describing the synoptic chart 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.
Expected JSON Format Example:
{
"central_node": "AI Project Lifecycle",
"nodes": [
{
"id": "phase1",
"label": "I. Problem Definition & Data Acquisition",
"relationship": "Starts\\nwith",
"subnodes": [
{
"id": "sub1_1",
"label": "1. Problem Formulation",
"relationship": "Involves\\n(Steps)",
"subnodes": [
{"id": "sub1_1_1", "label": "1.1. Identify Business Need", "relationship": "e.g.\\n(Need)"},
{"id": "sub1_1_2", "label": "1.2. Define KPIs", "relationship": "e.g.\\n(Metrics)"}
]
}
]
},
{
"id": "phase2",
"label": "II. Model Development",
"relationship": "Proceeds\\nto",
"subnodes": [
{
"id": "sub2_1",
"label": "1. Feature Engineering",
"relationship": "Comprises\\n(Features)",
"subnodes": [
{"id": "sub2_1_1", "label": "1.1. Feature Selection", "relationship": "e.g.\\n(Select)"}
]
}
]
}
]
}
"""
try:
if not json_input.strip():
return "Error: Empty input"
data = json.loads(json_input)
if 'central_node' not in data or 'nodes' not in data:
raise ValueError("Missing required fields: central_node or nodes")
dot = graphviz.Digraph(
name='SynopticChart',
format='png',
graph_attr={
'rankdir': 'LR', # Left-to-Right layout (horizontal hierarchy)
'splines': 'ortho', # Straight lines
'bgcolor': 'white', # White background
'pad': '0.5', # Padding around the graph
'ranksep': '0.7', # Reduced horizontal separation between ranks (columns)
'nodesep': '0.3' # Adjusted vertical separation between nodes in the same rank
}
)
# This line was REMOVED to ensure the passed base_color is used: base_color = '#19191a'
# Central node styling (rounded box, dark color)
dot.node(
'central',
data['central_node'],
shape='box', # Rectangular shape
style='filled,rounded', # Filled and rounded corners
fillcolor=base_color, # Darkest color
fontcolor='white', # White text for dark background
fontsize='16' # Larger font for central node
)
# Add child nodes and edges recursively starting from depth 1
add_nodes_and_edges(dot, 'central', data.get('nodes', []), current_depth=1, base_color=base_color)
# 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)}"
|