Update process_flow_generator.py

process_flow_generator.py

@@ -90,11 +90,9 @@ def generate_process_flow_diagram(json_input: str, output_format: str) -> str:
             
         data = json.loads(json_input)
         
-        # Validate required top-level keys for a flowchart
         if 'start_node' not in data or 'nodes' not in data or 'connections' not in data:
             raise ValueError("Missing required fields: 'start_node', 'nodes', or 'connections'")
 
-        # Define specific node shapes for flowchart types
         node_shapes = {
             "process": "box",          # Rectangle for processes
             "decision": "diamond",     # Diamond for decisions
@@ -118,15 +116,13 @@ def generate_process_flow_diagram(json_input: str, output_format: str) -> str:
             }
         )
         
-        base_color = '#19191a' # Hardcoded base color
+        base_color = '#19191a' 
 
         fill_color_for_nodes = base_color
         font_color_for_nodes = 'white' if base_color == '#19191a' or base_color.lower() in ['#000000', '#19191a'] else 'black'
         
-        # Store all nodes by ID for easy lookup
         all_defined_nodes = {node['id']: node for node in data['nodes']}
         
-        # Add start node explicitly
         start_node_id = data['start_node']
         dot.node(
             start_node_id,
@@ -138,7 +134,6 @@ def generate_process_flow_diagram(json_input: str, output_format: str) -> str:
             fontsize='14'
         )
 
-        # Add all other nodes (process, decision, etc.)
         for node_id, node_info in all_defined_nodes.items():
             if node_id == start_node_id: # Skip if it's the start node, already added
                 continue