Update process_flow_generator.py

process_flow_generator.py

@@ -15,32 +15,78 @@ def generate_process_flow_diagram(json_input: str) -> str: # Removed base_color
         json_input (str): A JSON string describing the process flow structure.
                           It must follow the Expected JSON Format Example below.
 
-    Returns:
-        str: The filepath to the generated PNG image file.
-
     Expected JSON Format Example:
     {
-      "start_node": "Start Order Process",
+      "start_node": "Start Inference Request",
       "nodes": [
-        {"id": "customer_order", "label": "Customer Places Order", "type": "process"},
-        {"id": "check_stock", "label": "Check Stock Availability", "type": "decision"},
-        {"id": "stock_available", "label": "Stock Available?", "type": "decision"},
-        {"id": "process_payment", "label": "Process Payment", "type": "process"},
-        {"id": "order_confirmed", "label": "Order Confirmed", "type": "process"},
-        {"id": "notify_customer_oos", "label": "Notify Customer (Out of Stock)", "type": "process"},
-        {"id": "end_order_process", "label": "End Order Process", "type": "end"}
+        {
+          "id": "user_input",
+          "label": "Receive User Input (Data)",
+          "type": "io"
+        },
+        {
+          "id": "preprocess_data",
+          "label": "Preprocess Data",
+          "type": "process"
+        },
+        {
+          "id": "validate_data",
+          "label": "Validate Data Format/Type",
+          "type": "decision"
+        },
+        {
+          "id": "data_valid_yes",
+          "label": "Data Valid?",
+          "type": "decision"
+        },
+        {
+          "id": "load_model",
+          "label": "Load AI Model (if not cached)",
+          "type": "process"
+        },
+        {
+          "id": "run_inference",
+          "label": "Run AI Model Inference",
+          "type": "process"
+        },
+        {
+          "id": "postprocess_output",
+          "label": "Postprocess Model Output",
+          "type": "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"}
       ]
     }
+
+    Returns:
+        str: The filepath to the generated PNG image file.
     """
     try:
         if not json_input.strip():