Update sample_data.py

sample_data.py

@@ -305,53 +305,71 @@ RADIAL_DIAGRAM_JSON = """
 
 PROCESS_FLOW_JSON = """
 {
-  "start_node": "Start Order Process",
+  "start_node": "Start Inference Request",
   "nodes": [
     {
-      "id": "customer_order",
-      "label": "Customer Places Order",
+      "id": "user_input",
+      "label": "Receive User Input (Data)",
+      "type": "io"
+    },
+    {
+      "id": "preprocess_data",
+      "label": "Preprocess Data",
       "type": "process"
     },
     {
-      "id": "check_stock",
-      "label": "Check Stock Availability",
+      "id": "validate_data",
+      "label": "Validate Data Format/Type",
       "type": "decision"
     },
     {
-      "id": "stock_available",
-      "label": "Stock Available?",
+      "id": "data_valid_yes",
+      "label": "Data Valid?",
       "type": "decision"
     },
     {
-      "id": "process_payment",
-      "label": "Process Payment",
+      "id": "load_model",
+      "label": "Load AI Model (if not cached)",
       "type": "process"
     },
     {
-      "id": "order_confirmed",
-      "label": "Order Confirmed",
+      "id": "run_inference",
+      "label": "Run AI Model Inference",
       "type": "process"
     },
     {
-      "id": "notify_customer_oos",
-      "label": "Notify Customer (Out of Stock)",
+      "id": "postprocess_output",
+      "label": "Postprocess Model Output",
       "type": "process"
     },
     {
-      "id": "end_order_process",
-      "label": "End Order Process",
+      "id": "send_response",
+      "label": "Send Response to User",
+      "type": "io"
+    },
+    {
+      "id": "log_error",
+      "label": "Log Error & Notify User",
+      "type": "process"
+    },
+    {
+      "id": "end_inference_process",
+      "label": "End Inference Process",
       "type": "end"
     }
   ],
   "connections": [
-    {"from": "start_node", "to": "customer_order", "label": "Initiate"},
-    {"from": "customer_order", "to": "check_stock", "label": "Proceed"},
-    {"from": "check_stock", "to": "stock_available", "label": "Result"},
-    {"from": "stock_available", "to": "process_payment", "label": "Yes"},
-    {"from": "stock_available", "to": "notify_customer_oos", "label": "No"},
-    {"from": "process_payment", "to": "order_confirmed", "label": "Success"},
-    {"from": "order_confirmed", "to": "end_order_process", "label": "Complete"},
-    {"from": "notify_customer_oos", "to": "end_order_process", "label": "Finish"}
+    {"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"}
   ]
 }
 """
@@ -359,46 +377,42 @@ PROCESS_FLOW_JSON = """
 # New JSON for Work Breakdown Structure (WBS) Diagram - similar to image, but not identical
 WBS_DIAGRAM_JSON = """
 {
-  "project_title": "Website Redesign Project",
+  "project_title": "AI Model Deployment Project",
   "phases": [
     {
-      "id": "phase_1_initiation",
-      "label": "1. Project Initiation",
+      "id": "phase_1_prep",
+      "label": "1. Preparation",
       "tasks": [
-        {"id": "task_1_1_define", "label": "1.1. Define Goals & Scope"},
-        {"id": "task_1_2_team", "label": "1.2. Assemble Team"},
-        {"id": "task_1_3_budget", "label": "1.3. Establish Budget"}
+        {"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 Phase",
-      "tasks": [
-        {"id": "task_2_1_req", "label": "2.1. Gather Requirements"},
-        {"id": "task_2_2_design", "label": "2.2. Design UI/UX"},
-        {"id": "task_2_3_tech_stack", "label": "2.3. Select Tech Stack"},
-        {"id": "task_2_4_sitemap", "label": "2.4. Create Sitemap"},
-        {"id": "task_2_5_content_plan", "label": "2.5. Plan Content Strategy"}
-      ]
-    },
-    {
-      "id": "phase_3_execution",
-      "label": "3. Execution Phase",
+      "label": "2. Planning",
       "tasks": [
-        {"id": "task_3_1_frontend", "label": "3.1. Develop Frontend"},
-        {"id": "task_3_2_backend", "label": "3.2. Develop Backend"},
-        {"id": "task_3_3_db", "label": "3.3. Database Integration"},
-        {"id": "task_3_4_content_creation", "label": "3.4. Content Creation"},
-        {"id": "task_3_5_testing", "label": "3.5. Internal Testing"}
+        {"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_4_deployment",
-      "label": "4. Deployment & Launch",
+      "id": "phase_3_dev",
+      "label": "3. Development",
       "tasks": [
-        {"id": "task_4_1_pre_launch", "label": "4.1. Pre-Launch Checks"},
-        {"id": "task_4_2_go_live", "label": "4.1. Go Live"},
-        {"id": "task_4_3_monitoring", "label": "4.3. Post-Launch Monitoring"}
+        {"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"}
       ]
     }
   ]