Update app_gradio.py

app_gradio.py

@@ -405,19 +405,50 @@ def execute_main_task(history, state, budget):
                     aggregated_state["status_update"] = str(node_output)
                 continue
             
+            # --- Robust State Aggregation (namespace status_update to avoid concurrent writes) ---
+            # Prevent multiple agents in the same step from clobbering the same top-level key
             for key, value in node_output.items():
+                # Normalize execution_path (keep existing logic)
                 if key == 'execution_path':
                     current_path = aggregated_state.get(key, [])
                     if isinstance(current_path, list):
-                         # Ensure we are always extending a list
                         new_items = value if isinstance(value, list) else [value]
                         aggregated_state[key] = current_path + new_items
                     else:
-                        # Recover from corrupted state
                         aggregated_state[key] = list(current_path) + (value if isinstance(value, list) else [value])
+                    continue
+
+                # Namespace status_update and other commonly-colliding keys from agents
+                if key == "status_update":
+                    # rename to a node-scoped key so LangGraph doesn't see concurrent writes to the same key
+                    node_status_key = f"status_update__{node_name}"
+                    # keep the node-scoped status
+                    aggregated_state[node_status_key] = value
+                    # also set a global 'status_update' if not already present, prefer explicit global writes
+                    if "status_update" not in aggregated_state or not aggregated_state.get("status_update"):
+                        aggregated_state["status_update"] = value
+                    else:
+                        # if there is already a global status, prefer the most recent node's status (override)
+                        aggregated_state["status_update"] = value
+                    continue
+
+                # If key already exists and is not a list, convert to list and append (safe merging)
+                if key in aggregated_state and key != "execution_path":
+                    existing = aggregated_state[key]
+                    # if both are lists, concatenate; otherwise turn into list of values
+                    if isinstance(existing, list) and isinstance(value, list):
+                        aggregated_state[key] = existing + value
+                    elif isinstance(existing, list):
+                        aggregated_state[key] = existing + [value]
+                    elif isinstance(value, list):
+                        aggregated_state[key] = [existing] + value
+                    else:
+                        # keep both values in list to avoid overwrite semantics
+                        aggregated_state[key] = [existing, value]
                 else:
                     aggregated_state[key] = value
 
+
             # --- Real-time Budget Check ---
             current_cost = cost_tracker.get_total_cost()
             aggregated_state["current_cost"] = current_cost