Update app.py

app.py

@@ -9,10 +9,7 @@ import base64
 import joblib
 from datetime import datetime
 import shutil
-from reportlab.lib.pagesizes import letter
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle, Image
-from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
-from reportlab.lib import colors
+import pdfkit
 import atexit
 import glob
 
@@ -130,112 +127,84 @@ def build_table(title, rows):
 # Function to save the health report to PDF
 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()
-        title_style = ParagraphStyle(
-            name='Title',
-            fontSize=16,
-            leading=20,
-            alignment=1,
-            spaceAfter=20,
-            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,
-            leading=14,
-            spaceAfter=10,
-            textColor=colors.black,
-            fontName='Helvetica'
-        )
-        
-        flowables = []
-
-        # Header with Profile Image and Patient Details
+        # Generate HTML content
         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, 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),
-                ('TEXTCOLOR', (0, 0), (-1, 0), colors.whitesmoke),
-                ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
-                ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
-                ('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),
-                ('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)
+        health_card_html = f"""
+        <!DOCTYPE html>
+        <html>
+        <head>
+            <style>
+                body {{ font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif; max-width: 700px; margin: 20px auto; color: #1a1a1a; }}
+                #health-card {{ border-radius: 16px; background: linear-gradient(135deg, #e3f2fd 0%, #f3e5f5 100%); border: 2px solid #ddd; box-shadow: 0 8px 32px rgba(0, 0, 0, 0.15); padding: 30px; }}
+                .header {{ background-color: rgba(255, 255, 255, 0.9); border-radius: 12px; padding: 20px; margin-bottom: 25px; border: 1px solid #e0e0e0; display: flex; align-items: center; }}
+                .header-title {{ background: linear-gradient(135deg, #64b5f6, #42a5f5); padding: 8px 16px; border-radius: 8px; margin-right: 20px; }}
+                .header-title h3 {{ margin: 0; color: white; font-size: 16px; font-weight: 600; }}
+                .header-date {{ margin-left: auto; text-align: right; color: #666; font-size: 12px; }}
+                .profile {{ display: flex; align-items: center; }}
+                .profile img {{ width: 90px; height: 90px; border-radius: 50%; margin-right: 20px; border: 3px solid #fff; box-shadow: 0 4px 12px rgba(0,0,0,0.1); }}
+                .results {{ background-color: rgba(255, 255, 255, 0.95); border-radius: 12px; padding: 25px; margin-bottom: 25px; border: 1px solid #e0e0e0; }}
+                .summary {{ background-color: rgba(255, 255, 255, 0.95); padding: 20px; border-radius: 12px; border: 1px solid #e0e0e0; margin-bottom: 25px; }}
+                .summary h4 {{ margin: 0 0 15px 0; color: #2c3e50; font-size: 18px; font-weight: 600; }}
+                .buttons {{ display: flex; gap: 15px; justify-content: center; flex-wrap: wrap; }}
+                button:disabled {{ 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); }}
+                button {{ padding: 12px 24px; background: linear-gradient(135deg, #2196f3, #1976d2); color: white; border: none; border-radius: 8px; cursor: pointer; font-weight: 600; font-size: 14px; box-shadow: 0 4px 12px rgba(33, 150, 243, 0.3); }}
+                @media print {{ .gradio-container {{ display: block; }} #health-card {{ display: block; }} }}
+            </style>
+        </head>
+        <body>
+            <div id="health-card">
+                <div class="header">
+                    <div class="header-title"><h3>HEALTH CARD</h3></div>
+                    <div class="header-date">
+                        <div>Report Date: {current_date}</div>
+                        {f'<div>Patient ID: {patient_id}</div>' if patient_id else ''}
+                    </div>
+                </div>
+                <div class="profile">
+                    <img src="data:image/png;base64,{profile_image_base64}" alt="Profile">
+                    <div>
+                        <h2>{patient_name if patient_name else 'Lab Test Results'}</h2>
+                        <p>{f'Age: {patient_age} | Gender: {patient_gender}' if patient_age and patient_gender else 'AI-Generated Health Analysis'}</p>
+                        <p>Face-Based Health Analysis Report</p>
+                    </div>
+                </div>
+                <div class="results">
+                    {test_results['Hematology']}
+                    {test_results['Iron Panel']}
+                    {test_results['Liver & Kidney']}
+                    {test_results['Electrolytes']}
+                    {test_results['Vitals']}
+                </div>
+                <div class="summary">
+                    <h4>📝 Summary & Recommendations</h4>
+                    <div>{summary}</div>
+                </div>
+                <div class="buttons">
+                    <button disabled>📥 Download Report</button>
+                    <button>📞 Find Labs Near Me</button>
+                </div>
+            </div>
+        </body>
+        </html>
+        """
+
+        # Save HTML to a temporary file
+        html_temp_path = "/tmp/health_card.html"
+        with open(html_temp_path, "w", encoding="utf-8") as f:
+            f.write(health_card_html)
+
+        # Convert HTML to PDF using pdfkit
+        pdfkit.from_file(html_temp_path, filename, options={"quiet": ""})
         print(f"PDF generated successfully at: {filename}")  # Debug log
-        return f"PDF saved successfully as {filename}", filename
+
+        # Move to /tmp for Gradio access
+        temp_pdf_path = "/tmp/" + os.path.basename(filename)
+        shutil.copy(filename, temp_pdf_path)
+
+        if os.path.exists(temp_pdf_path) and os.access(temp_pdf_path, os.R_OK):
+            return f"PDF saved successfully as {filename}", temp_pdf_path
+        else:
+            return "Error: PDF file not accessible.", None
     except Exception as e:
         print(f"Error saving PDF: {str(e)}")  # Debug log
         return f"Error saving PDF: {str(e)}", None