Update sample_data.py

sample_data.py

@@ -377,183 +377,42 @@ PROCESS_FLOW_JSON = """
 # New JSON for Work Breakdown Structure (WBS) Diagram - similar to image, but not identical
 WBS_DIAGRAM_JSON = """
 {
-  "project_title": "AI Model Deployment Project",
+  "project_title": "AI Project Planning",
   "phases": [
     {
-      "id": "phase_1_initiation",
-      "label": "1. Project Initiation",
+      "id": "phase_prep",
+      "label": "Preparation",
       "tasks": [
-        {
-          "id": "task_1_1_goal_scope",
-          "label": "1.1. Define Goal & Scope",
-          "subtasks": [
-            {
-              "id": "subtask_1_1_1_problem_def",
-              "label": "1.1.1. Problem Definition",
-              "sub_subtasks": [
-                {
-                  "id": "ss_task_1_1_1_1_req_analysis",
-                  "label": "1.1.1.1. Requirements Analysis",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_1_1_1_1_user_stories",
-                      "label": "1.1.1.1.1. User Stories",
-                      "final_level_tasks": [
-                        {"id": "ft_1_1_1_1_1_1_interview_users", "label": "1.1.1.1.1.1. Interview Users"},
-                        {"id": "ft_1_1_1_1_1_2_document_needs", "label": "1.1.1.1.1.2. Document Needs"},
-                        {"id": "ft_1_1_1_1_1_3_validate_scope", "label": "1.1.1.1.1.3. Validate Scope"}
-                      ]
-                    }
-                  ]
-                },
-                {
-                  "id": "ss_task_1_1_1_2_data_id",
-                  "label": "1.1.1.2. Data Identification",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_1_1_2_1_source_mapping",
-                      "label": "1.1.1.2.1. Source Mapping",
-                      "final_level_tasks": [
-                        {"id": "ft_1_1_1_2_1_1_list_databases", "label": "1.1.1.2.1.1. List Databases"},
-                        {"id": "ft_1_1_1_2_1_2_api_integration", "label": "1.1.1.2.1.2. API Integration"},
-                        {"id": "ft_1_1_1_2_1_3_external_feeds", "label": "1.1.1.2.1.3. External Feeds"}
-                      ]
-                    }
-                  ]
-                },
-                {
-                  "id": "ss_task_1_1_1_3_ml_approach",
-                  "label": "1.1.1.3. ML Approach Selection",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_1_1_3_1_algo_research",
-                      "label": "1.1.1.3.1. Algorithm Research",
-                      "final_level_tasks": [
-                        {"id": "ft_1_1_1_3_1_1_supervised_review", "label": "1.1.1.3.1.1. Supervised Review"},
-                        {"id": "ft_1_1_1_3_1_2_unsupervised_review", "label": "1.1.1.3.1.2. Unsupervised Review"},
-                        {"id": "ft_1_1_1_3_1_3_rl_consideration", "label": "1.1.1.3.1.3. RL Consideration"}
-                      ]
-                    }
-                  ]
-                }
-              ]
-            }
-          ]
-        },
-        {
-          "id": "task_1_2_resource_plan",
-          "label": "1.2. Resource Planning",
-          "subtasks": [
-            {
-              "id": "subtask_1_2_1_team_assembly",
-              "label": "1.2.1. Team Assembly",
-              "sub_subtasks": [
-                {
-                  "id": "ss_task_1_2_1_1_data_scientists",
-                  "label": "1.2.1.1. Data Scientists",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_2_1_1_1_identify_leads",
-                      "label": "1.2.1.1.1. Identify Leads",
-                      "final_level_tasks": [
-                        {"id": "ft_1_2_1_1_1_1_hire_ml_engineers", "label": "1.2.1.1.1.1. Hire ML Engineers"},
-                        {"id": "ft_1_2_1_1_1_2_allocate_researchers", "label": "1.2.1.1.1.2. Allocate Researchers"},
-                        {"id": "ft_1_2_1_1_1_3_assign_roles", "label": "1.2.1.1.1.3. Assign Roles"}
-                      ]
-                    }
-                  ]
-                },
-                {
-                  "id": "ss_task_1_2_1_2_engineers",
-                  "label": "1.2.1.2. Software Engineers",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_2_1_2_1_backend_devs",
-                      "label": "1.2.1.2.1. Backend Devs",
-                      "final_level_tasks": [
-                        {"id": "ft_1_2_1_2_1_1_recruit_py", "label": "1.2.1.2.1.1. Recruit Python"},
-                        {"id": "ft_1_2_1_2_1_2_recruit_java", "label": "1.2.1.2.1.2. Recruit Java"},
-                        {"id": "ft_1_2_1_2_1_3_onboard_team", "label": "1.2.1.2.1.3. Onboard Team"}
-                      ]
-                    }
-                  ]
-                },
-                {
-                  "id": "ss_task_1_2_1_3_domain_experts",
-                  "label": "1.2.1.3. Domain Experts",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_2_1_3_1_consultants",
-                      "label": "1.2.1.3.1. Consultants",
-                      "final_level_tasks": [
-                        {"id": "ft_1_2_1_3_1_1_engage_smes", "label": "1.2.1.3.1.1. Engage SMEs"},
-                        {"id": "ft_1_2_1_3_1_2_define_ksas", "label": "1.2.1.3.1.2. Define KSAs"},
-                        {"id": "ft_1_2_1_3_1_3_establish_collab", "label": "1.2.1.3.1.3. Establish Collab."}
-                      ]
-                    }
-                  ]
-                }
-              ]
-            }
-          ]
-        },
-        {
-          "id": "task_1_3_budget_timeline",
-          "label": "1.3. Budget & Timeline",
-          "subtasks": [
-            {
-              "id": "subtask_1_3_1_cost_estim",
-              "label": "1.3.1. Cost Estimation",
-              "sub_subtasks": [
-                {
-                  "id": "ss_task_1_3_1_1_hardware_costs",
-                  "label": "1.3.1.1. Hardware Costs",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_3_1_1_1_gpu_procurement",
-                      "label": "1.3.1.1.1. GPU Procurement",
-                      "final_level_tasks": [
-                        {"id": "ft_1_3_1_1_1_1_server_purchase", "label": "1.3.1.1.1.1. Server Purchase"},
-                        {"id": "ft_1_3_1_1_1_2_cloud_compute", "label": "1.3.1.1.1.2. Cloud Compute"},
-                        {"id": "ft_1_3_1_1_1_3_network_equip", "label": "1.3.1.1.1.3. Network Equip."}
-                      ]
-                    }
-                  ]
-                },
-                {
-                  "id": "ss_task_1_3_1_2_software_costs",
-                  "label": "1.3.1.2. Software Costs",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_3_1_2_1_licensing",
-                      "label": "1.3.1.2.1. Licensing",
-                      "final_level_tasks": [
-                        {"id": "ft_1_3_1_2_1_1_os_licenses", "label": "1.3.1.2.1.1. OS Licenses"},
-                        {"id": "ft_1_3_1_2_1_2_tool_subscriptions", "label": "1.3.1.2.1.2. Tool Subscriptions"},
-                        {"id": "ft_1_3_1_2_1_3_framework_fees", "label": "1.3.1.2.1.3. Framework Fees"}
-                      ]
-                    }
-                  ]
-                },
-                {
-                  "id": "ss_task_1_3_1_3_personnel_costs",
-                  "label": "1.3.1.3. Personnel Costs",
-                  "sub_sub_subtasks": [
-                    {
-                      "id": "sss_task_1_3_1_3_1_salaries",
-                      "label": "1.3.1.3.1. Salaries",
-                      "final_level_tasks": [
-                        {"id": "ft_1_3_1_3_1_1_engineer_salaries", "label": "1.3.1.3.1.1. Engineer Salaries"},
-                        {"id": "ft_1_3_1_3_1_2_consultant_fees", "label": "1.3.1.3.1.2. Consultant Fees"},
-                        {"id": "ft_1_3_1_3_1_3_training_budget", "label": "1.3.1.3.1.3. Training Budget"}
-                      ]
-                    }
-                  ]
-                }
-              ]
-            }
-          ]
-        }
+        {"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>"}
       ]
     }
   ]