Update app_logic.py

app_logic.py

@@ -61,16 +61,33 @@ def load_token_from_image_and_set_env(image_pil_object: Image.Image, password: s
     return "\n".join(status_messages_display)
 
 
+
 def process_commented_markdown(commented_input):
-    """Process a commented markdown string by stripping '# ' from each line if it matches the expected format."""
-    # Check for '# # Space:' at the start or within the input
-    if re.search(r'^\s*#+\s*Space:', commented_input, re.MULTILINE):
-        lines = commented_input.strip().split("\n")
-        # Strip '# ' from each line
-        cleaned_lines = [line.lstrip("# ") for line in lines]
-        return "\n".join(cleaned_lines)
-    return commented_input
+    """Process a commented markdown string by stripping '# ' from each line if it contains '# # Space:'."""
+    # Check for '# # Space:' or '# Space:' in the input
+    if not re.search(r'^\s*#+\s*Space:', commented_input, re.MULTILINE):
+        return commented_input
+
+    lines = commented_input.strip().split("\n")
+    cleaned_lines = []
+    in_code_block = False
+
+    for line in lines:
+        # Detect code block boundaries
+        if line.strip().startswith("# ```"):
+            in_code_block = not in_code_block
+
+        # Strip '# ' from the start of each line
+        cleaned_line = line.lstrip("# ")
+
+        # If in a code block and the line still starts with '# ', remove it
+        # This handles Gemini's over-commenting within code blocks
+        if in_code_block and cleaned_line.startswith("# "):
+            cleaned_line = cleaned_line[2:]
+
+        cleaned_lines.append(cleaned_line)
 
+    return "\n".join(cleaned_lines)
 
 # --- `parse_markdown` (Unchanged from previous corrected version) ---
 def parse_markdown(markdown_input):