Update sample_data.py

sample_data.py

@@ -383,36 +383,87 @@ WBS_DIAGRAM_JSON = """
       "id": "phase_1_prep",
       "label": "1. Preparation",
       "tasks": [
-        {"id": "task_1_1_vision", "label": "1.1. Define AI Goal"},
-        {"id": "task_1_2_data_id", "label": "1.2. Identify Data Sources"},
-        {"id": "task_1_3_infra_setup", "label": "1.3. Set Up Infra."},
-        {"id": "task_1_4_env_config", "label": "1.4. Configure Environment"},
-        {"id": "task_1_5_lib_install", "label": "1.5. Install Libraries"},
-        {"id": "task_1_6_access_grant", "label": "1.6. Grant Access"}
-      ]
-    },
-    {
-      "id": "phase_2_planning",
-      "label": "2. Planning",
-      "tasks": [
-        {"id": "task_2_1_model_sel", "label": "2.1. Model Selection"},
-        {"id": "task_2_2_data_strat", "label": "2.2. Data Strategy"},
-        {"id": "task_2_3_eval_metrics", "label": "2.3. Define Eval. Metrics"},
-        {"id": "task_2_4_resource_plan", "label": "2.4. Resource Planning"},
-        {"id": "task_2_5_risk_assess", "label": "2.5. Risk Assessment"},
-        {"id": "task_2_6_legal_review", "label": "2.6. Legal Review"}
-      ]
-    },
-    {
-      "id": "phase_3_dev",
-      "label": "3. Development",
-      "tasks": [
-        {"id": "task_3_1_data_prep", "label": "3.1. Data Preprocessing"},
-        {"id": "task_3_2_feature_eng", "label": "3.2. Feature Engineering"},
-        {"id": "task_3_3_model_train", "label": "3.3. Model Training"},
-        {"id": "task_3_4_model_eval", "label": "3.4. Model Evaluation"},
-        {"id": "task_3_5_hyperparam_tune", "label": "3.5. Hyperparam. Tuning"},
-        {"id": "task_3_6_version_control", "label": "3.6. Version Control"}
+        {
+          "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": "gt_1_1_1_1_understand_req",
+                  "label": "1.1.1.1. Understand Requirements",
+                  "sub_granular_tasks": [
+                    {
+                      "id": "sgt_1_1_1_1_1_gather_stakeholder_input",
+                      "label": "1.1.1.1.1. Gather Stakeholder Input",
+                      "final_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": "task_1_2_resource_id",
+          "label": "1.2. Identify Resources",
+          "subtasks": [
+            {
+              "id": "subtask_1_2_1_data_sources",
+              "label": "1.2.1. Data Sources Identification",
+              "granular_tasks": [
+                {
+                  "id": "gt_1_2_1_1_list_internal_data",
+                  "label": "1.2.1.1. List Internal Data",
+                  "sub_granular_tasks": [
+                    {
+                      "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": "task_1_3_env_setup",
+          "label": "1.3. Environment Setup",
+          "subtasks": [
+            {
+              "id": "subtask_1_3_1_cloud_infra",
+              "label": "1.3.1. Cloud Infra. Provisioning",
+              "granular_tasks": [
+                {
+                  "id": "gt_1_3_1_1_deploy_vms",
+                  "label": "1.3.1.1. Deploy VMs",
+                  "sub_granular_tasks": [
+                    {
+                      "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"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            }
+          ]
+        }
       ]
     }
   ]