Update app.py

app.py

@@ -2044,7 +2044,7 @@ def plan_opcode_counter_node(state: Dict[str, Any]) -> Dict[str, Any]:
                 # Invoke the main agent for logic refinement and relationship identification
                 response = agent.invoke({"messages": [{"role": "user", "content": refinement_prompt}]})
                 llm_output = response["messages"][-1].content
-                print(f"[RAW RESPONSE OPCODE COUTER LOGIC]: {response}")
+                #print(f"[RAW RESPONSE OPCODE COUTER LOGIC]: {response}")
                 llm_json = extract_json_from_llm_response(llm_output)                
                 logger.info(f"Successfully analyse the opcode requirement for {sprite} - {event}.")
 
@@ -3665,12 +3665,12 @@ workflow.add_node("time_delay_3", delay_for_tpm_node) # this is a temporary node
 workflow.add_node("time_delay_4", delay_for_tpm_node) # this is a temporary node to handle TPM issues
 workflow.add_node("time_delay_5", delay_for_tpm_node) # this is a temporary node to handle TPM issues
 workflow.add_node("time_delay_6", delay_for_tpm_node) # this is a temporary node to handle TPM issues
-workflow.add_node("time_delay_7", delay_for_tpm_node) # this is a temporary node to handle TPM issues
+#workflow.add_node("time_delay_7", delay_for_tpm_node) # this is a temporary node to handle TPM issues
 workflow.add_node("initial_plan_build", overall_planner_node) # High-level planning node
 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("opcode_counter", plan_opcode_counter_node)
 workflow.add_node("relation_builder", block_relationship_node)
 # Verifies the high-level plan
 workflow.add_node("block_builder", overall_block_builder_node) # Builds blocks from a plan
@@ -3698,24 +3698,22 @@ def decide_next_step_after_plan_verification(state: GameState):
         return "refined_planner"
     else:
         # If the plan is good, proceed to building blocks from this plan
-        return "opcode_counter"#"block_builder"
+        return "relation_builder"#"block_builder"
 
 workflow.add_conditional_edges(
     "plan_verifier",
     decide_next_step_after_plan_verification,
     {
         "refined_planner": "refined_planner",  # Path if plan needs refinement
-        "opcode_counter": "opcode_counter"#"block_builder": "block_builder"       # Path if plan is approved, proceeds to block building
+        "relation_builder": "relation_builder"#"block_builder": "block_builder"       # Path if plan is approved, proceeds to block building
     }
 )
 # --- CRITICAL CHANGE FOR THE PLAN REFINEMENT LOOP ---
 # After refining the plan, it should go back to plan_verifier for re-verification.
 workflow.add_edge("refined_planner", "time_delay_5")
 workflow.add_edge("time_delay_5", "plan_verifier") # This closes the loop for plan refinement and re-verification.
-workflow.add_edge("opcode_counter", "time_delay_6")
-workflow.add_edge("time_delay_6", "relation_builder")
-workflow.add_edge("relation_builder", "time_delay_7")
-workflow.add_edge("time_delay_7", "block_builder")
+workflow.add_edge("relation_builder", "time_delay_6")
+workflow.add_edge("time_delay_6", "block_builder")
 workflow.add_edge("block_builder", END)
 # Note: The original code had workflow.add_edge("time_delay", "block_builder") here,
 # but after refined_planner -> time_delay -> plan_verifier, the decision is made by plan_verifier.