Update app_gradio.py

app_gradio.py

@@ -347,7 +347,7 @@ def execute_main_task(history, state, budget):
             # Increment estimated cost per node execution
             estimated_cost += cost_per_loop
             
-            # Replace the problematic merge logic with this corrected version:
+            # Replace the state merging logic around line 260-280 with:
             for key, value in node_output.items():
                 if key == 'execution_path':
                     # Execution path should accumulate
@@ -356,8 +356,10 @@ def execute_main_task(history, state, budget):
                         aggregated_state[key] = current_path + value
                     else:
                         aggregated_state[key] = current_path + [value]
-                elif key in ['max_loops', 'rework_cycles', 'current_cost', 'approved', 'budget_exceeded']:
-                    # These should be single values, not lists - take the latest
+                elif key in ['max_loops', 'rework_cycles', 'current_cost', 'approved', 
+                             'budget_exceeded', 'pragmatistReport', 'governanceReport', 
+                             'complianceReport', 'observerReport', 'knowledgeInsights']:
+                    # These should be single values, not accumulate
                     aggregated_state[key] = value
                 elif key == 'status_updates':
                     # Status updates should accumulate as a list
@@ -366,10 +368,13 @@ def execute_main_task(history, state, budget):
                         aggregated_state[key] = current + value
                     else:
                         aggregated_state[key] = current + [value]
+                elif key == 'pmPlan' or key == 'experimentResults':
+                    # These should be replaced, not accumulated
+                    aggregated_state[key] = value
                 else:
                     # Default: replace with new value
                     aggregated_state[key] = value
-            
+                        
             # Update current cost in state
             aggregated_state["current_cost"] = estimated_cost