changed report UI

src/report_generator.py

@@ -6,9 +6,11 @@ from datetime import datetime
 import streamlit as st
 from reportlab.lib.pagesizes import A4
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
-from reportlab.lib.enums import TA_LEFT
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
+from reportlab.lib.enums import TA_LEFT, TA_CENTER, TA_RIGHT
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle, Image
 from reportlab.lib import colors
+from reportlab.lib.units import inch, cm
+from reportlab.platypus import PageBreak, KeepTogether
 
 from text_processors import format_conversation_history
 from model import orchestrator_chat
@@ -78,82 +80,210 @@ def build_medical_report(data: dict) -> bytes:
     fd, temp_path = tempfile.mkstemp(suffix='.pdf')
     os.close(fd)
     
+    # Set up the document with margins
     doc = SimpleDocTemplate(
         temp_path, 
         pagesize=A4,
-        rightMargin=40, 
-        leftMargin=40,
-        topMargin=60, 
-        bottomMargin=60
+        rightMargin=1.5*cm, 
+        leftMargin=1.5*cm,
+        topMargin=1.5*cm, 
+        bottomMargin=2*cm
     )
     
+    # Set up styles with more professional fonts
     styles = getSampleStyleSheet()
+    
+    # Title style
+    styles.add(ParagraphStyle(
+        name='ReportTitle',
+        fontName='Helvetica-Bold',
+        fontSize=16, 
+        leading=20,
+        alignment=TA_CENTER,
+        spaceAfter=0.2*inch,
+        textColor=colors.navy
+    ))
+    
+    # Section heading style
     styles.add(ParagraphStyle(
         name='SectionHeading',
-        fontSize=14, 
-        leading=16,
-        spaceAfter=8,
+        fontName='Helvetica-Bold',
+        fontSize=12, 
+        leading=14,
+        spaceAfter=0.1*inch,
         alignment=TA_LEFT,
-        textColor=colors.darkblue
+        textColor=colors.navy
     ))
     
+    # Normal text style
     styles.add(ParagraphStyle(
-        name='NormalLeft',
-        fontSize=11, 
-        leading=14,
-        spaceAfter=6,
+        name='NormalText',
+        fontName='Helvetica',
+        fontSize=10, 
+        leading=12,
+        spaceAfter=0.1*inch,
         alignment=TA_LEFT
     ))
     
+    # Company info style
+    styles.add(ParagraphStyle(
+        name='CompanyInfo',
+        fontName='Helvetica',
+        fontSize=9, 
+        leading=11,
+        alignment=TA_RIGHT,
+        textColor=colors.darkblue
+    ))
+    
+    # Initialize elements list
     elems = []
 
-    # Title
-    elems.append(Paragraph("Medical Consultation Report", styles['Title']))
-    elems.append(Spacer(1, 12))
+    # Add logo and company header
+    logo_path = "src/assets/logo.png"  # Adjust path as needed
+    
+    # Create a table for the header (logo on left, company info on right)
+    logo_img = Image(logo_path, width=1.5*inch, height=1.0*inch)
+    
+    company_info = Paragraph(
+        """
+        <b>Daease</b><br/>
+        Website: http://daease.com/<br/>
+        Email: [email protected]
+        """, 
+        styles['CompanyInfo']
+    )
+    
+    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),
+    ]))
+    
+    elems.append(header_table)
+    elems.append(Spacer(1, 0.3*inch))
 
-    # Patient info
+    # Add title
+    elems.append(Paragraph("Daease Medical Consultation Report", styles['ReportTitle']))
+    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),
+    ]))
+    elems.append(tbl_line)
+    elems.append(Spacer(1, 0.2*inch))
+
+    # Patient info in a clean table
     patient = data.get('patient', {})
-    info = [
+    
+    # Format the patient table data
+    patient_data = [
+        ['Patient Information', ''],
         ['Name:', patient.get('name', '–')],
         ['Age:', patient.get('age', '–')],
         ['Gender:', patient.get('gender', '–')],
-        ['Date:', data.get('visit_date', datetime.now().strftime("%Y-%m-%d"))],
+        ['Visit Date:', data.get('visit_date', datetime.now().strftime("%Y-%m-%d"))]
     ]
     
-    tbl = Table(info, colWidths=[80, 250])
-    tbl.setStyle(TableStyle([
-        ('BACKGROUND', (0, 0), (1, 0), colors.lightgrey),
-        ('BOX', (0, 0), (-1, -1), 0.5, colors.grey),
-        ('INNERGRID', (0, 0), (-1, -1), 0.5, colors.grey),
+    patient_table = Table(patient_data, colWidths=[doc.width * 0.3, doc.width * 0.7])
+    patient_table.setStyle(TableStyle([
+        ('BACKGROUND', (0, 0), (1, 0), colors.navy),
+        ('TEXTCOLOR', (0, 0), (1, 0), colors.white),
+        ('FONTNAME', (0, 0), (1, 0), 'Helvetica-Bold'),
+        ('SPAN', (0, 0), (1, 0)),
+        ('ALIGN', (0, 0), (1, 0), 'CENTER'),
         ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
+        ('GRID', (0, 0), (-1, -1), 0.5, colors.grey),
+        ('BOX', (0, 0), (-1, -1), 1, colors.darkblue),
     ]))
     
-    elems += [tbl, Spacer(1, 16)]
+    elems.append(patient_table)
+    elems.append(Spacer(1, 0.3*inch))
 
-    # Helper to add sections
-    def add_section(title, text):
-        # Convert text to string if it's a list or another non-string type
-        if isinstance(text, list):
-            text = '\n'.join(map(str, text))
-        elif text is not None and not isinstance(text, str):
-            text = str(text)
+    # Helper to add sections as tables
+    def add_section_table(title, content):
+        if isinstance(content, list):
+            content = '\n'.join(map(str, content))
+        elif content is not None and not isinstance(content, str):
+            content = str(content)
             
-        elems.append(Paragraph(title, styles['SectionHeading']))
-        elems.append(Paragraph(text or '–', styles['NormalLeft']))
-        elems.append(Spacer(1, 12))
-
-    # Populate all sections
-    add_section("Chief Complaint", data.get('chief_complaint'))
-    add_section("History of Present Illness", data.get('history_of_present_illness'))
-    add_section("Past Medical History", data.get('past_medical_history'))
-    add_section("Medications", data.get('medications'))
-    add_section("Allergies", data.get('allergies'))
-    add_section("Examination Findings", data.get('examination'))
-    add_section("Diagnosis", data.get('diagnosis'))
-    add_section("Recommendations", data.get('recommendations'))
-    add_section("Reasoning", data.get('reasoning'))
-    add_section("Sources", data.get('sources'))
+        if not content or content.strip() == '':
+            content = '–'
+            
+        section_data = [
+            [title, ''],
+            [Paragraph(content, styles['NormalText']), '']
+        ]
+        
+        # Create table and span the content cell
+        table = Table(section_data, colWidths=[doc.width * 0.97, doc.width * 0.03])
+        table.setStyle(TableStyle([
+            ('BACKGROUND', (0, 0), (1, 0), colors.lightblue),
+            ('TEXTCOLOR', (0, 0), (1, 0), colors.navy),
+            ('FONTNAME', (0, 0), (1, 0), 'Helvetica-Bold'),
+            ('SPAN', (0, 0), (1, 0)),  # Span the header row
+            ('SPAN', (0, 1), (1, 1)),  # Span the content row
+            ('ALIGN', (0, 0), (0, 0), 'LEFT'),
+            ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
+            ('BOX', (0, 0), (-1, -1), 1, colors.lightgrey),
+            ('LINEBELOW', (0, 0), (1, 0), 1, colors.navy),
+        ]))
+        
+        elems.append(KeepTogether([table, Spacer(1, 0.1*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'))
+    
+    # Add AI signature section
+    elems.append(Spacer(1, 0.4*inch))
+    
+    # Try to add AI signature image if available
+    signature_path = "assets/Logo/LOGO.png"  # Fallback to logo if no signature available
+    try:
+        signature_img = Image(signature_path, width=1.2*inch, height=0.8*inch)
+        signature_table = Table([
+            [Paragraph("<i>Generated by</i>", styles['NormalText'])],
+            [signature_img],
+            [Paragraph("<b>Daease AI Medical Assistant</b>", styles['NormalText'])]
+        ], colWidths=[doc.width * 0.3])
+        
+        signature_table.setStyle(TableStyle([
+            ('ALIGN', (0, 0), (0, -1), 'CENTER'),
+            ('VALIGN', (0, 0), (0, -1), 'MIDDLE'),
+        ]))
+        elems.append(signature_table)
+    except:
+        # Fallback if image can't be loaded
+        elems.append(Paragraph("<i>Generated by Daease AI Medical Assistant</i>", styles['NormalText']))
+    
+    # Add disclaimer
+    elems.append(Spacer(1, 0.2*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>",
+        ParagraphStyle(
+            name='Disclaimer',
+            fontName='Helvetica-Oblique',
+            fontSize=8,
+            textColor=colors.grey
+        )
+    )
+    elems.append(disclaimer)
+    
+    # Build the PDF document
     doc.build(elems)
     
     # Read PDF bytes
@@ -219,7 +349,7 @@ def generate_and_download_report():
             st.session_state.pdf_data = pdf_bytes  # Store PDF data for email
             
             # Display success message
-            st.success("Your medical report is ready!")
+            st.success("Report is ready!")
             
             # Create two columns for the download and email buttons
             col1, col2 = st.columns(2)
@@ -227,7 +357,7 @@ def generate_and_download_report():
             with col1:
                 # Offer download
                 st.download_button(
-                    label="📥 Download Report",
+                    label="Download Report",
                     data=pdf_bytes,
                     file_name=f"medical_report_{datetime.now().strftime('%Y%m%d')}.pdf",
                     mime="application/pdf",
@@ -236,7 +366,7 @@ def generate_and_download_report():
             
             with col2:
                 # Email button that shows email form when clicked
-                if st.button("📧 Email Report"):
+                if st.button("Email Report"):
                     st.session_state.show_email_form = True
                     st.rerun()