Update backend/utils/pdf_generator.py

backend/utils/pdf_generator.py

@@ -29,7 +29,7 @@ from cryptography.hazmat.backends import default_backend
 from reportlab.lib import colors
 from reportlab.lib.pagesizes import letter
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle, PageBreak, Image
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle, PageBreak, Image, KeepTogether
 from reportlab.graphics.shapes import Drawing
 from reportlab.graphics.charts.barcharts import VerticalBarChart
 
@@ -1062,7 +1062,26 @@ def generate_scan_pdf(scan, vulnerabilities):
 
         # Legal Disclaimer & Confidentiality Notice
         elements.append(PageBreak())
-        disclaimer_heading = Paragraph("<b>Legal Disclaimer, Scope Boundaries & Limitation of Liability</b>", styles['Heading3'])
+        
+        disclaimer_title_style = ParagraphStyle(
+            'DisclaimerTitleStyle',
+            parent=styles['Heading3'],
+            fontSize=11,
+            leading=14,
+            textColor=colors.HexColor("#0F172A"),
+            spaceAfter=6
+        )
+        
+        disclaimer_body_style = ParagraphStyle(
+            'DisclaimerBodyStyle',
+            parent=styles['Normal'],
+            fontSize=8,
+            leading=10.5,
+            textColor=colors.HexColor("#334155"),
+            alignment=4
+        )
+
+        disclaimer_heading = Paragraph("<b>Legal Disclaimer, Scope Boundaries & Limitation of Liability</b>", disclaimer_title_style)
         disclaimer_body = (
             "<b>1. Nature of Automated Security Assessment (\"AS-IS\" Provision)</b><br/>"
             "This vulnerability assessment report is completely system-generated by the LarShield automated engine. "
@@ -1086,16 +1105,16 @@ def generate_scan_pdf(scan, vulnerabilities):
             "This report contains sensitive, proprietary vulnerability intelligence regarding the client organization's digital assets. "
             "Unauthorized distribution, public exposure, or reproduction of this document without prior written authorization from both the client organization and LarShield is strictly prohibited."
         )
-        disclaimer_table = Table([[disclaimer_heading], [Paragraph(disclaimer_body, normal)]], colWidths=[460])
+        disclaimer_table = Table([[disclaimer_heading], [Paragraph(disclaimer_body, disclaimer_body_style)]], colWidths=[460])
         disclaimer_table.setStyle(TableStyle([
             ('BACKGROUND', (0,0), (-1,-1), colors.HexColor("#F8FAFC")),
             ('GRID', (0,0), (-1,-1), 0.5, colors.HexColor("#CBD5E1")),
-            ('TOPPADDING', (0,0), (-1,-1), 12),
-            ('BOTTOMPADDING', (0,0), (-1,-1), 12),
+            ('TOPPADDING', (0,0), (-1,-1), 10),
+            ('BOTTOMPADDING', (0,0), (-1,-1), 10),
             ('LEFTPADDING', (0,0), (-1,-1), 12),
             ('RIGHTPADDING', (0,0), (-1,-1), 12),
         ]))
-        elements.append(disclaimer_table)
+        elements.append(KeepTogether(disclaimer_table))
         elements.append(Spacer(1, 15))
 
         return elements