Update app.py

app.py

@@ -9,25 +9,17 @@ import base64
 import joblib
 from datetime import datetime
 import shutil
-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
+from reportlab.lib import colors
 import atexit
+import glob
 
 # Cleanup temporary files on exit
 def cleanup_temp_files():
-    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("health_card*.html"):
-        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_Report_*.pdf"):
+        os.remove(temp_file)
 atexit.register(cleanup_temp_files)
 
 # Initialize the face mesh model
@@ -98,107 +90,120 @@ models = {
 def get_risk_color(value, normal_range):
     low, high = normal_range
     if value < low:
-        return ("Low", "🔻", colors.yellow)
+        return ("Low", "🔻", "#fff3cd")
     elif value > high:
-        return ("High", "🔺", colors.red)
+        return ("High", "🔺", "#f8d7da")
     else:
-        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)
+        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
+        
         if "Count" in label or "Platelet" in label:
             value_str = f"{value:.0f}"
         else:
             value_str = f"{value:.2f}"
-        data.append([
-            label,
-            value_str,
-            f"{ref[0]} - {ref[1]}",
-            Paragraph(f"{icon} {level}", ParagraphStyle('Custom', textColor=color))
-        ])
-    return data
+            
+        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 using reportlab
-def save_results_to_pdf(profile_image_base64, test_results, summary, patient_name, patient_age, patient_gender, patient_id, filename):
+# Function to save the health report to PDF
+def save_results_to_pdf(test_results, test_values, filename):
     try:
-        # Create PDF document
-        pdf = SimpleDocTemplate(filename, pagesize=letter)
-        elements = []
-
-        # Header
+        doc = SimpleDocTemplate(filename, pagesize=letter)
         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])
+        title_style = ParagraphStyle(
+            name='Title',
+            fontSize=16,
+            leading=20,
+            alignment=1,
+            spaceAfter=20,
+            textColor=colors.black,
+            fontName='Helvetica-Bold'
+        )
+        body_style = ParagraphStyle(
+            name='Body',
+            fontSize=12,
+            leading=14,
+            spaceAfter=10,
+            textColor=colors.black,
+            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)
+        }
+        
+        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])
+            table = Table(table_data)
             table.setStyle(TableStyle([
-                ('BACKGROUND', (0, 0), (-1, 0), colors.lightgrey),
-                ('TEXTCOLOR', (0, 0), (-1, 0), colors.black),
+                ('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.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')
+                ('BACKGROUND', (0, 1), (-1, -1), colors.beige),
+                ('GRID', (0, 0), (-1, -1), 1, colors.black)
             ]))
-            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)
-        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
+            flowables.append(table)
+            flowables.append(Spacer(1, 12))
+        
+        doc.build(flowables)
+        return f"PDF saved successfully as {filename}", filename
     except Exception as e:
-        print(f"Error generating PDF: {str(e)}")
-        return f"Error: Failed to generate PDF - {str(e)}", None
+        return f"Error saving PDF: {str(e)}", None
 
-# Build health card layout (unchanged, using HTML for display)
+# Build health card layout
 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")
+    from datetime import datetime
+    current_date = datetime.now().strftime("%B %d, %Y")
     
     pdf_filename = os.path.basename(pdf_filepath) if pdf_filepath else "health_report.pdf"
     
@@ -241,6 +246,7 @@ 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>
@@ -265,7 +271,7 @@ current_patient_details = {'name': '', 'age': '', 'gender': '', 'id': ''}
 
 # Modified analyze_face function
 def analyze_face(input_data):
-    if isinstance(input_data, str):  # Video input (file path in Spaces)
+    if isinstance(input_data, str):  # Video input (file path in Replit)
         cap = cv2.VideoCapture(input_data)
         if not cap.isOpened():
             return "<div style='color:red;'>⚠️ Error: Could not open video.</div>", None
@@ -315,23 +321,23 @@ def analyze_face(input_data):
     })
 
     test_results = {
-        "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))])
+        "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))])
     }
 
     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>"
@@ -340,16 +346,7 @@ 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(
-        profile_image_base64,
-        test_results,
-        summary,
-        current_patient_details['name'],
-        current_patient_details['age'],
-        current_patient_details['gender'],
-        current_patient_details['id'],
-        pdf_filename
-    )
+    pdf_result, pdf_filepath = save_results_to_pdf(test_results, test_values, pdf_filename)
     
     if pdf_filepath:
         temp_pdf_path = "/tmp/" + os.path.basename(pdf_filepath)
@@ -390,7 +387,7 @@ def route_inputs(mode, image, video, patient_name, patient_age, patient_gender,
 
 # Create Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("""# 🧠 Face-Based Lab Test AI Report""")
+    gr.Markdown("""# 🧠 Face-Based Lab Test AI Report (Video Mode)""")
     with gr.Row():
         with gr.Column():
             gr.Markdown("### Patient Information")
@@ -408,12 +405,12 @@ with gr.Blocks() as demo:
                                    sources=["upload", "webcam"])
             submit_btn = gr.Button("🔍 Analyze")
         with gr.Column():
-            result_html = gr.HTML(label="🧪 Health Report")
+            result_html = gr.HTML(label="🧪 Health Report Table")
             result_pdf = gr.File(label="Download Health Report PDF", interactive=False)
 
             submit_btn.click(fn=route_inputs,
                              inputs=[mode_selector, image_input, video_input, patient_name, patient_age, patient_gender, patient_id],
                              outputs=[result_html, result_pdf])
 
-# Launch Gradio for Hugging Face Spaces
-demo.launch()
+# Launch Gradio for Replit
+demo.launch(server_name="0.0.0.0", server_port=7860)