Update synoptic_chart_generator.py

synoptic_chart_generator.py

@@ -12,9 +12,6 @@ def generate_synoptic_chart(json_input: str) -> str: # Removed base_color parame
         json_input (str): A JSON string describing the synoptic chart structure.
                           It must follow the Expected JSON Format Example below.
 
-    Returns:
-        str: The filepath to the generated PNG image file.
-
     Expected JSON Format Example:
     {
       "central_node": "AI Project Lifecycle",
@@ -22,15 +19,24 @@ def generate_synoptic_chart(json_input: str) -> str: # Removed base_color parame
         {
           "id": "phase1",
           "label": "I. Problem Definition & Data Acquisition",
-          "relationship": "Starts\\nwith",
+          "relationship": "Starts with",
           "subnodes": [
             {
               "id": "sub1_1",
               "label": "1. Problem Formulation",
-              "relationship": "Involves\\n(Steps)",
+              "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",
               "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": "sub1_2_1", "label": "2.1. Source Data", "relationship": "from"},
+                {"id": "sub1_2_2", "label": "2.2. Data Cleaning", "relationship": "includes"}
               ]
             }
           ]
@@ -38,20 +44,58 @@ def generate_synoptic_chart(json_input: str) -> str: # Removed base_color parame
         {
           "id": "phase2",
           "label": "II. Model Development",
-          "relationship": "Proceeds\\nto",
+          "relationship": "Proceeds to",
           "subnodes": [
             {
               "id": "sub2_1",
               "label": "1. Feature Engineering",
-              "relationship": "Comprises\\n(Features)",
+              "relationship": "Comprises",
               "subnodes": [
-                {"id": "sub2_1_1", "label": "1.1. Feature Selection", "relationship": "e.g.\\n(Select)"}
+                {"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",
+              "subnodes": [
+                {"id": "sub3_2_1", "label": "2.1. API/Integration Development", "relationship": "for"},
+                {"id": "sub3_2_2", "label": "2.2. Continuous Monitoring", "relationship": "ensures"}
               ]
             }
           ]
         }
       ]
     }
+
+    Returns:
+        str: The filepath to the generated PNG image file.
     """
     try:
         if not json_input.strip():