Update sample_data.py

sample_data.py

@@ -1,86 +1,181 @@
+# Enhanced Sample Data for AI Diagram Generator
+# Modern and practical AI-related sample data
+
 CONCEPT_MAP_JSON = """
 {
-  "central_node": "Artificial Intelligence (AI)",
+  "central_node": "Generative AI Ecosystem",
   "nodes": [
     {
-      "id": "ml_fundamental",
-      "label": "Machine Learning",
-      "relationship": "is essential for",
+      "id": "foundation_models",
+      "label": "Foundation Models",
+      "relationship": "powered by",
       "subnodes": [
         {
-          "id": "dl_branch",
-          "label": "Deep Learning",
-          "relationship": "for example",
+          "id": "llm_models",
+          "label": "Large Language Models",
+          "relationship": "includes",
           "subnodes": [
             {
-              "id": "cnn_example",
-              "label": "CNNs",
-              "relationship": "for example"
+              "id": "gpt_family",
+              "label": "GPT Family",
+              "relationship": "such as",
+              "subnodes": [
+                {
+                  "id": "gpt4",
+                  "label": "GPT-4",
+                  "relationship": "latest version"
+                },
+                {
+                  "id": "chatgpt",
+                  "label": "ChatGPT",
+                  "relationship": "application"
+                }
+              ]
             },
             {
-              "id": "rnn_example",
-              "label": "RNNs",
-              "relationship": "for example"
+              "id": "claude_family",
+              "label": "Claude Family",
+              "relationship": "such as",
+              "subnodes": [
+                {
+                  "id": "claude3",
+                  "label": "Claude 3 Opus",
+                  "relationship": "advanced model"
+                },
+                {
+                  "id": "claude_sonnet",
+                  "label": "Claude Sonnet",
+                  "relationship": "balanced model"
+                }
+              ]
+            },
+            {
+              "id": "opensource_llm",
+              "label": "Open Source LLMs",
+              "relationship": "alternatives",
+              "subnodes": [
+                {
+                  "id": "llama3",
+                  "label": "LLaMA 3",
+                  "relationship": "by Meta"
+                },
+                {
+                  "id": "mistral",
+                  "label": "Mistral 7B",
+                  "relationship": "efficient"
+                }
+              ]
             }
           ]
         },
         {
-          "id": "rl_branch",
-          "label": "Reinforcement Learning",
-          "relationship": "for example",
+          "id": "multimodal_models",
+          "label": "Multimodal Models",
+          "relationship": "extends to",
           "subnodes": [
             {
-              "id": "qlearning_example",
-              "label": "Q-Learning",
-              "relationship": "example"
+              "id": "vision_language",
+              "label": "Vision-Language Models",
+              "relationship": "including",
+              "subnodes": [
+                {
+                  "id": "dall_e3",
+                  "label": "DALL-E 3",
+                  "relationship": "image generation"
+                },
+                {
+                  "id": "midjourney",
+                  "label": "Midjourney v6",
+                  "relationship": "artistic creation"
+                }
+              ]
             },
             {
-              "id": "pg_example",
-              "label": "Policy Gradients",
-              "relationship": "example"
+              "id": "video_models",
+              "label": "Video Generation",
+              "relationship": "emerging with",
+              "subnodes": [
+                {
+                  "id": "sora",
+                  "label": "Sora",
+                  "relationship": "by OpenAI"
+                },
+                {
+                  "id": "runway",
+                  "label": "Runway Gen-2",
+                  "relationship": "creative tools"
+                }
+              ]
             }
           ]
         }
       ]
     },
     {
-      "id": "ai_types",
-      "label": "Types",
-      "relationship": "formed by",
+      "id": "ai_applications",
+      "label": "Real-World Applications",
+      "relationship": "deployed in",
       "subnodes": [
         {
-          "id": "agi_type",
-          "label": "AGI",
-          "relationship": "this is",
+          "id": "enterprise_ai",
+          "label": "Enterprise Solutions",
+          "relationship": "transforming",
           "subnodes": [
             {
-              "id": "strong_ai",
-              "label": "Strong AI",
-              "relationship": "provoked by",
+              "id": "ai_assistants",
+              "label": "AI Assistants",
+              "relationship": "like",
               "subnodes": [
                 {
-                  "id": "human_intel",
-                  "label": "Human-level Intel.",
-                  "relationship": "of"
+                  "id": "copilot",
+                  "label": "GitHub Copilot",
+                  "relationship": "code generation"
+                },
+                {
+                  "id": "duet_ai",
+                  "label": "Google Duet AI",
+                  "relationship": "workspace integration"
+                }
+              ]
+            },
+            {
+              "id": "automation",
+              "label": "Process Automation",
+              "relationship": "through",
+              "subnodes": [
+                {
+                  "id": "rpa_ai",
+                  "label": "AI-Enhanced RPA",
+                  "relationship": "intelligent automation"
+                },
+                {
+                  "id": "document_ai",
+                  "label": "Document Intelligence",
+                  "relationship": "data extraction"
                 }
               ]
             }
           ]
         },
         {
-          "id": "ani_type",
-          "label": "ANI",
-          "relationship": "this is",
+          "id": "creative_ai",
+          "label": "Creative Industries",
+          "relationship": "revolutionizing",
           "subnodes": [
             {
-              "id": "weak_ai",
-              "label": "Weak AI",
-              "relationship": "provoked by",
+              "id": "content_creation",
+              "label": "Content Generation",
+              "relationship": "enabling",
               "subnodes": [
                 {
-                  "id": "narrow_tasks",
-                  "label": "Narrow Tasks",
-                  "relationship": "of"
+                  "id": "marketing_ai",
+                  "label": "Marketing Copy",
+                  "relationship": "automated creation"
+                },
+                {
+                  "id": "game_assets",
+                  "label": "Game Asset Generation",
+                  "relationship": "procedural design"
                 }
               ]
             }
@@ -89,71 +184,53 @@ CONCEPT_MAP_JSON = """
       ]
     },
     {
-      "id": "ai_capabilities",
-      "label": "Capabilities",
-      "relationship": "change",
+      "id": "ai_infrastructure",
+      "label": "AI Infrastructure",
+      "relationship": "supported by",
       "subnodes": [
         {
-          "id": "data_proc",
-          "label": "Data Processing",
-          "relationship": "can",
+          "id": "compute_resources",
+          "label": "Compute Resources",
+          "relationship": "requiring",
           "subnodes": [
             {
-              "id": "big_data",
-              "label": "Big Data",
-              "relationship": "as",
+              "id": "gpu_clusters",
+              "label": "GPU Clusters",
+              "relationship": "powered by",
               "subnodes": [
                 {
-                  "id": "analysis_example",
-                  "label": "Data Analysis",
-                  "relationship": "example"
+                  "id": "nvidia_h100",
+                  "label": "NVIDIA H100",
+                  "relationship": "latest generation"
                 },
                 {
-                  "id": "prediction_example",
-                  "label": "Prediction",
-                  "relationship": "example"
+                  "id": "tpu_v5",
+                  "label": "Google TPU v5",
+                  "relationship": "specialized chips"
                 }
               ]
             }
           ]
         },
         {
-          "id": "decision_making",
-          "label": "Decision Making",
-          "relationship": "can be",
+          "id": "mlops_platform",
+          "label": "MLOps Platforms",
+          "relationship": "managed by",
           "subnodes": [
             {
-              "id": "automation",
-              "label": "Automation",
-              "relationship": "as",
+              "id": "model_serving",
+              "label": "Model Serving",
+              "relationship": "via",
               "subnodes": [
                 {
-                  "id": "robotics_example",
-                  "label": "Robotics",
-                  "relationship": "Example"},
-                {
-                  "id": "autonomous_example",
-                  "label": "Autonomous Vehicles",
-                  "relationship": "of one"
-                }
-              ]
-            }
-          ]
-        },
-        {
-          "id": "problem_solving",
-          "label": "Problem Solving",
-          "relationship": "can",
-          "subnodes": [
-            {
-              "id": "optimization",
-              "label": "Optimization",
-              "relationship": "as is",
-              "subnodes": [
+                  "id": "huggingface",
+                  "label": "Hugging Face",
+                  "relationship": "model hub"
+                },
                 {
-                  "id": "algorithms_example",
-                  "label": "Algorithms",
-                  "relationship": "for example"
+                  "id": "vertex_ai",
+                  "label": "Vertex AI",
+                  "relationship": "Google Cloud"
                 }
               ]
             }
@@ -165,82 +242,81 @@ CONCEPT_MAP_JSON = """
 }
 """
 
-# JSON for Synoptic Chart (horizontal hierarchy) - AI related, 4 levels
 SYNOPTIC_CHART_JSON = """
 {
-  "central_node": "AI Project Lifecycle",
+  "central_node": "Modern AI Development Pipeline",
   "nodes": [
     {
       "id": "phase1",
-      "label": "I. Problem Definition & Data Acquisition",
-      "relationship": "Starts with",
+      "label": "I. Data Engineering & Preparation",
+      "relationship": "begins with",
       "subnodes": [
         {
-          "id": "sub1_1",
-          "label": "1. Problem Formulation",
-          "relationship": "Involves",
+          "id": "data_strategy",
+          "label": "1. Data Strategy",
+          "relationship": "establishing",
           "subnodes": [
-            {"id": "sub1_1_1", "label": "1.1. Identify Business Need", "relationship": "e.g."},
-            {"id": "sub1_1_2", "label": "1.2. Define KPIs", "relationship": "e.g."}
+            {"id": "data_sources", "label": "1.1. Multi-Source Integration", "relationship": "includes", "subnodes": []},
+            {"id": "data_governance", "label": "1.2. Privacy & Compliance", "relationship": "ensures", "subnodes": []}
           ]
         },
         {
-          "id": "sub1_2",
-          "label": "2. Data Collection",
-          "relationship": "Followed by",
+          "id": "data_pipeline",
+          "label": "2. Data Pipeline",
+          "relationship": "building",
           "subnodes": [
-            {"id": "sub1_2_1", "label": "2.1. Source Data", "relationship": "from"},
-            {"id": "sub1_2_2", "label": "2.2. Data Cleaning", "relationship": "includes"}
+            {"id": "etl_process", "label": "2.1. ETL/ELT Workflows", "relationship": "implements", "subnodes": []},
+            {"id": "feature_store", "label": "2.2. Feature Store Setup", "relationship": "centralizes", "subnodes": []}
           ]
         }
       ]
     },
     {
       "id": "phase2",
-      "label": "II. Model Development",
-      "relationship": "Proceeds to",
+      "label": "II. Model Development & Training",
+      "relationship": "continues to",
       "subnodes": [
         {
-          "id": "sub2_1",
-          "label": "1. Feature Engineering",
-          "relationship": "Comprises",
+          "id": "experimentation",
+          "label": "1. Experimentation Platform",
+          "relationship": "enabling",
           "subnodes": [
-            {"id": "sub2_1_1", "label": "1.1. Feature Selection", "relationship": "e.g."},
-            {"id": "sub2_1_2", "label": "1.2. Feature Transformation", "relationship": "e.g."}
+            {"id": "experiment_tracking", "label": "1.1. MLflow/W&B Integration", "relationship": "tracks", "subnodes": []},
+            {"id": "hyperparameter_opt", "label": "1.2. AutoML & HPO", "relationship": "optimizes", "subnodes": []}
           ]
         },
         {
-          "id": "sub2_2",
-          "label": "2. Model Training",
-          "relationship": "Involves",
+          "id": "distributed_training",
+          "label": "2. Scalable Training",
+          "relationship": "leveraging",
           "subnodes": [
-            {"id": "sub2_2_1", "label": "2.1. Algorithm Selection", "relationship": "uses"},
-            {"id": "sub2_2_2", "label": "2.2. Hyperparameter Tuning", "relationship": "optimizes"}
+            {"id": "multi_gpu", "label": "2.1. Multi-GPU Training", "relationship": "accelerates", "subnodes": []},
+            {"id": "model_parallelism", "label": "2.2. Model Parallelism", "relationship": "enables", "subnodes": []}
           ]
         }
       ]
     },
     {
       "id": "phase3",
-      "label": "III. Evaluation & Deployment",
-      "relationship": "Culminates in",
+      "label": "III. Deployment & Operations",
+      "relationship": "culminates in",
       "subnodes": [
         {
-          "id": "sub3_1",
-          "label": "1. Model Evaluation",
-          "relationship": "Includes",
+          "id": "model_deployment",
+          "label": "1. Production Deployment",
+          "relationship": "implementing",
           "subnodes": [
-            {"id": "sub3_1_1", "label": "1.1. Performance Metrics", "relationship": "measures"},
-            {"id": "sub3_1_2", "label": "1.2. Bias & Fairness Audits", "relationship": "ensures"}
+            {"id": "edge_deployment", "label": "1.1. Edge Computing", "relationship": "optimizes", "subnodes": []},
+            {"id": "api_gateway", "label": "1.2. API Management", "relationship": "exposes", "subnodes": []}
           ]
         },
         {
-          "id": "sub3_2",
-          "label": "2. Deployment & Monitoring",
-          "relationship": "Requires",
+          "id": "monitoring_ops",
+          "label": "2. AI Operations",
+          "relationship": "maintaining",
           "subnodes": [
-            {"id": "sub3_2_1", "label": "2.1. API/Integration Development", "relationship": "for"},
-            {"id": "sub3_2_2", "label": "2.2. Continuous Monitoring", "relationship": "ensures"}
+            {"id": "model_monitoring", "label": "2.1. Drift Detection", "relationship": "monitors", "subnodes": []},
+            {"id": "feedback_loop", "label": "2.2. Continuous Learning", "relationship": "improves", "subnodes": []}
           ]
         }
       ]
@@ -249,54 +325,53 @@ SYNOPTIC_CHART_JSON = """
 }
 """
 
-# JSON for Radial Diagram (central expansion) - AI related, 3 levels with 5->10 structure
 RADIAL_DIAGRAM_JSON = """
 {
-  "central_node": "AI Core Concepts & Domains",
+  "central_node": "AI Innovation Hub",
   "nodes": [
     {
-      "id": "foundational_ml",
-      "label": "Foundational ML",
-      "relationship": "builds on",
+      "id": "emerging_tech",
+      "label": "Emerging Technologies",
+      "relationship": "advancing through",
       "subnodes": [
-        {"id": "supervised_l", "label": "Supervised Learning", "relationship": "e.g."},
-        {"id": "unsupervised_l", "label": "Unsupervised Learning", "relationship": "e.g."}
+        {"id": "quantum_ml", "label": "Quantum Machine Learning", "relationship": "breakthrough in", "subnodes": []},
+        {"id": "neuromorphic", "label": "Neuromorphic Computing", "relationship": "inspired by", "subnodes": []}
       ]
     },
     {
-      "id": "dl_architectures",
-      "label": "Deep Learning Arch.",
-      "relationship": "evolved from",
+      "id": "ai_safety",
+      "label": "AI Safety & Ethics",
+      "relationship": "ensuring through",
       "subnodes": [
-        {"id": "cnns_rad", "label": "CNNs", "relationship": "e.g."},
-        {"id": "rnns_rad", "label": "RNNs", "relationship": "e.g."}
+        {"id": "alignment", "label": "AI Alignment Research", "relationship": "focusing on", "subnodes": []},
+        {"id": "explainable_ai", "label": "Explainable AI (XAI)", "relationship": "providing", "subnodes": []}
       ]
     },
     {
-      "id": "major_applications",
-      "label": "Major AI Applications",
-      "relationship": "applied in",
+      "id": "industry_impact",
+      "label": "Industry Transformation",
+      "relationship": "disrupting",
       "subnodes": [
-        {"id": "nlp_rad", "label": "Natural Language Processing", "relationship": "e.g."},
-        {"id": "cv_rad", "label": "Computer Vision", "relationship": "e.g."}
+        {"id": "healthcare_ai", "label": "AI in Healthcare", "relationship": "revolutionizing", "subnodes": []},
+        {"id": "fintech_ai", "label": "AI in Finance", "relationship": "automating", "subnodes": []}
       ]
     },
     {
-      "id": "ethical_concerns",
-      "label": "Ethical AI Concerns",
-      "relationship": "addresses",
+      "id": "research_frontiers",
+      "label": "Research Frontiers",
+      "relationship": "exploring",
       "subnodes": [
-        {"id": "fairness_rad", "label": "Fairness & Bias", "relationship": "e.g."},
-        {"id": "explainability", "label": "Explainability (XAI)", "relationship": "e.g."}
+        {"id": "agi_research", "label": "AGI Development", "relationship": "pursuing", "subnodes": []},
+        {"id": "consciousness_ai", "label": "Machine Consciousness", "relationship": "investigating", "subnodes": []}
       ]
     },
     {
-      "id": "future_trends",
-      "label": "Future AI Trends",
-      "relationship": "looking at",
+      "id": "ai_ecosystem",
+      "label": "AI Ecosystem",
+      "relationship": "building",
       "subnodes": [
-        {"id": "agi_future", "label": "AGI Development", "relationship": "e.g."},
-        {"id": "quantum_ai", "label": "Quantum AI", "relationship": "e.g."}
+        {"id": "open_source", "label": "Open Source Community", "relationship": "contributing to", "subnodes": []},
+        {"id": "ai_startups", "label": "AI Startup Ecosystem", "relationship": "innovating in", "subnodes": []}
       ]
     }
   ]
@@ -305,101 +380,176 @@ RADIAL_DIAGRAM_JSON = """
 
 PROCESS_FLOW_JSON = """
 {
-  "start_node": "Start Inference Request",
+  "start_node": "AI Request Initiated",
   "nodes": [
     {
-      "id": "user_input",
-      "label": "Receive User Input (Data)",
-      "type": "io"
+      "id": "auth_check",
+      "label": "Authentication & Rate Limiting",
+      "type": "process"
     },
     {
-      "id": "preprocess_data",
-      "label": "Preprocess Data",
+      "id": "auth_valid",
+      "label": "Valid Request?",
+      "type": "decision"
+    },
+    {
+      "id": "request_router",
+      "label": "Request Router",
       "type": "process"
     },
     {
-      "id": "validate_data",
-      "label": "Validate Data Format/Type",
+      "id": "cache_check",
+      "label": "Check Response Cache",
       "type": "decision"
     },
     {
-      "id": "data_valid_yes",
-      "label": "Data Valid?",
+      "id": "model_selector",
+      "label": "Model Selection Logic",
+      "type": "process"
+    },
+    {
+      "id": "load_balancer",
+      "label": "Load Balancer",
+      "type": "process"
+    },
+    {
+      "id": "inference_engine",
+      "label": "Run Inference",
+      "type": "process"
+    },
+    {
+      "id": "quality_check",
+      "label": "Quality Assessment",
       "type": "decision"
     },
     {
-      "id": "load_model",
-      "label": "Load AI Model (if not cached)",
+      "id": "post_processor",
+      "label": "Post-Processing & Filtering",
       "type": "process"
     },
     {
-      "id": "run_inference",
-      "label": "Run AI Model Inference",
+      "id": "response_cache",
+      "label": "Update Cache",
       "type": "process"
     },
     {
-      "id": "postprocess_output",
-      "label": "Postprocess Model Output",
+      "id": "analytics_log",
+      "label": "Log Analytics",
       "type": "process"
     },
     {
-      "id": "send_response",
-      "label": "Send Response to User",
+      "id": "deliver_response",
+      "label": "Deliver Response",
       "type": "io"
     },
     {
-      "id": "log_error",
-      "label": "Log Error & Notify User",
+      "id": "error_handler",
+      "label": "Error Handler",
       "type": "process"
     },
     {
-      "id": "end_inference_process",
-      "label": "End Inference Process",
+      "id": "fallback_model",
+      "label": "Fallback Model",
+      "type": "process"
+    },
+    {
+      "id": "end_process",
+      "label": "Complete",
       "type": "end"
     }
   ],
   "connections": [
-    {"from": "start_node", "to": "user_input", "label": "Request"},
-    {"from": "user_input", "to": "preprocess_data", "label": "Data Received"},
-    {"from": "preprocess_data", "to": "validate_data", "label": "Cleaned"},
-    {"from": "validate_data", "to": "data_valid_yes", "label": "Check"},
-    {"from": "data_valid_yes", "to": "load_model", "label": "Yes"},
-    {"from": "data_valid_yes", "to": "log_error", "label": "No"},
-    {"from": "load_model", "to": "run_inference", "label": "Model Ready"},
-    {"from": "run_inference", "to": "postprocess_output", "label": "Output Generated"},
-    {"from": "postprocess_output", "to": "send_response", "label": "Ready"},
-    {"from": "send_response", "to": "end_inference_process", "label": "Response Sent"},
-    {"from": "log_error", "to": "end_inference_process", "label": "Error Handled"}
+    {"from": "start_node", "to": "auth_check", "label": "Begin"},
+    {"from": "auth_check", "to": "auth_valid", "label": "Verify"},
+    {"from": "auth_valid", "to": "request_router", "label": "Authorized"},
+    {"from": "auth_valid", "to": "error_handler", "label": "Denied"},
+    {"from": "request_router", "to": "cache_check", "label": "Route"},
+    {"from": "cache_check", "to": "deliver_response", "label": "Cache Hit"},
+    {"from": "cache_check", "to": "model_selector", "label": "Cache Miss"},
+    {"from": "model_selector", "to": "load_balancer", "label": "Selected"},
+    {"from": "load_balancer", "to": "inference_engine", "label": "Assigned"},
+    {"from": "inference_engine", "to": "quality_check", "label": "Generated"},
+    {"from": "quality_check", "to": "post_processor", "label": "Pass"},
+    {"from": "quality_check", "to": "fallback_model", "label": "Fail"},
+    {"from": "fallback_model", "to": "post_processor", "label": "Retry"},
+    {"from": "post_processor", "to": "response_cache", "label": "Processed"},
+    {"from": "response_cache", "to": "analytics_log", "label": "Cached"},
+    {"from": "analytics_log", "to": "deliver_response", "label": "Logged"},
+    {"from": "deliver_response", "to": "end_process", "label": "Delivered"},
+    {"from": "error_handler", "to": "end_process", "label": "Handled"}
   ]
 }
 """
 
-# New JSON for Work Breakdown Structure (WBS) Diagram - similar to image, but not identical
 WBS_DIAGRAM_JSON = """
 {
-  "project_title": "AI Model Development Project",
+  "project_title": "Enterprise AI Platform Implementation",
   "phases": [
     {
-      "id": "phase_prep",
-      "label": "Preparation",
+      "id": "phase_foundation",
+      "label": "Foundation & Architecture",
       "tasks": [
         {
-          "id": "task_1_1_vision",
-          "label": "Identify Vision",
+          "id": "task_requirements",
+          "label": "Requirements Analysis",
+          "subtasks": [
+            {
+              "id": "subtask_stakeholder",
+              "label": "Stakeholder Mapping",
+              "sub_subtasks": [
+                {
+                  "id": "ss_task_interviews",
+                  "label": "Executive Interviews",
+                  "sub_sub_subtasks": [
+                    {
+                      "id": "sss_task_schedule",
+                      "label": "Schedule Sessions",
+                      "final_level_tasks": [
+                        {"id": "ft_prep_materials", "label": "Prepare Materials"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            },
+            {
+              "id": "subtask_technical",
+              "label": "Technical Assessment",
+              "sub_subtasks": [
+                {
+                  "id": "ss_task_infra",
+                  "label": "Infrastructure Audit",
+                  "sub_sub_subtasks": [
+                    {
+                      "id": "sss_task_compute",
+                      "label": "Compute Resources",
+                      "final_level_tasks": [
+                        {"id": "ft_gpu_assessment", "label": "GPU Capacity Planning"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            }
+          ]
+        },
+        {
+          "id": "task_architecture",
+          "label": "Solution Architecture",
           "subtasks": [
             {
-              "id": "subtask_1_1_1_design_staff",
-              "label": "Design & Staffing",
+              "id": "subtask_design",
+              "label": "System Design",
               "sub_subtasks": [
                 {
-                  "id": "ss_task_1_1_1_1_env_setup",
-                  "label": "Environment Setup",
+                  "id": "ss_task_microservices",
+                  "label": "Microservices Architecture",
                   "sub_sub_subtasks": [
                     {
-                      "id": "sss_task_1_1_1_1_1_lib_install",
-                      "label": "Install Libraries",
+                      "id": "sss_task_api_design",
+                      "label": "API Design",
                       "final_level_tasks": [
-                        {"id": "ft_1_1_1_1_1_1_data_access", "label": "Grant Data Access"}
+                        {"id": "ft_api_specs", "label": "OpenAPI Specifications"}
                       ]
                     }
                   ]
@@ -411,26 +561,70 @@ WBS_DIAGRAM_JSON = """
       ]
     },
     {
-      "id": "phase_plan",
-      "label": "Planning",
+      "id": "phase_implementation",
+      "label": "Core Implementation",
       "tasks": [
         {
-          "id": "task_2_1_cost_analysis",
-          "label": "Cost Analysis",
+          "id": "task_data_platform",
+          "label": "Data Platform",
+          "subtasks": [
+            {
+              "id": "subtask_data_lake",
+              "label": "Data Lake Setup",
+              "sub_subtasks": [
+                {
+                  "id": "ss_task_storage",
+                  "label": "Storage Configuration",
+                  "sub_sub_subtasks": [
+                    {
+                      "id": "sss_task_s3_setup",
+                      "label": "S3 Bucket Setup",
+                      "final_level_tasks": [
+                        {"id": "ft_lifecycle_policies", "label": "Lifecycle Policies"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            },
+            {
+              "id": "subtask_streaming",
+              "label": "Real-time Pipeline",
+              "sub_subtasks": [
+                {
+                  "id": "ss_task_kafka",
+                  "label": "Kafka Implementation",
+                  "sub_sub_subtasks": [
+                    {
+                      "id": "sss_task_topics",
+                      "label": "Topic Design",
+                      "final_level_tasks": [
+                        {"id": "ft_schema_registry", "label": "Schema Registry"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            }
+          ]
+        },
+        {
+          "id": "task_ml_platform",
+          "label": "ML Platform",
           "subtasks": [
             {
-              "id": "subtask_2_1_1_benefit_analysis",
-              "label": "Benefit Analysis",
+              "id": "subtask_mlops",
+              "label": "MLOps Setup",
               "sub_subtasks": [
                 {
-                  "id": "ss_task_2_1_1_1_risk_assess",
-                  "label": "AI Risk Assessment",
+                  "id": "ss_task_ci_cd",
+                  "label": "CI/CD Pipeline",
                   "sub_sub_subtasks": [
                     {
-                      "id": "sss_task_2_1_1_1_1_model_selection",
-                      "label": "Model Selection",
+                      "id": "sss_task_github_actions",
+                      "label": "GitHub Actions",
                       "final_level_tasks": [
-                        {"id": "ft_2_1_1_1_1_1_data_strategy", "label": "Data Strategy"}
+                        {"id": "ft_model_registry", "label": "Model Registry Integration"}
                       ]
                     }
                   ]
@@ -442,26 +636,45 @@ WBS_DIAGRAM_JSON = """
       ]
     },
     {
-      "id": "phase_dev",
-      "label": "Development",
+      "id": "phase_deployment",
+      "label": "Deployment & Operations",
       "tasks": [
         {
-          "id": "task_3_1_change_mgmt",
-          "label": "Data Preprocessing",
+          "id": "task_production",
+          "label": "Production Deployment",
           "subtasks": [
             {
-              "id": "subtask_3_1_1_implementation",
-              "label": "Feature Engineering",
+              "id": "subtask_kubernetes",
+              "label": "Kubernetes Orchestration",
+              "sub_subtasks": [
+                {
+                  "id": "ss_task_helm",
+                  "label": "Helm Charts",
+                  "sub_sub_subtasks": [
+                    {
+                      "id": "sss_task_autoscaling",
+                      "label": "Auto-scaling Config",
+                      "final_level_tasks": [
+                        {"id": "ft_hpa_setup", "label": "HPA Configuration"}
+                      ]
+                    }
+                  ]
+                }
+              ]
+            },
+            {
+              "id": "subtask_monitoring",
+              "label": "Observability Stack",
               "sub_subtasks": [
                 {
-                  "id": "ss_task_3_1_1_1_beta_testing",
-                  "label": "Model Training",
+                  "id": "ss_task_prometheus",
+                  "label": "Prometheus Setup",
                   "sub_sub_subtasks": [
                     {
-                      "id": "sss_task_3_1_1_1_1_other_task",
-                      "label": "Model Evaluation",
+                      "id": "sss_task_grafana",
+                      "label": "Grafana Dashboards",
                       "final_level_tasks": [
-                        {"id": "ft_3_1_1_1_1_1_hyperparam_tune", "label": "Hyperparameter Tuning"}
+                        {"id": "ft_alert_rules", "label": "Alert Configuration"}
                       ]
                     }
                   ]
@@ -474,5 +687,4 @@ WBS_DIAGRAM_JSON = """
     }
   ]
 }
-
-"""
+"""