Update process_flow_generator.py

process_flow_generator.py

@@ -4,7 +4,7 @@ from tempfile import NamedTemporaryFile
 import os
 from graph_generator_utils import add_nodes_and_edges # Reusing common utility
 
-def generate_process_flow_diagram(json_input: str, base_color: str) -> str:
+def generate_process_flow_diagram(json_input: str, base_color: str) -> str: # base_color is now correctly used
     """
     Generates a Process Flow Diagram (Flowchart) from JSON input.
 
@@ -52,7 +52,7 @@ def generate_process_flow_diagram(json_input: str, base_color: str) -> str:
         }
       ],
       "connections": [
-        {"from": "start_node", "to": "step1", "label": "Start"},
+        {"from": "start_node", "to": "step1", "label": "Initiate"},
         {"from": "step1", "to": "decision1", "label": "Continue"},
         {"from": "decision1", "to": "path_yes", "label": "Yes"},
         {"from": "decision1", "to": "path_no", "label": "No"},
@@ -67,16 +67,8 @@ def generate_process_flow_diagram(json_input: str, base_color: str) -> str:
             
         data = json.loads(json_input)
         
-        # Determine specific node shapes for flowchart types
-        node_shapes = {
-            "process": "box",          # Rectangle for processes
-            "decision": "diamond",     # Diamond for decisions
-            "start": "oval",           # Oval for start
-            "end": "oval",             # Oval for end
-            "io": "parallelogram",     # Input/Output
-            "document": "note",        # Document symbol
-            "default": "box"           # Fallback
-        }
+        if 'central_node' not in data or 'nodes' not in data:
+            raise ValueError("Missing required fields: central_node or nodes")
 
         dot = graphviz.Digraph(
             name='ProcessFlowDiagram',
@@ -91,9 +83,18 @@ def generate_process_flow_diagram(json_input: str, base_color: str) -> str:
             }
         )
         
-        # Ensure base_color is valid, fallback if not
-        if not isinstance(base_color, str) or not base_color.startswith('#') or len(base_color) != 7:
-            base_color = '#19191a' # Fallback to default dark if invalid
+        # This line was REMOVED to ensure the passed base_color is used: base_color = '#19191a' 
+
+        # Determine specific node shapes for flowchart types
+        node_shapes = {
+            "process": "box",          # Rectangle for processes
+            "decision": "diamond",     # Diamond for decisions
+            "start": "oval",           # Oval for start
+            "end": "oval",             # Oval for end
+            "io": "parallelogram",     # Input/Output
+            "document": "note",        # Document symbol
+            "default": "box"           # Fallback
+        }
 
         # Add all nodes based on JSON structure
         all_nodes_data = {}
@@ -109,7 +110,6 @@ def generate_process_flow_diagram(json_input: str, base_color: str) -> str:
             shape = node_shapes.get(node_type, "box") # Default to box if type is unknown
 
             # Use base_color for all nodes for consistent flowchart look
-            # You can adapt add_nodes_and_edges if you want color gradient per depth for flowcharts too
             fill_color_for_node = base_color
             font_color_for_node = 'white' if base_color == '#19191a' else 'black' # Keep text readable