go

pdf_report_generation_service.py

@@ -1,6 +1,6 @@
 """
-Professional Q4 Report Generator - Complete Working Version
-Generates executive-quality PDF reports with proper image rendering
+Professional Q4 Report Generator - Enhanced Markdown Parsing
+Generates executive-quality PDF reports with proper list rendering
 """
 import markdown
 from weasyprint import HTML, CSS
@@ -22,7 +22,7 @@ logging.basicConfig(
 logger = logging.getLogger(__name__)
 
 # ---------------------------------------------------------
-# EXECUTIVE Q4 REPORT THEME
+# EXECUTIVE Q4 REPORT THEME - ENHANCED LIST STYLING
 # ---------------------------------------------------------
 CSS_STYLE = """
 @page {
@@ -253,20 +253,72 @@ blockquote {
     page-break-after: always;
 }
 
-ul, ol {
-    margin: 14px 0;
-    padding-left: 28px;
+/* ENHANCED LIST STYLING - ENSURES PROPER BULLET RENDERING */
+ul {
+    margin: 16px 0;
+    padding-left: 35px;
+    list-style-type: disc;
+    list-style-position: outside;
+}
+
+ol {
+    margin: 16px 0;
+    padding-left: 35px;
+    list-style-type: decimal;
+    list-style-position: outside;
 }
 
 li {
-    margin: 8px 0;
+    margin: 6px 0;
     line-height: 1.6;
+    padding-left: 5px;
+    display: list-item;
+}
+
+ul ul {
+    margin: 8px 0;
+    padding-left: 25px;
+    list-style-type: circle;
+}
+
+ul ul ul {
+    list-style-type: square;
+}
+
+ol ol {
+    margin: 8px 0;
+    padding-left: 25px;
+    list-style-type: lower-alpha;
 }
 
 strong {
     font-weight: 600;
     color: #000;
 }
+
+em {
+    font-style: italic;
+}
+
+code {
+    font-family: 'Courier New', monospace;
+    background-color: #f4f4f4;
+    padding: 2px 6px;
+    border-radius: 3px;
+    font-size: 10pt;
+}
+
+pre {
+    background-color: #f4f4f4;
+    border: 1px solid #ddd;
+    border-radius: 4px;
+    padding: 12px;
+    overflow-x: auto;
+    font-family: 'Courier New', monospace;
+    font-size: 9.5pt;
+    line-height: 1.4;
+    margin: 16px 0;
+}
 """
 
 # ---------------------------------------------------------
@@ -371,16 +423,33 @@ class Q4ReportGenerator:
         """
     
     def _process_section(self, section: Dict, output_dir: str) -> str:
+        """Process section with enhanced markdown parsing for proper list rendering."""
         html = ""
         
         content = section.get("content", "")
         if content:
+            # Use comprehensive markdown extensions for full parsing
             html_content = markdown.markdown(
                 content,
-                extensions=['extra', 'sane_lists', 'tables', 'fenced_code']
+                extensions=[
+                    'extra',           # Tables, footnotes, attributes
+                    'sane_lists',      # Better list handling
+                    'tables',          # Table support
+                    'fenced_code',     # Code blocks
+                    'nl2br',           # Newline to <br>
+                    'codehilite',      # Syntax highlighting
+                    'toc'              # Table of contents
+                ],
+                extension_configs={
+                    'codehilite': {
+                        'guess_lang': False,
+                        'noclasses': True
+                    }
+                }
             )
             html += f"<div class='section'>{html_content}</div>"
         
+        # Handle images
         images = section.get("images", [])
         for img_data in images:
             if isinstance(img_data, dict):
@@ -401,6 +470,7 @@ class Q4ReportGenerator:
                     </div>
                     """
         
+        # Page break if requested
         if section.get("page_break", False):
             html += '<div class="page-break"></div>'
         
@@ -411,20 +481,24 @@ class Q4ReportGenerator:
                  filename: str = "Q4_Report.pdf",
                  output_dir: str = "./output",
                  cover_page: bool = True) -> bool:
+        """Generate PDF report with proper markdown parsing."""
         try:
             os.makedirs(output_dir, exist_ok=True)
             output_path = os.path.join(output_dir, filename)
             
             logger.info(f"Starting Q4 report generation: {output_path}")
             
+            # Build HTML body
             body_html = ""
             if cover_page:
                 body_html += self._build_cover_page()
             
+            # Process all sections
             for idx, section in enumerate(sections):
                 logger.info(f"Processing section {idx + 1}/{len(sections)}")
                 body_html += self._process_section(section, output_dir)
             
+            # Construct full HTML document
             full_html = f"""
             <!DOCTYPE html>
             <html>
@@ -440,15 +514,78 @@ class Q4ReportGenerator:
             </html>
             """
             
+            # Generate PDF
             logger.info("Rendering PDF with WeasyPrint...")
             html_doc = HTML(string=full_html)
             css_doc = CSS(string=CSS_STYLE)
             
             html_doc.write_pdf(output_path, stylesheets=[css_doc])
             
-            logger.info(f"PDF Generated: {output_path}")
+            logger.info(f"✓ PDF Generated Successfully: {output_path}")
             return True
             
         except Exception as e:
-            logger.error(f"Error generating report: {e}", exc_info=True)
-            return False
+            logger.error(f"✗ Error generating report: {e}", exc_info=True)
+            return False
+    
+    def cleanup_images(self, output_dir: str):
+        """Optional: Clean up downloaded images after PDF generation."""
+        try:
+            img_dir = os.path.join(output_dir, 'images')
+            if os.path.exists(img_dir):
+                import shutil
+                shutil.rmtree(img_dir)
+                logger.info("Cleaned up temporary images")
+        except Exception as e:
+            logger.warning(f"Failed to cleanup images: {e}")
+
+
+# ---------------------------------------------------------
+# USAGE EXAMPLE
+# ---------------------------------------------------------
+# if __name__ == "__main__":
+#     # Example usage
+#     generator = Q4ReportGenerator(
+#         title="Q4 2024 Performance Report",
+#         subtitle="Product Analytics Summary",
+#         author="Analytics Team",
+#         department="Product Department"
+#     )
+    
+#     sections = [
+#         {
+#             "content": """
+# # Executive Summary
+
+# This report provides a comprehensive analysis of Q4 2024 product performance.
+
+# ## Key Highlights
+
+# * Revenue increased by 23% compared to Q3
+# * Customer satisfaction improved to 4.7/5.0
+# * Successfully launched 3 new product features
+
+# ## Daily Product Additions
+
+# Example Daily Additions:
+
+# * 2023-09-30: 3 products
+# * 2023-10-01: 1 product
+# * 2023-10-02: 4 products
+# * 2024-09-27: 3 products
+# * 2024-09-28: 3 products
+# * 2024-09-29: 2 products
+
+# ## Growth Metrics
+
+# 1. User acquisition up 45%
+# 2. Retention rate improved to 87%
+# 3. Average session time increased by 12 minutes
+#             """,
+#             "page_break": False
+#         }
+#     ]
+    
+#     success = generator.generate(sections, filename="Q4_Report.pdf")
+#     if success:
+#         print("Report generated successfully!")