Update sample_data.py

sample_data.py

@@ -380,28 +380,58 @@ WBS_DIAGRAM_JSON = """
   "project_title": "AI Model Deployment Project",
   "phases": [
     {
-      "id": "phase_1_prep",
-      "label": "1. Preparation",
+      "id": "phase_1_initiation",
+      "label": "1. Project Initiation",
       "tasks": [
         {
           "id": "task_1_1_goal_scope",
           "label": "1.1. Define Goal & Scope",
           "subtasks": [
             {
-              "id": "subtask_1_1_1_ai_problem",
-              "label": "1.1.1. AI Problem Definition",
-              "granular_tasks": [
+              "id": "subtask_1_1_1_problem_def",
+              "label": "1.1.1. Problem Definition",
+              "sub_subtasks": [
                 {
-                  "id": "gt_1_1_1_1_understand_req",
-                  "label": "1.1.1.1. Understand Requirements",
-                  "sub_granular_tasks": [
+                  "id": "ss_task_1_1_1_1_req_analysis",
+                  "label": "1.1.1.1. Requirements Analysis",
+                  "sub_sub_subtasks": [
                     {
-                      "id": "sgt_1_1_1_1_1_gather_stakeholder_input",
-                      "label": "1.1.1.1.1. Gather Stakeholder Input",
-                      "final_tasks": [
+                      "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_analyze_feedback", "label": "1.1.1.1.1.2. Analyze Feedback"},
-                        {"id": "ft_1_1_1_1_1_3_document_needs", "label": "1.1.1.1.1.3. Document Needs"}
+                        {"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"}
                       ]
                     }
                   ]
@@ -411,24 +441,54 @@ WBS_DIAGRAM_JSON = """
           ]
         },
         {
-          "id": "task_1_2_resource_id",
-          "label": "1.2. Identify Resources",
+          "id": "task_1_2_resource_plan",
+          "label": "1.2. Resource Planning",
           "subtasks": [
             {
-              "id": "subtask_1_2_1_data_sources",
-              "label": "1.2.1. Data Sources Identification",
-              "granular_tasks": [
+              "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": "gt_1_2_1_1_list_internal_data",
-                  "label": "1.2.1.1. List Internal Data",
-                  "sub_granular_tasks": [
+                  "id": "ss_task_1_2_1_2_engineers",
+                  "label": "1.2.1.2. Software Engineers",
+                  "sub_sub_subtasks": [
                     {
-                      "id": "sgt_1_2_1_1_1_check_db_access",
-                      "label": "1.2.1.1.1. Check DB Access",
-                      "final_tasks": [
-                        {"id": "ft_1_2_1_1_1_1_verify_credentials", "label": "1.2.1.1.1.1. Verify Credentials"},
-                        {"id": "ft_1_2_1_1_1_2_test_connections", "label": "1.2.1.1.1.2. Test Connections"},
-                        {"id": "ft_1_2_1_1_1_3_secure_apis", "label": "1.2.1.1.1.3. Secure APIs"}
+                      "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."}
                       ]
                     }
                   ]
@@ -438,24 +498,54 @@ WBS_DIAGRAM_JSON = """
           ]
         },
         {
-          "id": "task_1_3_env_setup",
-          "label": "1.3. Environment Setup",
+          "id": "task_1_3_budget_timeline",
+          "label": "1.3. Budget & Timeline",
           "subtasks": [
             {
-              "id": "subtask_1_3_1_cloud_infra",
-              "label": "1.3.1. Cloud Infra. Provisioning",
-              "granular_tasks": [
+              "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": "gt_1_3_1_1_deploy_vms",
-                  "label": "1.3.1.1. Deploy VMs",
-                  "sub_granular_tasks": [
+                  "id": "ss_task_1_3_1_3_personnel_costs",
+                  "label": "1.3.1.3. Personnel Costs",
+                  "sub_sub_subtasks": [
                     {
-                      "id": "sgt_1_3_1_1_1_select_instance_types",
-                      "label": "1.3.1.1.1. Select Instance Types",
-                      "final_tasks": [
-                        {"id": "ft_1_3_1_1_1_1_gpu_spec", "label": "1.3.1.1.1.1. GPU Spec"},
-                        {"id": "ft_1_3_1_1_1_2_cpu_mem", "label": "1.3.1.1.1.2. CPU/Mem"},
-                        {"id": "ft_1_3_1_1_1_3_storage_opts", "label": "1.3.1.1.1.3. Storage Options"}
+                      "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"}
                       ]
                     }
                   ]