update pages

document_generator_v3.py

@@ -402,24 +402,6 @@ class MarkdownConverter:
     def slugify(text: str) -> str:
         return re.sub(r'\W+', '-', text.lower())
 
-    @classmethod
-    def generate_toc(cls, sections: List[Dict]) -> str:
-        toc = "<div style='page-break-before: always;'></div>\n\n"
-        toc += "<h2 style='color: #2c3e50; text-align: center;'>Table of Contents</h2>\n\n"
-        toc += "<nav style='background-color: #f8f9fa; padding: 20px; border-radius: 5px; line-height: 1.6;'>\n\n"
-        for section in sections:
-            section_number = section['PageNumber']
-            section_title = section['Title']
-            toc += f"<p><a href='#{cls.slugify(section_title)}' style='color: #3498db; text-decoration: none;'>{section_number}. {section_title}</a></p>\n\n"
-            
-            for subsection in section.get('Subsections', []):
-                subsection_number = subsection['PageNumber']
-                subsection_title = subsection['Title']
-                toc += f"<p style='margin-left: 20px;'><a href='#{cls.slugify(subsection_title)}' style='color: #2980b9; text-decoration: none;'>{subsection_number} {subsection_title}</a></p>\n\n"
-        
-        toc += "</nav>\n\n"
-        return toc
-
     @classmethod
     def convert_to_markdown(cls, document: Dict) -> str:
         markdown = "<div style='text-align: center; padding-top: 33vh;'>\n\n"
@@ -428,23 +410,33 @@ class MarkdownConverter:
         markdown += f"<p style='color: #95a5a6;'>Version {document['Version']} | {document['Date']}</p>\n\n"
         markdown += "</div>\n\n"
         
-        markdown += cls.generate_toc(document['Sections'])
+        # Generate Table of Contents
+        markdown += "<div style='page-break-before: always;'></div>\n\n"
+        markdown += "<h2 style='color: #2c3e50; text-align: center;'>Table of Contents</h2>\n\n"
+        markdown += "<nav style='background-color: #f8f9fa; padding: 20px; border-radius: 5px; line-height: 1.6;'>\n\n"
         
+        for section in document['Pages']:
+            section_number = section['PageNumber']
+            section_title = section['Title']
+            markdown += f"<p><a href='#{cls.slugify(section_title)}' style='color: #3498db; text-decoration: none;'>{section_number}. {section_title}</a></p>\n\n"
+            
+            for subsection in section.get('Subsections', []):
+                subsection_number = subsection['PageNumber']
+                subsection_title = subsection['Title']
+                markdown += f"<p style='margin-left: 20px;'><a href='#{cls.slugify(subsection_title)}' style='color: #2980b9; text-decoration: none;'>{subsection_number} {subsection_title}</a></p>\n\n"
+        
+        markdown += "</nav>\n\n"
+        
+        # Generate Content
         markdown += "<div style='max-width: 800px; margin: 0 auto; font-family: \"Segoe UI\", Arial, sans-serif; line-height: 1.6;'>\n\n"
         
-        for section in document['Sections']:
+        for section in document['Pages']:
             markdown += "<div style='page-break-before: always;'></div>\n\n"
             section_number = section['PageNumber']
             section_title = section['Title']
             markdown += f"<h2 id='{cls.slugify(section_title)}' style='color: #2c3e50; border-bottom: 1px solid #bdc3c7; padding-bottom: 5px;'>{section_number}. {section_title}</h2>\n\n"
             markdown += f"<div style='color: #34495e; margin-bottom: 20px;'>\n\n{section['Content']}\n\n</div>\n\n"
             
-            # for subsection in section.get('Subsections', []):
-            #     subsection_number = subsection['PageNumber']
-            #     subsection_title = subsection['Title']
-            #     markdown += f"<h3 id='{cls.slugify(subsection_title)}' style='color: #34495e;'>{subsection_number} {subsection_title}</h3>\n\n"
-            #     markdown += f"<div style='color: #34495e; margin-bottom: 20px;'>\n\n{subsection['Content']}\n\n</div>\n\n"
-        
         markdown += "</div>"
         return markdown