Update app.py

app.py

@@ -22,7 +22,7 @@ from langchain_core.utils.utils import secret_from_env
 from io import BytesIO
 from pathlib import Path
 import os
-from utils.block_relation_builder import block_builder#, variable_adder_main
+from utils.block_relation_builder import block_builder, separate_scripts, transform_logic_to_action_flow#, variable_adder_main
 from langchain.chat_models import ChatOpenAI
 from langchain_openai import ChatOpenAI
 from pydantic import Field, SecretStr
@@ -1009,6 +1009,23 @@ 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
+
+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", "")
+    try:
+        refined_logic_data["pseudocode"] = separate_scripts(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
+        state["action_plan"] = transform_logic_to_action_flow(state["pseudo_code"])
+        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):
     """
@@ -2353,16 +2370,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)
@@ -2384,13 +2402,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