fix: Remove markdown code fences in HTML conversion and improve solution link generation

- Add code fence removal in convert_to_html.py to prevent HTML conversion errors
- Improve _build_problems_lookup to better match problem IDs from various sources
- Enhance _add_github_solution_links to correctly add Solution links when solution files exist
- Solution links now properly generated as: [LeetCode X](url) | [Solution](github_url)

Author: lufftw

tools/ai-markmap-agent/convert_to_html.py

@@ -135,6 +135,18 @@ def convert_file_to_html(
         title = f"{input_path.stem.replace('_', ' ').title()} - NeetCode Mind Maps"
 
     markdown_content = input_path.read_text(encoding="utf-8")
+    
+    # Remove markdown code fence if present (LLM sometimes wraps output)
+    markdown_content = markdown_content.strip()
+    if markdown_content.startswith("```markdown"):
+        markdown_content = markdown_content[len("```markdown"):].strip()
+    if markdown_content.startswith("```md"):
+        markdown_content = markdown_content[len("```md"):].strip()
+    if markdown_content.startswith("```"):
+        markdown_content = markdown_content[3:].strip()
+    if markdown_content.endswith("```"):
+        markdown_content = markdown_content[:-3].strip()
+    
     converter = StandaloneHTMLConverter(template_path=template_path)
     html_content = converter.convert(markdown_content, title=title)
     output_path.write_text(html_content, encoding="utf-8")

tools/ai-markmap-agent/src/post_processing.py

@@ -64,16 +64,35 @@ def _build_problems_lookup(self, problems: dict[str, Any]) -> dict[str, dict]:
 
         for key, value in problems.items():
             if isinstance(value, dict):
-                problem_id = value.get("id", key)
-                # Normalize ID to 4 digits
-                if isinstance(problem_id, str) and problem_id.isdigit():
-                    problem_id = problem_id.zfill(4)
-                lookup[problem_id] = value
-                # Also store without leading zeros for flexibility
-                try:
-                    lookup[str(int(problem_id))] = value
-                except (ValueError, TypeError):
-                    pass
+                # Try to get ID from various sources
+                problem_id = value.get("id") or value.get("leetcode_id")
+                
+                # If no ID found, try to extract from key (slug format: "0079_word_search")
+                if not problem_id:
+                    # Extract ID from slug if it starts with digits
+                    match = re.match(r'^(\d+)_', key)
+                    if match:
+                        problem_id = match.group(1)
+                
+                # If still no ID, use key as fallback (but this is less reliable)
+                if not problem_id:
+                    problem_id = key
+                
+                # Normalize ID to string and ensure 4 digits
+                if isinstance(problem_id, int):
+                    problem_id = str(problem_id)
+                elif not isinstance(problem_id, str):
+                    problem_id = str(problem_id)
+                
+                # Store with 4-digit format
+                if problem_id.isdigit():
+                    normalized_id = problem_id.zfill(4)
+                    lookup[normalized_id] = value
+                    # Also store without leading zeros for flexibility
+                    lookup[problem_id] = value
+                    # Also store as integer string if different
+                    if normalized_id != problem_id:
+                        lookup[str(int(problem_id))] = value
 
         return lookup
 
@@ -283,7 +302,12 @@ def add_solution_link(match: re.Match) -> str:
             problem_id = id_match.group(1)
 
             # Look up problem in our data
-            problem = self.problems_lookup.get(problem_id.zfill(4)) or self.problems_lookup.get(problem_id)
+            # Try both 4-digit format and numeric format
+            problem = (
+                self.problems_lookup.get(problem_id.zfill(4)) or 
+                self.problems_lookup.get(problem_id) or
+                self.problems_lookup.get(str(int(problem_id)).zfill(4)) if problem_id.isdigit() else None
+            )
             if not problem:
                 return full_text