Update src/report_generator.py

src/report_generator.py

@@ -150,6 +150,7 @@ Do NOT invent any data. Return ONLY the JSON object.
 def build_medical_report(data: dict) -> bytes:
     """
     Generates a PDF from the extracted JSON and returns PDF bytes.
+    Modern dark theme version with minimal design.
     """
     # Create a temporary file for the PDF
     fd, temp_path = tempfile.mkstemp(suffix='.pdf')
@@ -165,29 +166,36 @@ def build_medical_report(data: dict) -> bytes:
         bottomMargin=2*cm
     )
     
+    # Custom colors for dark theme
+    background_color = colors.black
+    text_color = colors.white
+    accent_color = colors.purple
+    secondary_color = colors.lavender
+    highlight_color = colors.mediumorchid
+    
     # Set up styles with more professional fonts
     styles = getSampleStyleSheet()
     
-    # Title style
+    # Title style with gradient effect
     styles.add(ParagraphStyle(
         name='ReportTitle',
         fontName='Helvetica-Bold',
-        fontSize=16, 
-        leading=20,
+        fontSize=20, 
+        leading=24,
         alignment=TA_CENTER,
-        spaceAfter=0.2*inch,
-        textColor=colors.navy
+        spaceAfter=0.3*inch,
+        textColor=highlight_color
     ))
     
     # Section heading style
     styles.add(ParagraphStyle(
         name='SectionHeading',
         fontName='Helvetica-Bold',
-        fontSize=12, 
-        leading=14,
+        fontSize=14, 
+        leading=16,
         spaceAfter=0.1*inch,
         alignment=TA_LEFT,
-        textColor=colors.navy
+        textColor=accent_color
     ))
     
     # Normal text style
@@ -197,7 +205,8 @@ def build_medical_report(data: dict) -> bytes:
         fontSize=10, 
         leading=12,
         spaceAfter=0.1*inch,
-        alignment=TA_LEFT
+        alignment=TA_LEFT,
+        textColor=text_color
     ))
     
     # Company info style
@@ -207,248 +216,237 @@ def build_medical_report(data: dict) -> bytes:
         fontSize=9, 
         leading=11,
         alignment=TA_RIGHT,
-        textColor=colors.darkblue
+        textColor=secondary_color
     ))
     
+    # Setup canvas callback to draw background
+    def add_page_background(canvas, doc):
+        canvas.saveState()
+        canvas.setFillColor(background_color)
+        canvas.rect(0, 0, doc.pagesize[0], doc.pagesize[1], fill=1)
+        canvas.restoreState()
+    
     # Initialize elements list
     elems = []
 
-    # Add logo and company header - use path from app.py
-    logo_path = "src/assets/Logo_no_background.png" 
+    # Add logo and company header
+    logo_path = "src/assets/Logo_no_background.png"
     
-    # Get logo image using our helper
+    # Get logo image
     logo_img = get_reportlab_image(logo_path)
     
-    # Create a table for the header (logo on left, company info on right)
+    # Company info
     company_info = Paragraph(
         """
-        <b>Daease</b><br/>
-        Website: http://daease.com/<br/>
-        Email: [email protected]
+        <b><font color="#E0B0FF">Daease</font></b><br/>
+        Website: <font color="#E0B0FF">http://daease.com/</font><br/>
+        Email: <font color="#E0B0FF">[email protected]</font>
         """, 
         styles['CompanyInfo']
     )
     
+    # Add header with logo or text-only fallback
     if logo_img:
-        # Use the logo in the header table
         header_table = Table([[logo_img, company_info]], colWidths=[doc.width/2.0]*2)
         header_table.setStyle(TableStyle([
             ('VALIGN', (0, 0), (-1, -1), 'TOP'),
             ('ALIGN', (0, 0), (0, 0), 'LEFT'),
             ('ALIGN', (1, 0), (1, 0), 'RIGHT'),
             ('BOTTOMPADDING', (0, 0), (-1, -1), 10),
+            ('BACKGROUND', (0, 0), (-1, -1), background_color),
         ]))
+        elems.append(header_table)
     else:
-        # Fallback to text-only header if logo can't be loaded
-        header_table = Table([['Daease', company_info]], colWidths=[doc.width/2.0]*2)
-        header_table.setStyle(TableStyle([
-            ('VALIGN', (0, 0), (-1, -1), 'TOP'),
-            ('ALIGN', (0, 0), (0, 0), 'LEFT'),
-            ('ALIGN', (1, 0), (1, 0), 'RIGHT'),
-            ('BOTTOMPADDING', (0, 0), (-1, -1), 10),
-            ('FONTNAME', (0, 0), (0, 0), 'Helvetica-Bold'),
-            ('FONTSIZE', (0, 0), (0, 0), 14),
-            ('TEXTCOLOR', (0, 0), (0, 0), colors.navy),
-        ]))
+        company_header = Paragraph(
+            """
+            <b><font color="#E0B0FF" size="14">Daease</font></b><br/>
+            Website: <font color="#E0B0FF">http://daease.com/</font><br/>
+            Email: <font color="#E0B0FF">[email protected]</font>
+            """, 
+            styles['CompanyInfo']
+        )
+        elems.append(company_header)
     
-    elems.append(header_table)
     elems.append(Spacer(1, 0.3*inch))
 
-    # Add title - with more prominent styling
-    title_style = ParagraphStyle(
-        name='ReportTitleEnhanced',
-        fontName='Helvetica-Bold',
-        fontSize=18, 
-        leading=22,
-        alignment=TA_CENTER,
-        spaceAfter=0.3*inch,
-        textColor=colors.navy
+    # Add title with subtle gradient-like effect
+    title = Paragraph(
+        '<font color="#E0B0FF">Daease</font> '
+        '<font color="#D8BFD8">Medical</font> '
+        '<font color="#DA70D6">Consultation</font> '
+        '<font color="#BA55D3">Report</font>', 
+        styles['ReportTitle']
     )
-    elems.append(Paragraph("Daease Medical Consultation Report", title_style))
+    elems.append(title)
     elems.append(Spacer(1, 0.2*inch))
 
-    # Add a horizontal line
-    tbl_line = Table([['']],  colWidths=[doc.width], rowHeights=[1])
-    tbl_line.setStyle(TableStyle([
-        ('LINEABOVE', (0, 0), (-1, -1), 1, colors.navy),
+    # Add a horizontal separator line
+    separator = Table([['']],  colWidths=[doc.width], rowHeights=[1])
+    separator.setStyle(TableStyle([
+        ('LINEABOVE', (0, 0), (-1, -1), 1, accent_color),
+        ('BACKGROUND', (0, 0), (-1, -1), background_color),
     ]))
-    elems.append(tbl_line)
-    elems.append(Spacer(1, 0.2*inch))
+    elems.append(separator)
+    elems.append(Spacer(1, 0.3*inch))
 
-    # Patient info in a clean table
+    # --- TABLE 1: Patient Information ---
     patient = data.get('patient', {})
     
-    # Format the patient table data with improved styling
+    # Create a cleaner patient info table
     patient_data = [
-        ['Patient Information', '', ''],
-        ['Name:', patient.get('name', '–'), ''],
-        ['Age:', patient.get('age', '–'), ''],
-        ['Gender:', patient.get('gender', '–'), ''],
-        ['Visit Date:', data.get('visit_date', datetime.now().strftime("%Y-%m-%d")), '']
+        [Paragraph('<font color="#BA55D3"><b>Patient Information</b></font>', styles['SectionHeading']), ''],
+        ['<font color="#D8BFD8">Name:</font>', patient.get('name', '–')],
+        ['<font color="#D8BFD8">Age:</font>', patient.get('age', '–')],
+        ['<font color="#D8BFD8">Gender:</font>', patient.get('gender', '–')],
+        ['<font color="#D8BFD8">Visit Date:</font>', data.get('visit_date', datetime.now().strftime("%Y-%m-%d"))]
     ]
     
-    # Create a more structured patient table with better column distribution
     patient_table = Table(
         patient_data, 
-        colWidths=[doc.width * 0.2, doc.width * 0.5, doc.width * 0.3],
-        rowHeights=[0.4*inch, 0.3*inch, 0.3*inch, 0.3*inch, 0.3*inch]
+        colWidths=[doc.width * 0.3, doc.width * 0.7],
+        rowHeights=[0.4*inch, 0.25*inch, 0.25*inch, 0.25*inch, 0.25*inch]
     )
+    
     patient_table.setStyle(TableStyle([
-        # Header row styling
-        ('BACKGROUND', (0, 0), (2, 0), colors.navy),
-        ('TEXTCOLOR', (0, 0), (2, 0), colors.white),
-        ('FONTNAME', (0, 0), (2, 0), 'Helvetica-Bold'),
-        ('SPAN', (0, 0), (2, 0)),  # Span all columns in header
-        ('ALIGN', (0, 0), (2, 0), 'CENTER'),
+        # Header styling
+        ('SPAN', (0, 0), (1, 0)),  # Span the header cell
+        ('ALIGN', (0, 0), (0, 0), 'CENTER'),
         
-        # Label column styling (first column)
-        ('FONTNAME', (0, 1), (0, -1), 'Helvetica-Bold'),
+        # Label column styling
         ('ALIGN', (0, 1), (0, -1), 'RIGHT'),
-        ('RIGHTPADDING', (0, 1), (0, -1), 10),
-        
-        # Value column styling (second column)
-        ('ALIGN', (1, 1), (1, -1), 'LEFT'),
-        
-        # Table borders and grid
         ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
-        ('GRID', (0, 0), (1, -1), 0.5, colors.lightgrey),
-        ('BOX', (0, 0), (1, -1), 1, colors.darkblue),
+        ('RIGHTPADDING', (0, 1), (0, -1), 12),
         
-        # Alternating row colors for better readability
-        ('BACKGROUND', (0, 1), (1, 1), colors.whitesmoke),
-        ('BACKGROUND', (0, 3), (1, 3), colors.whitesmoke),
+        # Cell styling
+        ('BACKGROUND', (0, 0), (-1, -1), background_color),
+        ('TEXTCOLOR', (1, 1), (1, -1), text_color),
+        
+        # Borders
+        ('GRID', (0, 1), (1, -1), 0.5, secondary_color),
+        ('BOX', (0, 1), (1, -1), 1, accent_color),
     ]))
     
     elems.append(patient_table)
-    elems.append(Spacer(1, 0.3*inch))
-
-    # Helper to add sections as tables with improved styling
-    def add_section_table(title, content):
+    elems.append(Spacer(1, 0.4*inch))
+    
+    # --- TABLE 2: Medical Information (All in one table) ---
+    # Collect all medical data
+    medical_sections = [
+        ("Chief Complaint", data.get('chief_complaint')),
+        ("History of Present Illness", data.get('history_of_present_illness')),
+        ("Past Medical History", data.get('past_medical_history')),
+        ("Medications", data.get('medications')),
+        ("Allergies", data.get('allergies')),
+        ("Examination Findings", data.get('examination')),
+        ("Diagnosis", data.get('diagnosis')),
+        ("Recommendations", data.get('recommendations')),
+        ("Reasoning", data.get('reasoning')),
+        ("Sources", data.get('sources'))
+    ]
+    
+    # Create table data
+    elems.append(Paragraph('<font color="#BA55D3"><b>Medical Information</b></font>', styles['SectionHeading']))
+    elems.append(Spacer(1, 0.1*inch))
+    
+    # Create medical info rows
+    for title, content in medical_sections:
+        if not content or str(content).strip() == '':
+            content = '–'
+        
+        # Format content as paragraphs
         if isinstance(content, list):
             content = '\n'.join(map(str, content))
         elif content is not None and not isinstance(content, str):
             content = str(content)
-            
-        if not content or content.strip() == '':
-            content = '–'
-            
-        # Create a gradient-like effect with nested tables
+        
+        # Add section
         section_data = [
-            # Header row with gradient-like styling
-            [Paragraph(f"<b>{title}</b>", 
-                      ParagraphStyle(
-                          name='TableHeader',
-                          fontName='Helvetica-Bold',
-                          fontSize=11,
-                          textColor=colors.white,
-                          alignment=TA_LEFT
-                      ))],
-            # Content row
-            [Paragraph(content, styles['NormalText'])]
+            [f'<font color="#D8BFD8"><b>{title}</b></font>', 
+             Paragraph(content, styles['NormalText'])]
         ]
         
-        # Create the main table
-        table = Table(section_data, colWidths=[doc.width * 0.97], rowHeights=[0.35*inch, None])
-        table.setStyle(TableStyle([
-            # Header styling with gradient-like effect
-            ('BACKGROUND', (0, 0), (0, 0), colors.navy),
-            ('LINEBELOW', (0, 0), (0, 0), 1, colors.darkblue),
-            ('LEFTPADDING', (0, 0), (0, 0), 10),
-            ('TOPPADDING', (0, 0), (0, 0), 6),
-            ('BOTTOMPADDING', (0, 0), (0, 0), 6),
+        section_table = Table(
+            section_data,
+            colWidths=[doc.width * 0.25, doc.width * 0.75],
+            rowHeights=None  # Auto height
+        )
+        
+        section_table.setStyle(TableStyle([
+            # Cell styling
+            ('VALIGN', (0, 0), (0, 0), 'TOP'),
+            ('ALIGN', (0, 0), (0, 0), 'RIGHT'),
+            ('RIGHTPADDING', (0, 0), (0, 0), 12),
+            ('TOPPADDING', (0, 0), (1, 0), 8),
+            ('BOTTOMPADDING', (0, 0), (1, 0), 8),
             
-            # Content styling
-            ('LEFTPADDING', (0, 1), (0, 1), 10),
-            ('RIGHTPADDING', (0, 1), (0, 1), 10),
-            ('TOPPADDING', (0, 1), (0, 1), 8),
-            ('BOTTOMPADDING', (0, 1), (0, 1), 8),
-            ('BACKGROUND', (0, 1), (0, 1), colors.white),
+            # Background
+            ('BACKGROUND', (0, 0), (-1, -1), background_color),
             
-            # Border styling
-            ('BOX', (0, 0), (-1, -1), 1, colors.lightgrey),
+            # Bottom border only
+            ('LINEBELOW', (0, 0), (1, 0), 0.5, secondary_color),
         ]))
         
-        elems.append(KeepTogether([table, Spacer(1, 0.15*inch)]))
-
-    # Add all medical information sections as tables
-    add_section_table("Chief Complaint", data.get('chief_complaint'))
-    add_section_table("History of Present Illness", data.get('history_of_present_illness'))
-    add_section_table("Past Medical History", data.get('past_medical_history'))
-    add_section_table("Medications", data.get('medications'))
-    add_section_table("Allergies", data.get('allergies'))
-    add_section_table("Examination Findings", data.get('examination'))
-    add_section_table("Diagnosis", data.get('diagnosis'))
-    add_section_table("Recommendations", data.get('recommendations'))
-    add_section_table("Reasoning", data.get('reasoning'))
-    add_section_table("Sources", data.get('sources'))
+        elems.append(section_table)
     
     # Add AI signature section
     elems.append(Spacer(1, 0.4*inch))
     
     # Add a horizontal line before signature
-    sig_line = Table([['']],  colWidths=[doc.width], rowHeights=[1])
-    sig_line.setStyle(TableStyle([
-        ('LINEABOVE', (0, 0), (-1, -1), 0.5, colors.grey),
+    sig_separator = Table([['']],  colWidths=[doc.width], rowHeights=[1])
+    sig_separator.setStyle(TableStyle([
+        ('LINEABOVE', (0, 0), (-1, -1), 0.5, accent_color),
+        ('BACKGROUND', (0, 0), (-1, -1), background_color),
     ]))
-    elems.append(sig_line)
+    elems.append(sig_separator)
     elems.append(Spacer(1, 0.2*inch))
     
-    # Create a robust signature section
-    signature_text = Paragraph("<i>Generated by</i>", styles['NormalText'])
-    ai_name = Paragraph("<b>Daease AI Medical Assistant</b>", styles['NormalText'])
+    # Signature text
+    signature_text = Paragraph('<font color="#D8BFD8"><i>Generated by</i></font>', styles['NormalText'])
+    ai_name = Paragraph('<font color="#BA55D3"><b>Daease AI Medical Assistant</b></font>', styles['NormalText'])
     
-    # Use the same logo path for signature
-    signature_img = get_reportlab_image("src/assets/logo.png", width=1.2*inch, height=0.8*inch)
+    # Use the logo as signature if available
+    signature_img = get_reportlab_image("src/assets/Logo_no_background.png", width=1.2*inch, height=0.8*inch)
     
     if signature_img:
-        # Create a 3-column table for better alignment
         sig_table = Table(
-            [[signature_text, '', ''], 
-             [signature_img, '', ''], 
-             [ai_name, '', '']], 
-            colWidths=[doc.width * 0.3, doc.width * 0.35, doc.width * 0.35]
+            [[signature_text, ''], 
+             [signature_img, ''], 
+             [ai_name, '']], 
+            colWidths=[doc.width * 0.3, doc.width * 0.7],
+            rowHeights=[None, None, None]
         )
         
         sig_table.setStyle(TableStyle([
             ('ALIGN', (0, 0), (0, -1), 'CENTER'),
             ('VALIGN', (0, 0), (0, -1), 'MIDDLE'),
+            ('BACKGROUND', (0, 0), (-1, -1), background_color),
         ]))
         elems.append(sig_table)
     else:
-        # Text-only fallback for signature
         sig_text = Paragraph(
-            "<i>Generated by</i><br/><b>Daease AI Medical Assistant</b>", 
-            ParagraphStyle(
-                name='SignatureText',
-                fontName='Helvetica-Bold',
-                fontSize=10,
-                alignment=TA_LEFT,
-                spaceAfter=0.1*inch
-            )
+            '<font color="#D8BFD8"><i>Generated by</i></font><br/>'
+            '<font color="#BA55D3"><b>Daease AI Medical Assistant</b></font>', 
+            styles['NormalText']
         )
         elems.append(sig_text)
     
-    # Add disclaimer with more distinctive styling
+    # Add disclaimer with subtle styling
     elems.append(Spacer(1, 0.3*inch))
     disclaimer = Paragraph(
-        "<i>Disclaimer: This report is generated by AI for informational purposes only. "
-        "It should not replace professional medical advice, diagnosis, or treatment. "
-        "Always consult with a qualified healthcare provider for medical concerns.</i>",
+        '<font color="#A9A9A9"><i>Disclaimer: This report is generated by AI for informational purposes only. '
+        'It should not replace professional medical advice, diagnosis, or treatment. '
+        'Always consult with a qualified healthcare provider for medical concerns.</i></font>',
         ParagraphStyle(
             name='Disclaimer',
             fontName='Helvetica-Oblique',
             fontSize=8,
-            textColor=colors.grey,
-            borderWidth=1,
-            borderColor=colors.lightgrey,
-            borderPadding=5,
-            borderRadius=2
+            alignment=TA_CENTER
         )
     )
     elems.append(disclaimer)
     
-    # Build the PDF document
-    doc.build(elems)
+    # Build the PDF document with background
+    doc.build(elems, onFirstPage=add_page_background, onLaterPages=add_page_background)
     
     # Read PDF bytes
     with open(temp_path, 'rb') as file: