Update app.py

app.py

@@ -1458,6 +1458,87 @@ Example output:
     logger.info("=== OPCODE COUTER LOGIC completed ===")
     return state
 
+# Node 10:Function based block builder node
+# def overall_block_builder_node_2(state: dict):
+#     logger.info("--- Running OverallBlockBuilderNode ---")
+
+#     project_json = state["project_json"]
+#     targets = project_json["targets"]
+#     # --- Sprite and Stage Target Mapping ---
+#     sprite_map = {target["name"]: target for target in targets if not target["isStage"]}
+#     stage_target = next((target for target in targets if target["isStage"]), None)
+#     if stage_target:
+#         sprite_map[stage_target["name"]] = stage_target
+
+#     action_plan = state.get("action_plan", {})
+#     print("[Overall Action Plan received at the block generator]:", json.dumps(action_plan, indent=2))
+#     if not action_plan:
+#         logger.warning("No action plan found in state. Skipping OverallBlockBuilderNode.")
+#         return state
+
+#     # Initialize offsets for script placement on the Scratch canvas
+#     script_y_offset = {}
+#     script_x_offset_per_sprite = {name: 0 for name in sprite_map.keys()}
+
+#     # This handles potential variations in the action_plan structure.
+#     if action_plan.get("action_overall_flow", {}) == {}:
+#         plan_data = action_plan.items()
+#     else:
+#         plan_data = action_plan.get("action_overall_flow", {}).items()
+
+#     # --- Extract global project context for LLM ---
+#     all_sprite_names = list(sprite_map.keys())
+#     all_variable_names = {}
+#     all_list_names = {}
+#     all_broadcast_messages = {}
+
+#     for target in targets:
+#         for var_id, var_info in target.get("variables", {}).items():
+#             all_variable_names[var_info[0]] = var_id  # Store name -> ID mapping (e.g., "myVariable": "myVarId123")
+#         for list_id, list_info in target.get("lists", {}).items():
+#             all_list_names[list_info[0]] = list_id  # Store name -> ID mapping
+#         for broadcast_id, broadcast_name in target.get("broadcasts", {}).items():
+#             all_broadcast_messages[broadcast_name] = broadcast_id  # Store name -> ID mapping
+
+#     # --- Process each sprite's action plan ---
+#     for sprite_name, sprite_actions_data in plan_data:
+#         if sprite_name in sprite_map:
+#             current_sprite_target = sprite_map[sprite_name]
+#             if "blocks" not in current_sprite_target:
+#                 current_sprite_target["blocks"] = {}
+
+#             if sprite_name not in script_y_offset:
+#                 script_y_offset[sprite_name] = 0
+
+#             for plan_entry in sprite_actions_data.get("plans", []):
+#                 logic_sequence = str(plan_entry["logic"])
+#                 opcode_counts = plan_entry.get("opcode_counts", {})
+
+#                 try:
+#                     generated_blocks=block_builder(opcode_counts,logic_sequence)                
+#                     if "blocks" in generated_blocks and isinstance(generated_blocks["blocks"], dict):
+#                         logger.warning(f"LLM returned nested 'blocks' key for {sprite_name}. Unwrapping.")
+#                         generated_blocks = generated_blocks["blocks"]
+
+#                     # Update block positions for top-level script
+#                     for block_id, block_data in generated_blocks.items():
+#                         if block_data.get("topLevel"):
+#                             block_data["x"] = script_x_offset_per_sprite.get(sprite_name, 0)
+#                             block_data["y"] = script_y_offset[sprite_name]
+#                             script_y_offset[sprite_name] += 150 # Increment for next script
+
+#                     current_sprite_target["blocks"].update(generated_blocks)
+#                     print(f"[current_sprite_target block updated]: {current_sprite_target['blocks']}")
+#                     state["iteration_count"] = 0
+#                     logger.info(f"Action blocks added for sprite '{sprite_name}' by OverallBlockBuilderNode.")
+#                 except Exception as e:
+#                     logger.error(f"Error generating blocks for sprite '{sprite_name}': {e}")
+                    
+#     state["project_json"] = project_json
+#     # with open("debug_state.json", "w", encoding="utf-8") as f:
+#     #     json.dump(state, f, indent=2, ensure_ascii=False)
+    
+#     return state
 # Node 10:Function based block builder node
 def overall_block_builder_node_2(state: dict):
     logger.info("--- Running OverallBlockBuilderNode ---")
@@ -1515,7 +1596,13 @@ def overall_block_builder_node_2(state: dict):
                 opcode_counts = plan_entry.get("opcode_counts", {})
 
                 try:
-                    generated_blocks=block_builder(opcode_counts,logic_sequence)                
+                    generated_blocks = block_builder(opcode_counts, logic_sequence)
+                    
+                    # Ensure generated_blocks is a dictionary
+                    if not isinstance(generated_blocks, dict):
+                        logger.error(f"block_builder for sprite '{sprite_name}' returned non-dict type: {type(generated_blocks)}. Skipping block update.")
+                        continue # Skip to next plan_entry if output is not a dictionary
+
                     if "blocks" in generated_blocks and isinstance(generated_blocks["blocks"], dict):
                         logger.warning(f"LLM returned nested 'blocks' key for {sprite_name}. Unwrapping.")
                         generated_blocks = generated_blocks["blocks"]
@@ -1533,6 +1620,9 @@ def overall_block_builder_node_2(state: dict):
                     logger.info(f"Action blocks added for sprite '{sprite_name}' by OverallBlockBuilderNode.")
                 except Exception as e:
                     logger.error(f"Error generating blocks for sprite '{sprite_name}': {e}")
+                    # Consider adding more specific error handling here if a malformed output
+                    # from block_builder should cause a specific state change, but generally
+                    # avoid nulling the entire project_json.
                     
     state["project_json"] = project_json
     # with open("debug_state.json", "w", encoding="utf-8") as f: