Update app_logic.py

app_logic.py

@@ -75,7 +75,7 @@ def process_commented_markdown(commented_input):
     in_code_block = False
 
     for line in lines:
-        # Detect code block boundaries (handle both '# ```' and '```')
+        # Detect code block boundaries (handle '# ```' or '```')
         if line.strip() in ("# ```", "```"):
             in_code_block = not in_code_block
             # Preserve backticks as-is (strip '# ' if present)
@@ -86,7 +86,7 @@ def process_commented_markdown(commented_input):
         # Strip '# ' from lines that have it
         cleaned_line = line.lstrip("# ")
 
-        # In code blocks, remove additional '# ' if present (handles Gemini's over-commenting)
+        # In code blocks, remove additional '# ' if present (handles over-commenting)
         if in_code_block and cleaned_line.startswith("# "):
             cleaned_line = cleaned_line[2:]
 
@@ -94,7 +94,6 @@ def process_commented_markdown(commented_input):
 
     return "\n".join(cleaned_lines)
 
-# --- `parse_markdown` (Unchanged from previous corrected version) ---
 def parse_markdown(markdown_input):
     """Parse markdown input to extract space details and file structure."""
     # Preprocess to remove comments
@@ -104,44 +103,82 @@ def parse_markdown(markdown_input):
     file_content = []
     in_file_content = False
     in_code_block = False
+    language = None
 
     lines = cleaned_markdown.strip().split("\n")
-    for line in lines:
-        line = line.strip()
+    i = 0
+    while i < len(lines):
+        line = lines[i].strip()
+
+        # Extract space name
         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"):
+            i += 1
+            continue
+
+        # Skip file structure header
+        if line.startswith("## File Structure"):
+            i += 1
             continue
-        elif line.startswith("📄") or line.startswith("📁"):
+
+        # Parse file structure
+        if 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:"):
+            language = None
+            i += 1
+            continue
+
+        # Start of a new file content section
+        if 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
-            if in_code_block:
-                # First line after ``` is the language
-                file_content = []
+            language = None
+            i += 1
+
+            # Handle content (code block or plain text)
+            if i < len(lines) and lines[i].strip() == "```":
+                in_code_block = True
+                i += 1
+                # Check for language identifier
+                if i < len(lines) and lines[i].strip() and not lines[i].strip().startswith("```"):
+                    language = lines[i].strip()
+                    i += 1
+                # Collect content until closing backtick
+                while i < len(lines) and lines[i].strip() != "```":
+                    file_content.append(lines[i].rstrip())
+                    i += 1
+                if i < len(lines) and lines[i].strip() == "```":
+                    i += 1
+                in_code_block = False
+                in_file_content = False
             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 = []
+                # Handle non-code-block content (e.g., binary file)
+                while i < len(lines) and not (
+                    lines[i].strip().startswith("### File:")
+                    or lines[i].strip().startswith("#")
+                    or lines[i].strip().startswith("##")
+                    or lines[i].strip() == "```"
+                ):
+                    if lines[i].strip():
+                        file_content.append(lines[i].rstrip())
+                    i += 1
                 in_file_content = False
-        elif in_file_content and in_code_block:
-            file_content.append(line)
 
+        else:
+            i += 1
+
+    # Save the last file if present
     if current_file and file_content and in_file_content:
         space_info["files"].append({"path": current_file, "content": "\n".join(file_content)})