Update process_flow_generator.py
Browse files
process_flow_generator.py
CHANGED
@@ -7,15 +7,13 @@ import os
|
|
7 |
# as process flow diagrams have a different structure and node addition logic.
|
8 |
# However, we'll re-implement the color gradient logic directly for consistency.
|
9 |
|
10 |
-
def generate_process_flow_diagram(json_input: str
|
11 |
"""
|
12 |
Generates a Process Flow Diagram (Flowchart) from JSON input.
|
13 |
|
14 |
Args:
|
15 |
json_input (str): A JSON string describing the process flow structure.
|
16 |
It must follow the Expected JSON Format Example below.
|
17 |
-
base_color (str): The hexadecimal color string (e.g., '#19191a') for the base
|
18 |
-
color of the nodes, from which a gradient will be generated.
|
19 |
|
20 |
Returns:
|
21 |
str: The filepath to the generated PNG image file.
|
@@ -78,9 +76,7 @@ def generate_process_flow_diagram(json_input: str, base_color: str) -> str:
|
|
78 |
}
|
79 |
)
|
80 |
|
81 |
-
|
82 |
-
if not isinstance(base_color, str) or not base_color.startswith('#') or len(base_color) != 7:
|
83 |
-
base_color = '#19191a' # Fallback to default dark if invalid
|
84 |
|
85 |
# Helper for color and font based on depth (simulated for flowcharts if needed, or consistent color)
|
86 |
# For flowcharts, often all process/decision nodes are the same color.
|
|
|
7 |
# as process flow diagrams have a different structure and node addition logic.
|
8 |
# However, we'll re-implement the color gradient logic directly for consistency.
|
9 |
|
10 |
+
def generate_process_flow_diagram(json_input: str) -> str: # Removed base_color parameter
|
11 |
"""
|
12 |
Generates a Process Flow Diagram (Flowchart) from JSON input.
|
13 |
|
14 |
Args:
|
15 |
json_input (str): A JSON string describing the process flow structure.
|
16 |
It must follow the Expected JSON Format Example below.
|
|
|
|
|
17 |
|
18 |
Returns:
|
19 |
str: The filepath to the generated PNG image file.
|
|
|
76 |
}
|
77 |
)
|
78 |
|
79 |
+
base_color = '#19191a' # Hardcoded base color
|
|
|
|
|
80 |
|
81 |
# Helper for color and font based on depth (simulated for flowcharts if needed, or consistent color)
|
82 |
# For flowcharts, often all process/decision nodes are the same color.
|