Update app_logic.py

app_logic.py

@@ -71,34 +71,55 @@ def process_commented_markdown(lines):
 
 # --- `parse_markdown` (Unchanged from previous corrected version) ---
 def parse_markdown(markdown_input):
-    space_info = {"repo_name_md": "", "owner_md": "", "files": []}
-    current_file_path = None; current_file_content_lines = []
-    in_file_definition = False; in_code_block = False
-    lines = markdown_input.strip().split("\n")
-    lines = process_commented_markdown(lines)
-    #print(lines)
-    for line_content_orig in lines:
-        line_content_stripped = line_content_orig.strip()
-        if line_content_stripped.startswith("### File:"):
-            if current_file_path and in_file_definition:
-                space_info["files"].append({"path": current_file_path, "content": "\n".join(current_file_content_lines)})
-            current_file_path = line_content_stripped.replace("### File:", "").strip()
-            current_file_content_lines = []
-            in_file_definition = True; in_code_block = False
-            continue
-        if not in_file_definition:
-            if line_content_stripped.startswith("# Space:"):
-                full_space_name_md = line_content_stripped.replace("# Space:", "").strip()
-                if "/" in full_space_name_md: space_info["owner_md"], space_info["repo_name_md"] = full_space_name_md.split("/", 1)
-                else: space_info["repo_name_md"] = full_space_name_md
+    """Parse markdown input to extract space details and file structure."""
+    # Preprocess to remove comments
+    cleaned_markdown = process_commented_markdown(markdown_input)
+    space_info = {"repo_name": "", "owner": "", "files": []}
+    current_file = None
+    file_content = []
+    in_file_content = False
+    in_code_block = False
+
+    lines = cleaned_markdown.strip().split("\n")
+    for line in lines:
+        line = line.strip()
+        if line.startswith("# Space:"):
+            space_info["repo_name"] = line.replace("# Space:", "").strip()
+            if "/" in space_info["repo_name"]:
+                space_info["owner"], space_info["repo_name"] = space_info["repo_name"].split("/", 1)
+        elif line.startswith("## File Structure"):
             continue
-        if line_content_stripped.startswith("```"):
+        elif line.startswith("📄") or line.startswith("📁"):
+            if current_file and file_content and in_file_content:
+                space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
+                file_content = []
+            current_file = line[2:].strip()
+            in_file_content = False
+            in_code_block = False
+        elif line.startswith("### File:"):
+            if current_file and file_content and in_file_content:
+                space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
+                file_content = []
+            current_file = line.replace("### File:", "").strip()
+            in_file_content = True
+            in_code_block = False
+        elif in_file_content and line.startswith("```"):
             in_code_block = not in_code_block
-            continue
-        current_file_content_lines.append(line_content_orig)
-    if current_file_path and in_file_definition:
-        space_info["files"].append({"path": current_file_path, "content": "\n".join(current_file_content_lines)})
-    space_info["files"] = [f for f in space_info["files"] if f.get("path")]
+            if in_code_block:
+                # First line after ``` is the language
+                file_content = []
+            else:
+                # End of code block, store file content
+                if current_file and file_content:
+                    space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
+                file_content = []
+                in_file_content = False
+        elif in_file_content and in_code_block:
+            file_content.append(line)
+
+    if current_file and file_content and in_file_content:
+        space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})
+
     return space_info
 
 # --- `_determine_repo_id` (Unchanged) ---