Update synoptic_chart_generator.py

synoptic_chart_generator.py

@@ -4,13 +4,18 @@ from tempfile import NamedTemporaryFile
 import os
 from graph_generator_utils import add_nodes_and_edges
 
-def generate_synoptic_chart(json_input: str) -> str:
+def generate_synoptic_chart(json_input: str, base_color: str) -> str:
     """
     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:
     {
@@ -19,24 +24,15 @@ def generate_synoptic_chart(json_input: str) -> str:
         {
           "id": "phase1",
           "label": "I. Problem Definition & Data Acquisition",
-          "relationship": "Starts with",
+          "relationship": "Starts\\nwith",
           "subnodes": [
             {
               "id": "sub1_1",
               "label": "1. Problem Formulation",
-              "relationship": "Involves",
-              "subnodes": [
-                {"id": "sub1_1_1", "label": "1.1. Identify Business Need", "relationship": "e.g."},
-                {"id": "sub1_1_2", "label": "1.2. Define KPIs", "relationship": "e.g."}
-              ]
-            },
-            {
-              "id": "sub1_2",
-              "label": "2. Data Collection",
-              "relationship": "Followed by",
+              "relationship": "Involves\\n(Steps)",
               "subnodes": [
-                {"id": "sub1_2_1", "label": "2.1. Source Data", "relationship": "from"},
-                {"id": "sub1_2_2", "label": "2.2. Data Cleaning", "relationship": "includes"}
+                {"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)"}
               ]
             }
           ]
@@ -44,58 +40,20 @@ def generate_synoptic_chart(json_input: str) -> str:
         {
           "id": "phase2",
           "label": "II. Model Development",
-          "relationship": "Proceeds to",
+          "relationship": "Proceeds\\nto",
           "subnodes": [
             {
               "id": "sub2_1",
               "label": "1. Feature Engineering",
-              "relationship": "Comprises",
-              "subnodes": [
-                {"id": "sub2_1_1", "label": "1.1. Feature Selection", "relationship": "e.g."},
-                {"id": "sub2_1_2", "label": "1.2. Feature Transformation", "relationship": "e.g."}
-              ]
-            },
-            {
-              "id": "sub2_2",
-              "label": "2. Model Training",
-              "relationship": "Involves",
-              "subnodes": [
-                {"id": "sub2_2_1", "label": "2.1. Algorithm Selection", "relationship": "uses"},
-                {"id": "sub2_2_2", "label": "2.2. Hyperparameter Tuning", "relationship": "optimizes"}
-              ]
-            }
-          ]
-        },
-        {
-          "id": "phase3",
-          "label": "III. Evaluation & Deployment",
-          "relationship": "Culminates in",
-          "subnodes": [
-            {
-              "id": "sub3_1",
-              "label": "1. Model Evaluation",
-              "relationship": "Includes",
-              "subnodes": [
-                {"id": "sub3_1_1", "label": "1.1. Performance Metrics", "relationship": "measures"},
-                {"id": "sub3_1_2", "label": "1.2. Bias & Fairness Audits", "relationship": "ensures"}
-              ]
-            },
-            {
-              "id": "sub3_2",
-              "label": "2. Deployment & Monitoring",
-              "relationship": "Requires",
+              "relationship": "Comprises\\n(Features)",
               "subnodes": [
-                {"id": "sub3_2_1", "label": "2.1. API/Integration Development", "relationship": "for"},
-                {"id": "sub3_2_2", "label": "2.2. Continuous Monitoring", "relationship": "ensures"}
+                {"id": "sub2_1_1", "label": "1.1. Feature Selection", "relationship": "e.g.\\n(Select)"}
               ]
             }
           ]
         }
       ]
     }
-
-    Returns:
-        str: The filepath to the generated PNG image file.
     """
     try:
         if not json_input.strip():
@@ -119,6 +77,10 @@ def generate_synoptic_chart(json_input: 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
+
         base_color = '#19191a' # Base color for the central node and gradient
 
         # Central node styling (rounded box, dark color)