Update sample_data.py

sample_data.py

@@ -377,44 +377,282 @@ PROCESS_FLOW_JSON = """
 # New JSON for Work Breakdown Structure (WBS) Diagram - similar to image, but not identical
 WBS_DIAGRAM_JSON = """
 {
-  "project_title": "AI Project Planning",
-  "phases": [
-    {
-      "id": "phase_prep",
-      "label": "Preparation",
-      "tasks": [
-        {"id": "task_id_vision", "label": "Identify Vision"},
-        {"id": "task_design_staff", "label": "Design & Staffing"},
-        {"id": "task_other_prep_1", "label": "<Other Task>"},
-        {"id": "task_other_prep_2", "label": "<Other Task>"},
-        {"id": "task_other_prep_3", "label": "<Other Task>"},
-        {"id": "task_other_prep_4", "label": "<Other Task>"}
-      ]
-    },
-    {
-      "id": "phase_plan",
-      "label": "Planning",
-      "tasks": [
-        {"id": "task_cost_analysis", "label": "Cost Analysis"},
-        {"id": "task_benefit_analysis", "label": "Benefit Analysis"},
-        {"id": "task_risk_assess", "label": "Risk Assessment"},
-        {"id": "task_other_plan_1", "label": "<Other Task>"},
-        {"id": "task_other_plan_2", "label": "<Other Task>"},
-        {"id": "task_other_plan_3", "label": "<Other Task>"}
-      ]
-    },
-    {
-      "id": "phase_dev",
-      "label": "Development",
-      "tasks": [
-        {"id": "task_change_mgmt", "label": "Change Management"},
-        {"id": "task_implementation", "label": "Implementation"},
-        {"id": "task_beta_testing", "label": "Beta Testing"},
-        {"id": "task_other_dev_1", "label": "<Other Task>"},
-        {"id": "task_other_dev_2", "label": "<Other Task>"},
-        {"id": "task_other_dev_3", "label": "<Other Task>"}
-      ]
-    }
-  ]
+  "project_title": "AI Development Project",
+  "root": {
+    "id": "level_1",
+    "label": "<Project Title>",
+    "children": [
+      {
+        "id": "level_2_prep",
+        "label": "Preparation",
+        "children": [
+          {
+            "id": "level_3_vision",
+            "label": "Identify Vision",
+            "children": [
+              {
+                "id": "level_4_analysis",
+                "label": "Stakeholder Analysis",
+                "children": [
+                  {
+                    "id": "level_5_requirements",
+                    "label": "Requirements Gathering",
+                    "children": [
+                      {
+                        "id": "level_6_validation",
+                        "label": "Validation Process",
+                        "children": [
+                          {
+                            "id": "level_7_final",
+                            "label": "Final Documentation"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          },
+          {
+            "id": "level_3_design",
+            "label": "Design and Staffing",
+            "children": [
+              {
+                "id": "level_4_architecture",
+                "label": "AI Architecture",
+                "children": [
+                  {
+                    "id": "level_5_model_selection",
+                    "label": "Model Selection",
+                    "children": [
+                      {
+                        "id": "level_6_benchmarking",
+                        "label": "Performance Benchmarking",
+                        "children": [
+                          {
+                            "id": "level_7_optimization",
+                            "label": "Model Optimization"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          },
+          {
+            "id": "level_3_other_prep",
+            "label": "<Other Task>",
+            "children": [
+              {
+                "id": "level_4_data_prep",
+                "label": "Data Preparation",
+                "children": [
+                  {
+                    "id": "level_5_data_collection",
+                    "label": "Data Collection",
+                    "children": [
+                      {
+                        "id": "level_6_data_cleaning",
+                        "label": "Data Cleaning",
+                        "children": [
+                          {
+                            "id": "level_7_data_validation",
+                            "label": "Data Validation"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          }
+        ]
+      },
+      {
+        "id": "level_2_planning",
+        "label": "Planning",
+        "children": [
+          {
+            "id": "level_3_cost",
+            "label": "Cost Analysis",
+            "children": [
+              {
+                "id": "level_4_budget",
+                "label": "Budget Planning",
+                "children": [
+                  {
+                    "id": "level_5_resource_alloc",
+                    "label": "Resource Allocation",
+                    "children": [
+                      {
+                        "id": "level_6_cost_tracking",
+                        "label": "Cost Tracking",
+                        "children": [
+                          {
+                            "id": "level_7_budget_review",
+                            "label": "Budget Review"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          },
+          {
+            "id": "level_3_benefit",
+            "label": "Benefit Analysis",
+            "children": [
+              {
+                "id": "level_4_roi",
+                "label": "ROI Calculation",
+                "children": [
+                  {
+                    "id": "level_5_metrics",
+                    "label": "Success Metrics",
+                    "children": [
+                      {
+                        "id": "level_6_kpi",
+                        "label": "KPI Definition",
+                        "children": [
+                          {
+                            "id": "level_7_measurement",
+                            "label": "Measurement Plan"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          },
+          {
+            "id": "level_3_risk",
+            "label": "Risk Assessment",
+            "children": [
+              {
+                "id": "level_4_risk_id",
+                "label": "Risk Identification",
+                "children": [
+                  {
+                    "id": "level_5_mitigation",
+                    "label": "Mitigation Strategies",
+                    "children": [
+                      {
+                        "id": "level_6_contingency",
+                        "label": "Contingency Planning",
+                        "children": [
+                          {
+                            "id": "level_7_risk_monitoring",
+                            "label": "Risk Monitoring"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          }
+        ]
+      },
+      {
+        "id": "level_2_development",
+        "label": "Development",
+        "children": [
+          {
+            "id": "level_3_change",
+            "label": "Change Management",
+            "children": [
+              {
+                "id": "level_4_process",
+                "label": "Process Definition",
+                "children": [
+                  {
+                    "id": "level_5_workflow",
+                    "label": "Workflow Design",
+                    "children": [
+                      {
+                        "id": "level_6_automation",
+                        "label": "Process Automation",
+                        "children": [
+                          {
+                            "id": "level_7_monitoring",
+                            "label": "Process Monitoring"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          },
+          {
+            "id": "level_3_implementation",
+            "label": "Implementation",
+            "children": [
+              {
+                "id": "level_4_deployment",
+                "label": "Model Deployment",
+                "children": [
+                  {
+                    "id": "level_5_integration",
+                    "label": "System Integration",
+                    "children": [
+                      {
+                        "id": "level_6_api",
+                        "label": "API Development",
+                        "children": [
+                          {
+                            "id": "level_7_documentation",
+                            "label": "API Documentation"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          },
+          {
+            "id": "level_3_testing",
+            "label": "Beta Testing",
+            "children": [
+              {
+                "id": "level_4_validation",
+                "label": "Model Validation",
+                "children": [
+                  {
+                    "id": "level_5_performance",
+                    "label": "Performance Testing",
+                    "children": [
+                      {
+                        "id": "level_6_user_testing",
+                        "label": "User Acceptance Testing",
+                        "children": [
+                          {
+                            "id": "level_7_release",
+                            "label": "Release Preparation"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            ]
+          }
+        ]
+      }
+    ]
+  }
 }
 """