Update app.py

app.py

@@ -9,18 +9,21 @@ import base64
 import joblib
 from datetime import datetime
 import shutil
-import pdfkit
-import atexit
 import glob
+from reportlab.lib import colors
+from reportlab.lib.pagesizes import letter
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Table, TableStyle
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+import atexit
 
 # Cleanup temporary files on exit
 def cleanup_temp_files():
-    for temp_file in glob.glob("/tmp/Health_Report_*.pdf"):
+    for temp_file in glob.glob("Health_Report_*.pdf"):
         try:
             os.remove(temp_file)
         except Exception as e:
             print(f"Failed to clean up {temp_file}: {e}")
-    for temp_file in glob.glob("/tmp/health_card*.html"):
+    for temp_file in glob.glob("health_card*.html"):
         try:
             os.remove(temp_file)
         except Exception as e:
@@ -95,159 +98,105 @@ models = {
 def get_risk_color(value, normal_range):
     low, high = normal_range
     if value < low:
-        return ("Low", "🔻", "#fff3cd")
+        return ("Low", "🔻", colors.yellow)
     elif value > high:
-        return ("High", "🔺", "#f8d7da")
+        return ("High", "🔺", colors.red)
     else:
-        return ("Normal", "✅", "#d4edda")
-
-# Function to build table for test results
-def build_table(title, rows):
-    color_map = {
-        "Normal": "#28a745",
-        "High": "#dc3545",
-        "Low": "#ffc107"
-    }
-    
-    html = (
-        f'<div style="margin-bottom: 25px; border-radius: 8px; overflow: hidden; border: 1px solid #e0e0e0;">'
-        f'<div style="background: linear-gradient(135deg, #f5f7fa, #c3cfe2); padding: 12px 16px; border-bottom: 1px solid #e0e0e0;">'
-        f'<h4 style="margin: 0; color: #2c3e50; font-size: 16px; font-weight: 600;">{title}</h4>'
-        f'</div>'
-        f'<table style="width:100%; border-collapse:collapse; background: white;">'
-        f'<thead><tr style="background:#f8f9fa;"><th style="padding:12px 8px;border-bottom:2px solid #dee2e6;color:#495057;font-weight:600;text-align:left;font-size:13px;">Test</th><th style="padding:12px 8px;border-bottom:2px solid #dee2e6;color:#495057;font-weight:600;text-align:center;font-size:13px;">Result</th><th style="padding:12px 8px;border-bottom:2px solid #dee2e6;color:#495057;font-weight:600;text-align:center;font-size:13px;">Range</th><th style="padding:12px 8px;border-bottom:2px solid #dee2e6;color:#495057;font-weight:600;text-align:center;font-size:13px;">Level</th></tr></thead><tbody>'
-    )
-    for i, (label, value, ref) in enumerate(rows):
-        level, icon, bg = get_risk_color(value, ref)
-        row_bg = "#f8f9fa" if i % 2 == 0 else "white"
-        if level != "Normal":
-            row_bg = bg
-        
+        return ("Normal", "✅", colors.green)
+
+# Function to build table data for PDF
+def build_table_data(title, rows):
+    data = [[Paragraph(title, getSampleStyleSheet()['Heading2'])]]
+    data.append(["Test", "Result", "Range", "Level"])
+    for label, value, ref in rows:
+        level, icon, color = get_risk_color(value, ref)
         if "Count" in label or "Platelet" in label:
             value_str = f"{value:.0f}"
         else:
             value_str = f"{value:.2f}"
-            
-        html += f'<tr style="background:{row_bg};border-bottom:1px solid #e9ecef;"><td style="padding:10px 8px;color:#2c3e50;font-weight:500;">{label}</td><td style="padding:10px 8px;text-align:center;color:#2c3e50;font-weight:600;">{value_str}</td><td style="padding:10px 8px;text-align:center;color:#6c757d;font-size:12px;">{ref[0]} - {ref[1]}</td><td style="padding:10px 8px;text-align:center;font-weight:600;color:{color_map[level]};">{icon} {level}</td></tr>'
-    html += '</tbody></table></div>'
-    return html
-
-# Function to save the health report to PDF
+        data.append([
+            label,
+            value_str,
+            f"{ref[0]} - {ref[1]}",
+            Paragraph(f"{icon} {level}", ParagraphStyle('Custom', textColor=color))
+        ])
+    return data
+
+# Function to save the health report to PDF using reportlab
 def save_results_to_pdf(profile_image_base64, test_results, summary, patient_name, patient_age, patient_gender, patient_id, filename):
     try:
-        # Generate HTML content
-        current_date = datetime.now().strftime("%B %d, %Y %I:%M %p IST")
-        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>
-        """
+        # Create PDF document
+        pdf = SimpleDocTemplate(filename, pagesize=letter)
+        elements = []
 
-        # Debug: Print HTML length and sample
-        print(f"HTML generated, length: {len(health_card_html)}")
-        print(f"HTML sample: {health_card_html[:500]}...")  # Print first 500 chars for inspection
-
-        # Save HTML to a temporary file
-        html_temp_path = "/tmp/health_card.html"
-        try:
-            os.makedirs(os.path.dirname(html_temp_path), exist_ok=True)
-            with open(html_temp_path, "w", encoding="utf-8") as f:
-                f.write(health_card_html)
-            print(f"HTML saved to: {html_temp_path}")
-        except Exception as e:
-            print(f"Error saving HTML: {e}")
-            return f"Error: Failed to save HTML - {str(e)}", None
-
-        # Convert HTML to PDF using pdfkit with verbose logging
-        try:
-            pdfkit_options = {
-                "quiet": "",
-                "enable-local-file-access": "",  # Allow local file access for images
-                "margin-top": "0",
-                "margin-bottom": "0",
-                "margin-left": "0",
-                "margin-right": "0"
-            }
-            pdfkit.from_file(html_temp_path, filename, options=pdfkit_options)
-            print(f"PDF generated successfully at: {filename}")
-        except Exception as e:
-            print(f"Error converting to PDF: {str(e)}")
-            # Save fallback HTML with timestamp
-            fallback_path = f"/tmp/health_card_fallback_{datetime.now().strftime('%Y%m%d_%H%M%S')}.html"
-            with open(fallback_path, "w", encoding="utf-8") as f:
-                f.write(health_card_html)
-            print(f"Fallback HTML saved to: {fallback_path}")
-            return f"Error: Failed to generate PDF - {str(e)}. Fallback HTML at {fallback_path}", None
+        # Header
+        styles = getSampleStyleSheet()
+        current_date = datetime.now().strftime("%B %d, %Y %I:%M %p IST")
+        header_text = f"HEALTH CARD - Report Date: {current_date}"
+        if patient_id:
+            header_text += f" | Patient ID: {patient_id}"
+        elements.append(Paragraph(header_text, styles['Title']))
+        elements.append(Spacer(1, 12))
+
+        # Profile Section
+        profile_text = f"{patient_name if patient_name else 'Lab Test Results'}"
+        if patient_age and patient_gender:
+            profile_text += f" | Age: {patient_age} | Gender: {patient_gender}"
+        else:
+            profile_text += " | AI-Generated Health Analysis"
+        profile_text += " | Face-Based Health Analysis Report"
+        elements.append(Paragraph(profile_text, styles['Heading1']))
+        elements.append(Spacer(1, 12))
+
+        # Test Results Tables
+        for title, rows in [
+            ("Hematology", test_results['Hematology']),
+            ("Iron Panel", test_results['Iron Panel']),
+            ("Liver & Kidney", test_results['Liver & Kidney']),
+            ("Electrolytes", test_results['Electrolytes']),
+            ("Vitals", test_results['Vitals'])
+        ]:
+            table_data = build_table_data(title, rows)
+            table = Table(table_data, colWidths=[100, 70, 70, 70])
+            table.setStyle(TableStyle([
+                ('BACKGROUND', (0, 0), (-1, 0), colors.lightgrey),
+                ('TEXTCOLOR', (0, 0), (-1, 0), colors.black),
+                ('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.white),
+                ('TEXTCOLOR', (0, 1), (-1, -1), colors.black),
+                ('ALIGN', (0, 1), (-1, -1), 'LEFT'),
+                ('FONTNAME', (0, 1), (-1, -1), 'Helvetica'),
+                ('FONTSIZE', (0, 1), (-1, -1), 10),
+                ('GRID', (0, 0), (-1, -1), 1, colors.black),
+                ('VALIGN', (0, 0), (-1, -1), 'MIDDLE')
+            ]))
+            elements.append(table)
+            elements.append(Spacer(1, 12))
+
+        # Summary
+        elements.append(Paragraph("Summary & Recommendations", styles['Heading2']))
+        elements.append(Paragraph(summary, styles['Normal']))
+        elements.append(Spacer(1, 12))
+
+        # Build PDF
+        pdf.build(elements)
+        print(f"PDF generated successfully at: {filename}")
 
         # Move to /tmp for Gradio access
         temp_pdf_path = "/tmp/" + os.path.basename(filename)
-        try:
-            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 moving PDF: {e}")
-            return f"Error: Failed to move PDF - {str(e)}", None
+        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"Unexpected error in save_results_to_pdf: {str(e)}")
-        return f"Error: Unexpected failure - {str(e)}", None
+        print(f"Error generating PDF: {str(e)}")
+        return f"Error: Failed to generate PDF - {str(e)}", None
 
-# Build health card layout
+# Build health card layout (unchanged, using HTML for display)
 def build_health_card(profile_image, test_results, summary, pdf_filepath, patient_name="", patient_age="", patient_gender="", patient_id=""):
     current_date = datetime.now().strftime("%B %d, %Y %I:%M %p IST")
     
@@ -305,7 +254,7 @@ def build_health_card(profile_image, test_results, summary, pdf_filepath, patien
         /* Hide input sections during print */
         .gradio-container {{ display: block; }}
         /* Keep only the health card visible */
-        #health-card {{ display: block; }} 
+        #health-card {{ display: block; }}
     }}
     </style>
     """
@@ -366,23 +315,23 @@ def analyze_face(input_data):
     })
 
     test_results = {
-        "Hematology": build_table("🩸 Hematology", [("Hemoglobin", test_values["Hemoglobin"], (13.5, 17.5)),
-                                                  ("WBC Count", test_values["WBC Count"], (4.0, 11.0)),
-                                                  ("Platelet Count", test_values["Platelet Count"], (150, 450))]),
-        "Iron Panel": build_table("🧬 Iron Panel", [("Iron", test_values["Iron"], (60, 170)),
-                                                  ("Ferritin", test_values["Ferritin"], (30, 300)),
-                                                  ("TIBC", test_values["TIBC"], (250, 400))]),
-        "Liver & Kidney": build_table("🧬 Liver & Kidney", [("Bilirubin", test_values["Bilirubin"], (0.3, 1.2)),
-                                                          ("Creatinine", test_values["Creatinine"], (0.6, 1.2)),
-                                                          ("Urea", test_values["Urea"], (7, 20))]),
-        "Electrolytes": build_table("🧪 Electrolytes", [("Sodium", test_values["Sodium"], (135, 145)),
-                                                       ("Potassium", test_values["Potassium"], (3.5, 5.1))]),
-        "Vitals": build_table("❤️ Vitals", [("SpO2", test_values["SpO2"], (95, 100)),
-                                           ("Heart Rate", test_values["Heart Rate"], (60, 100)),
-                                           ("Respiratory Rate", test_values["Respiratory Rate"], (12, 20)),
-                                           ("Temperature", test_values["Temperature"], (97, 99)),
-                                           ("BP Systolic", test_values["BP Systolic"], (90, 120)),
-                                           ("BP Diastolic", test_values["BP Diastolic"], (60, 80))])
+        "Hematology": build_table_data("🩸 Hematology", [("Hemoglobin", test_values["Hemoglobin"], (13.5, 17.5)),
+                                                       ("WBC Count", test_values["WBC Count"], (4.0, 11.0)),
+                                                       ("Platelet Count", test_values["Platelet Count"], (150, 450))]),
+        "Iron Panel": build_table_data("🧬 Iron Panel", [("Iron", test_values["Iron"], (60, 170)),
+                                                       ("Ferritin", test_values["Ferritin"], (30, 300)),
+                                                       ("TIBC", test_values["TIBC"], (250, 400))]),
+        "Liver & Kidney": build_table_data("🧬 Liver & Kidney", [("Bilirubin", test_values["Bilirubin"], (0.3, 1.2)),
+                                                               ("Creatinine", test_values["Creatinine"], (0.6, 1.2)),
+                                                               ("Urea", test_values["Urea"], (7, 20))]),
+        "Electrolytes": build_table_data("🧪 Electrolytes", [("Sodium", test_values["Sodium"], (135, 145)),
+                                                            ("Potassium", test_values["Potassium"], (3.5, 5.1))]),
+        "Vitals": build_table_data("❤️ Vitals", [("SpO2", test_values["SpO2"], (95, 100)),
+                                                ("Heart Rate", test_values["Heart Rate"], (60, 100)),
+                                                ("Respiratory Rate", test_values["Respiratory Rate"], (12, 20)),
+                                                ("Temperature", test_values["Temperature"], (97, 99)),
+                                                ("BP Systolic", test_values["BP Systolic"], (90, 120)),
+                                                ("BP Diastolic", test_values["BP Diastolic"], (60, 80))])
     }
 
     summary = "<ul><li>Your hemoglobin is a bit low — this could mean mild anemia.</li><li>Low iron storage detected — consider an iron profile test.</li><li>Elevated bilirubin — possible jaundice. Recommend LFT.</li><li>High HbA1c — prediabetes indication. Recommend glucose check.</li><li>Low SpO₂ — suggest retesting with a pulse oximeter.</li></ul>"