Update app.py

app.py

@@ -489,6 +489,56 @@ def clean_base64_for_model(raw_b64):
     # 6. Return with the correct data URI prefix
     return f"data:image/png;base64,{clean_b64}"
 
+def format_scratch_pseudo_code(code_string):
+    """
+    Parses and formats Scratch pseudo-code with correct indentation,
+    specifically handling if/else/end structures correctly.
+
+    Args:
+        code_string (str): A string containing Scratch pseudo-code with
+                         potentially inconsistent indentation.
+
+    Returns:
+        str: The correctly formatted and indented pseudo-code string.
+    """
+    lines = code_string.strip().split('\n')
+    formatted_lines = []
+    indent_level = 0
+    
+    # Keywords that increase indentation for the NEXT line
+    indent_keywords = ['when', 'forever', 'if', 'repeat', 'else']
+    
+    # Keywords that decrease indentation for the CURRENT line
+    unindent_keywords = ['end', 'else']
+
+    for line in lines:
+        stripped_line = line.strip()
+        if not stripped_line:
+            continue
+
+        # Check for keywords that should un-indent the current line
+        if any(keyword in stripped_line for keyword in unindent_keywords):
+            # Special case for 'else': it should align with its 'if'
+            if 'else' in stripped_line:
+                # Decrease indentation for 'else' and its following lines
+                indentation = '    ' * (indent_level -1)
+                formatted_lines.append(indentation + stripped_line)
+                continue
+            
+            # For 'end', decrease the level before formatting
+            indent_level = max(0, indent_level - 1)
+        
+        indentation = '    ' * indent_level
+        formatted_lines.append(indentation + stripped_line)
+
+        # Check for keywords that should indent the next line
+        if any(keyword in stripped_line for keyword in indent_keywords):
+            # 'else' both un-indents and indents, so the level remains the same for the next block
+            if 'else' not in stripped_line:
+                indent_level += 1
+
+    return '\n'.join(formatted_lines)
+
 
 # Node 1: Logic updating if any issue here
 def pseudo_generator_node(state: GameState):
@@ -1636,9 +1686,10 @@ def overall_block_builder_node_2(state: dict):
             for plan_entry in sprite_actions_data.get("plans", []):
                 logic_sequence = str(plan_entry["logic"])
                 opcode_counts = plan_entry.get("opcode_counts", {})
-
+                refined_indent_logic = format_scratch_pseudo_code(logic_sequence)
+                print(f"\n--------------------------- refined indent logic: {refined_indent_logic}-------------------------------\n")
                 try:
-                    generated_blocks = block_builder(opcode_counts, logic_sequence)
+                    generated_blocks = block_builder(opcode_counts, refined_indent_logic)
                     
                     # Ensure generated_blocks is a dictionary
                     if not isinstance(generated_blocks, dict):