Update app_logic.py

app_logic.py

@@ -3,6 +3,8 @@ import re
 import tempfile
 import shutil
 import git
+import re
+
 from huggingface_hub import (
     create_repo,
     upload_folder,
@@ -58,15 +60,16 @@ def load_token_from_image_and_set_env(image_pil_object: Image.Image, password: s
     except Exception as e: status_messages_display.append(f"**Unexpected decoding error:** {str(e)}")
     return "\n".join(status_messages_display)
 
-def process_commented_markdown(lines):
 
-    """Process a commented markdown string by stripping '# ' from each line if any line starts with '# # Space:'."""
-    if any(line.strip().startswith(("# # Space:", "# Space:")) for line in lines):
-        print("FOUND ALT MARKDOWN")
+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 cleaned_lines
-    
-    return lines
+        return "\n".join(cleaned_lines)
+    return commented_input
 
 
 # --- `parse_markdown` (Unchanged from previous corrected version) ---