Update app.py

app.py

@@ -10,7 +10,7 @@ import joblib
 from datetime import datetime
 import shutil
 from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle, Image
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.lib import colors
 import atexit
@@ -128,7 +128,7 @@ def build_table(title, rows):
     return html
 
 # Function to save the health report to PDF
-def save_results_to_pdf(test_results, test_values, filename):
+def save_results_to_pdf(profile_image_base64, test_results, summary, patient_name, patient_age, patient_gender, patient_id, filename):
     try:
         doc = SimpleDocTemplate(filename, pagesize=letter)
         styles = getSampleStyleSheet()
@@ -141,6 +141,13 @@ def save_results_to_pdf(test_results, test_values, filename):
             textColor=colors.black,
             fontName='Helvetica-Bold'
         )
+        header_style = ParagraphStyle(
+            name='Header',
+            fontSize=14,
+            leading=16,
+            textColor=colors.darkblue,
+            fontName='Helvetica-Bold'
+        )
         body_style = ParagraphStyle(
             name='Body',
             fontSize=12,
@@ -150,37 +157,62 @@ def save_results_to_pdf(test_results, test_values, filename):
             fontName='Helvetica'
         )
         
-        flowables = [Paragraph("Health Report", title_style), Spacer(1, 12)]
-        
-        # Define ranges for all tests
-        test_ranges = {
-            "Hemoglobin": (13.5, 17.5),
-            "WBC Count": (4.0, 11.0),
-            "Platelet Count": (150, 450),
-            "Iron": (60, 170),
-            "Ferritin": (30, 300),
-            "TIBC": (250, 400),
-            "Bilirubin": (0.3, 1.2),
-            "Creatinine": (0.6, 1.2),
-            "Urea": (7, 20),
-            "Sodium": (135, 145),
-            "Potassium": (3.5, 5.1),
-            "SpO2": (95, 100),
-            "Heart Rate": (60, 100),
-            "Respiratory Rate": (12, 20),
-            "Temperature": (97, 99),
-            "BP Systolic": (90, 120),
-            "BP Diastolic": (60, 80)
+        flowables = []
+
+        # Header with Profile Image and Patient Details
+        current_date = datetime.now().strftime("%B %d, %Y")
+        header_table_data = [
+            [Paragraph("HEALTH CARD", header_style), "", Paragraph(f"Report Date: {current_date}<br/>{f'Patient ID: {patient_id}' if patient_id else ''}", body_style)]
+        ]
+        header_table = Table(header_table_data, colWidths=[100, 300, 150])
+        header_table.setStyle(TableStyle([
+            ('BACKGROUND', (0, 0), (0, 0), colors.lightblue),
+            ('BACKGROUND', (2, 0), (2, 0), colors.lightgrey),
+            ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
+            ('ALIGN', (0, 0), (0, 0), 'LEFT'),
+            ('ALIGN', (2, 0), (2, 0), 'RIGHT'),
+        ]))
+        flowables.append(header_table)
+
+        # Profile Image and Patient Info
+        if profile_image_base64:
+            import io
+            from reportlab.lib.utils import ImageReader
+            img_data = base64.b64decode(profile_image_base64)
+            img_file = io.BytesIO(img_data)
+            img = ImageReader(img_file)
+            img_width = 90
+            img_height = 90
+            img.drawHeight = img_height * 0.5
+            img.drawWidth = img_width * 0.5
+            patient_info = f"{patient_name if patient_name else 'Lab Test Results'}<br/>{f'Age: {patient_age} | Gender: {patient_gender}' if patient_age and patient_gender else 'AI-Generated Health Analysis'}<br/>Face-Based Health Analysis Report"
+            profile_table_data = [[Image(img, width=img_width, height=img_height), Paragraph(patient_info, body_style)]]
+            profile_table = Table(profile_table_data, colWidths=[100, 400])
+            profile_table.setStyle(TableStyle([
+                ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
+                ('LEFTPADDING', (0, 0), (0, 0), 0),
+                ('RIGHTPADDING', (1, 0), (1, 0), 10),
+            ]))
+            flowables.append(profile_table)
+        flowables.append(Spacer(1, 20))
+
+        # Test Results Sections
+        section_tests = {
+            "Hematology": [("Hemoglobin", (13.5, 17.5)), ("WBC Count", (4.0, 11.0)), ("Platelet Count", (150, 450))],
+            "Iron Panel": [("Iron", (60, 170)), ("Ferritin", (30, 300)), ("TIBC", (250, 400))],
+            "Liver & Kidney": [("Bilirubin", (0.3, 1.2)), ("Creatinine", (0.6, 1.2)), ("Urea", (7, 20))],
+            "Electrolytes": [("Sodium", (135, 145)), ("Potassium", (3.5, 5.1))],
+            "Vitals": [("SpO2", (95, 100)), ("Heart Rate", (60, 100)), ("Respiratory Rate", (12, 20)),
+                       ("Temperature", (97, 99)), ("BP Systolic", (90, 120)), ("BP Diastolic", (60, 80))]
         }
-        
+
         for section_name, html in test_results.items():
             flowables.append(Paragraph(section_name, styles['Heading2']))
             table_data = [["Test", "Result", "Range", "Level"]]
-            for label in test_ranges:
-                if any(label in html for section_html in test_results.values()):
-                    value = test_values.get(label, test_ranges[label][0] + random.uniform(-1, 1))  # Use actual value or fallback
-                    level, _, _ = get_risk_color(value, test_ranges[label])
-                    table_data.append([label, f"{value:.2f}" if label not in ["WBC Count", "Platelet Count"] else f"{value:.0f}", f"{test_ranges[label][0]} - {test_ranges[label][1]}", level])
+            for label, ref_range in section_tests.get(section_name, []):
+                value = float(test_results[section_name].split(f'{label}</td><td style="padding:10px 8px;text-align:center;color:#2c3e50;font-weight:600;">')[1].split('</td>')[0]) if label in test_results[section_name] else ref_range[0] + random.uniform(-1, 1)
+                level, _, _ = get_risk_color(value, ref_range)
+                table_data.append([label, f"{value:.2f}" if label not in ["WBC Count", "Platelet Count"] else f"{value:.0f}", f"{ref_range[0]} - {ref_range[1]}", f"{level}"])
             table = Table(table_data)
             table.setStyle(TableStyle([
                 ('BACKGROUND', (0, 0), (-1, 0), colors.grey),
@@ -190,14 +222,22 @@ def save_results_to_pdf(test_results, test_values, filename):
                 ('FONTSIZE', (0, 0), (-1, 0), 12),
                 ('BOTTOMPADDING', (0, 0), (-1, 0), 12),
                 ('BACKGROUND', (0, 1), (-1, -1), colors.beige),
-                ('GRID', (0, 0), (-1, -1), 1, colors.black)
+                ('GRID', (0, 0), (-1, -1), 1, colors.black),
+                ('VALIGN', (0, 0), (-1, -1), 'MIDDLE'),
             ]))
             flowables.append(table)
             flowables.append(Spacer(1, 12))
-        
+
+        # Summary & Recommendations
+        flowables.append(Paragraph("Summary & Recommendations", styles['Heading2']))
+        flowables.append(Paragraph(summary.replace("<ul>", "").replace("</ul>", "").replace("<li>", "\u2022 ").replace("</li>", "<br/>"), body_style))
+        flowables.append(Spacer(1, 20))
+
         doc.build(flowables)
+        print(f"PDF generated successfully at: {filename}")  # Debug log
         return f"PDF saved successfully as {filename}", filename
     except Exception as e:
+        print(f"Error saving PDF: {str(e)}")  # Debug log
         return f"Error saving PDF: {str(e)}", None
 
 # Build health card layout
@@ -246,7 +286,6 @@ def build_health_card(profile_image, test_results, summary, pdf_filepath, patien
         </div>
 
         <div style="display: flex; gap: 15px; justify-content: center; flex-wrap: wrap;">
-            <!-- Disabled the direct download link to rely on Gradio File component -->
             <button disabled style="padding: 12px 24px; background: #ccc; color: white; border: none; border-radius: 8px; cursor: not-allowed; font-weight: 600; font-size: 14px; box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1);">
                 📥 Download Report
             </button>
@@ -346,7 +385,16 @@ def analyze_face(input_data):
     profile_image_base64 = base64.b64encode(buffer).decode('utf-8')
 
     pdf_filename = f"Health_Report_{datetime.now().strftime('%Y-%m-%d_%H-%M-%S')}.pdf"
-    pdf_result, pdf_filepath = save_results_to_pdf(test_results, test_values, pdf_filename)
+    pdf_result, pdf_filepath = save_results_to_pdf(
+        profile_image_base64,
+        test_results,
+        summary,
+        current_patient_details['name'],
+        current_patient_details['age'],
+        current_patient_details['gender'],
+        current_patient_details['id'],
+        pdf_filename
+    )
     
     if pdf_filepath:
         temp_pdf_path = "/tmp/" + os.path.basename(pdf_filepath)