Update sample_data.py

sample_data.py

@@ -377,114 +377,116 @@ PROCESS_FLOW_JSON = """
 # New JSON for Work Breakdown Structure (WBS) Diagram - similar to image, but not identical
 WBS_DIAGRAM_JSON = """
 {
-  "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_design",
-                "label": "Design and Staffing",
-                "children": [
-                  {
-                    "id": "level_5_task1",
-                    "label": "<Other Task>",
-                    "children": [
-                      {
-                        "id": "level_6_task1",
-                        "label": "<Other Task>",
-                        "children": [
-                          {
-                            "id": "level_7_task1",
-                            "label": "<Other Task>"
-                          }
-                        ]
-                      }
-                    ]
-                  }
-                ]
-              }
-            ]
-          }
-        ]
-      },
-      {
-        "id": "level_2_planning",
-        "label": "Planning",
-        "children": [
-          {
-            "id": "level_3_cost",
-            "label": "Cost Analysis",
-            "children": [
-              {
-                "id": "level_4_benefit",
-                "label": "Benefit Analysis",
-                "children": [
-                  {
-                    "id": "level_5_risk",
-                    "label": "Risk Assessment",
-                    "children": [
-                      {
-                        "id": "level_6_task2",
-                        "label": "<Other Task>",
-                        "children": [
-                          {
-                            "id": "level_7_task2",
-                            "label": "<Other Task>"
-                          }
-                        ]
-                      }
-                    ]
-                  }
-                ]
-              }
-            ]
-          }
-        ]
-      },
-      {
-        "id": "level_2_development",
-        "label": "Development",
-        "children": [
-          {
-            "id": "level_3_change",
-            "label": "Change Management",
-            "children": [
-              {
-                "id": "level_4_implementation",
-                "label": "Implementation",
-                "children": [
-                  {
-                    "id": "level_5_testing",
-                    "label": "Beta Testing",
-                    "children": [
-                      {
-                        "id": "level_6_task3",
-                        "label": "<Other Task>",
-                        "children": [
-                          {
-                            "id": "level_7_task3",
-                            "label": "<Other Task>"
-                          }
-                        ]
-                      }
-                    ]
-                  }
-                ]
-              }
-            ]
-          }
-        ]
-      }
-    ]
-  }
+  "project_title": "AI Model Development Project",
+  "phases": [
+    {
+      "id": "phase_prep",
+      "label": "Preparation",
+      "tasks": [
+        {
+          "id": "task_1_1_vision",
+          "label": "Identify Vision",
+          "subtasks": [
+            {
+              "id": "subtask_1_1_1_design_staff",
+              "label": "Design & Staffing",
+              "sub_subtasks": [
+                {
+                  "id": "ss_task_1_1_1_1_env_setup",
+                  "label": "Environment Setup",
+                  "sub_sub_subtasks": [
+                    {
+                      "id": "sss_task_1_1_1_1_1_lib_install",
+                      "label": "Install Libraries",
+                      "final_level_tasks": [
+                        {"id": "ft_1_1_1_1_1_1_data_access", "label": "Grant Data Access"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            }
+          ]
+        },
+        {"id": "task_other_prep_1", "label": "Define Resources"},
+        {"id": "task_other_prep_2", "label": "Initial Data Fetch"},
+        {"id": "task_other_prep_3", "label": "Toolchain Setup"},
+        {"id": "task_other_prep_4", "label": "Kick-off Meeting"}
+      ]
+    },
+    {
+      "id": "phase_plan",
+      "label": "Planning",
+      "tasks": [
+        {
+          "id": "task_2_1_cost_analysis",
+          "label": "Cost Analysis",
+          "subtasks": [
+            {
+              "id": "subtask_2_1_1_benefit_analysis",
+              "label": "Benefit Analysis",
+              "sub_subtasks": [
+                {
+                  "id": "ss_task_2_1_1_1_risk_assess",
+                  "label": "AI Risk Assessment",
+                  "sub_sub_subtasks": [
+                    {
+                      "id": "sss_task_2_1_1_1_1_model_selection",
+                      "label": "Model Selection",
+                      "final_level_tasks": [
+                        {"id": "ft_2_1_1_1_1_1_data_strategy", "label": "Data Strategy"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            }
+          ]
+        },
+        {"id": "task_other_plan_1", "label": "Evaluation Plan"},
+        {"id": "task_other_plan_2", "label": "Deployment Plan"},
+        {"id": "task_other_plan_3", "label": "Compliance Check"},
+        {"id": "task_other_plan_4", "label": "Ethical Review"},
+        {"id": "task_other_plan_5", "label": "Roadmap Finalization"}
+      ]
+    },
+    {
+      "id": "phase_dev",
+      "label": "Development",
+      "tasks": [
+        {
+          "id": "task_3_1_change_mgmt",
+          "label": "Data Preprocessing",
+          "subtasks": [
+            {
+              "id": "subtask_3_1_1_implementation",
+              "label": "Feature Engineering",
+              "sub_subtasks": [
+                {
+                  "id": "ss_task_3_1_1_1_beta_testing",
+                  "label": "Model Training",
+                  "sub_sub_subtasks": [
+                    {
+                      "id": "sss_task_3_1_1_1_1_other_task",
+                      "label": "Model Evaluation",
+                      "final_level_tasks": [
+                        {"id": "ft_3_1_1_1_1_1_hyperparam_tune", "label": "Hyperparameter Tuning"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            }
+          ]
+        },
+        {"id": "task_other_dev_1", "label": "Version Control"},
+        {"id": "task_other_dev_2", "label": "API Development"},
+        {"id": "task_other_dev_3", "label": "Integration Testing"},
+        {"id": "task_other_dev_4", "label": "Documentation"},
+        {"id": "task_other_dev_5", "label": "User Acceptance Testing"}
+      ]
+    }
+  ]
 }
+
 """