Update app.py

app.py

@@ -1156,6 +1156,54 @@ If you find any "Code-Blocks" then,
     print(f"[OVREALL LISTS OF LOGICS]: {Data}")
     logger.info("Plan refinement and block relation analysis completed for all plans.")
     return state
+
+# Node: Node Optimizer node
+def node_optimizer(state: GameState):
+    logger.info("--- Running Node Optimizer Node ---")
+    project_json = state["project_json"]
+    raw = state.get("pseudo_code", {})
+    refined_logic_data = raw.get("refined_logic", {})
+    sprite_name = refined_logic_data.get("name_variable", "<unknown>")
+    pseudo      = refined_logic_data.get("pseudocode", "")
+    sprite_name = {}
+    project_json_targets = state.get("project_json", {}).get("targets", [])
+    for target in project_json_targets:
+        sprite_name[target["name"]] = target["name"]
+    action_flow = state.get("action_plan", {}) 
+    
+    try:
+        refined_logic_data["pseudocode"] = separate_scripts(str(pseudo))
+        # Step 4: If you want to update the `state` dictionary with the new refined_logic_data
+        state["pseudo_code"]["refined_logic"] = refined_logic_data
+        print(f"[The pseudo_code generated here]: { state['pseudo_code']}")
+        state["action_plan"] = transform_logic_to_action_flow(state["pseudo_code"])
+        print(f"[The action plan generated here]: { state['action_plan']}")
+ 
+        action_flow = state.get("action_plan", {})     
+        if action_flow.get("action_overall_flow", {}) == {}:
+            plan_data = action_flow.items()
+        else:
+            plan_data = action_flow.get("action_overall_flow", {}).items()
+            
+        refined_flow: Dict[str, Any] = {}
+        for sprite, sprite_data in plan_data:
+            refined_plans = []
+            for plan in sprite_data.get("plans", []):
+                logic = plan.get("logic", "")
+                plan["opcode_counts"]= analyze_opcode_counts(str(logic))
+                refined_plans.append(plan)
+            refined_flow[sprite] = {
+                "description": sprite_data.get("description", ""),
+                "plans": refined_plans
+            }
+        if refined_flow:
+            state["action_plan"] = refined_flow  
+            logger.info("Node Optimization completed.")
+        
+        return state
+    except Exception as e:
+        logger.error(f"Error in Node Optimizer Node: {e}")
+
 # Node 2: planner node
 def overall_planner_node(state: GameState):
     """
@@ -2505,16 +2553,17 @@ def delay_for_tpm_node(state: GameState):
 workflow = StateGraph(GameState)
 
 # Add all nodes to the workflow
-workflow.add_node("time_delay_1", delay_for_tpm_node)
-workflow.add_node("time_delay_2", delay_for_tpm_node)
-workflow.add_node("time_delay_3", delay_for_tpm_node)
+# workflow.add_node("time_delay_1", delay_for_tpm_node)
+# workflow.add_node("time_delay_2", delay_for_tpm_node)
+# workflow.add_node("time_delay_3", delay_for_tpm_node)
 # workflow.add_node("time_delay_4", delay_for_tpm_node)
 workflow.add_node("pseudo_generator", pseudo_generator_node)
-workflow.add_node("plan_generator", overall_planner_node)
+#workflow.add_node("plan_generator", overall_planner_node)
+workflow.add_node("Node_optimizer", node_optimizer)
 #workflow.add_node("logic_alignment", plan_logic_aligner_node)
 #workflow.add_node("plan_verifier", plan_verification_node)
-workflow.add_node("refined_planner", refined_planner_node) # Refines the action plan
-workflow.add_node("opcode_counter", plan_opcode_counter_node)
+#workflow.add_node("refined_planner", refined_planner_node) # Refines the action plan
+#workflow.add_node("opcode_counter", plan_opcode_counter_node)
 workflow.add_node("block_builder", overall_block_builder_node_2)
 workflow.add_node("variable_initializer", variable_adder_node)
 workflow.add_node("page_processed", processed_page_node)
@@ -2536,13 +2585,14 @@ workflow.add_conditional_edges(
     }
 )
 # Main chain
-workflow.add_edge("pseudo_generator", "time_delay_1")
-workflow.add_edge("time_delay_1", "plan_generator")
-workflow.add_edge("plan_generator", "time_delay_2")
-workflow.add_edge("time_delay_2", "refined_planner")
-workflow.add_edge("refined_planner", "time_delay_3")
-workflow.add_edge("time_delay_3", "opcode_counter")
-workflow.add_edge("opcode_counter", "block_builder")
+# workflow.add_edge("pseudo_generator", "time_delay_1")
+workflow.add_edge("pseudo_generator", "Node_optimizer")
+# workflow.add_edge("time_delay_1", "plan_generator")
+# workflow.add_edge("plan_generator", "time_delay_2")
+# workflow.add_edge("time_delay_2", "refined_planner")
+# workflow.add_edge("refined_planner", "time_delay_3")
+# workflow.add_edge("time_delay_3", "opcode_counter")
+workflow.add_edge("Node_optimizer", "block_builder")
 workflow.add_edge("block_builder", "variable_initializer")
 
 # After last node, check again