Update app.py

app.py

@@ -13,6 +13,14 @@ 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("/tmp/Health_Report_*.pdf"):
+        os.remove(temp_file)
+atexit.register(cleanup_temp_files)
 
 # Initialize the face mesh model
 mp_face_mesh = mp.solutions.face_mesh
@@ -104,7 +112,6 @@ def build_table(title, rows):
         if level != "Normal":
             row_bg = bg
         
-        # Format the value with appropriate units
         if "Count" in label or "Platelet" in label:
             value_str = f"{value:.0f}"
         else:
@@ -117,16 +124,13 @@ def build_table(title, rows):
 # Function to save the health report to PDF
 def save_results_to_pdf(test_results, filename):
     try:
-        # Create a PDF document
         doc = SimpleDocTemplate(filename, pagesize=letter)
         styles = getSampleStyleSheet()
-        
-        # Define custom styles
         title_style = ParagraphStyle(
             name='Title',
             fontSize=16,
             leading=20,
-            alignment=1,  # Center
+            alignment=1,
             spaceAfter=20,
             textColor=colors.black,
             fontName='Helvetica-Bold'
@@ -140,19 +144,49 @@ def save_results_to_pdf(test_results, filename):
             fontName='Helvetica'
         )
         
-        # Build the PDF content
-        flowables = []
-        
-        # Add title
-        flowables.append(Paragraph("Health Report", title_style))
+        flowables = [Paragraph("Health Report", title_style), Spacer(1, 12)]
         
-        # Add test results to the report
-        for label, value in test_results.items():
-            line = f"{label}: {value}"
-            flowables.append(Paragraph(line, body_style))
+        test_values = {
+            "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, value in test_values.items():
+                if any(label in html for section_html in test_results.values()):
+                    simulated_value = test_values[label][0] + random.uniform(-1, 1)
+                    level, _, _ = get_risk_color(simulated_value, value)
+                    table_data.append([label, f"{simulated_value:.2f}", f"{value[0]} - {value[1]}", 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)
+            ]))
+            flowables.append(table)
             flowables.append(Spacer(1, 12))
         
-        # Build the PDF
         doc.build(flowables)
         return f"PDF saved successfully as {filename}", filename
     except Exception as e:
@@ -163,7 +197,6 @@ def build_health_card(profile_image, test_results, summary, pdf_filepath, patien
     from datetime import datetime
     current_date = datetime.now().strftime("%B %d, %Y")
     
-    # Use a relative path for the download link to work in Gradio
     pdf_filename = os.path.basename(pdf_filepath) if pdf_filepath else "health_report.pdf"
     
     html = f"""
@@ -182,8 +215,8 @@ def build_health_card(profile_image, test_results, summary, pdf_filepath, patien
             <div style="display: flex; align-items: center;">
                 <img src="data:image/png;base64,{profile_image}" alt="Profile" style="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);">
                 <div>
-                    <h2 style="margin: 0; font-size: 28px; color: #2c3e50; font-weight: 700;">{patient_name if patient_name else "Lab Test Results"}</h2>
-                    <p style="margin: 4px 0 0 0; color: #666; font-size: 14px;">{f"Age: {patient_age} | Gender: {patient_gender}" if patient_age and patient_gender else "AI-Generated Health Analysis"}</p>
+                    <h2 style="margin: 0; font-size: 28px; color: #2c3e50; font-weight: 700;">{patient_name if patient_name else 'Lab Test Results'}</h2>
+                    <p style="margin: 4px 0 0 0; color: #666; font-size: 14px;">{f'Age: {patient_age} | Gender: {patient_gender}' if patient_age and patient_gender else 'AI-Generated Health Analysis'}</p>
                     <p style="margin: 4px 0 0 0; color: #888; font-size: 12px;">Face-Based Health Analysis Report</p>
                 </div>
             </div>
@@ -213,6 +246,14 @@ def build_health_card(profile_image, test_results, summary, pdf_filepath, patien
             </button>
         </div>
     </div>
+    <style>
+    @media print {
+        /* Hide input sections during print */
+        .gradio-container { display: block; }
+        /* Keep only the health card visible */
+        #health-card { display: block; }
+    }
+    </style>
     """
     return html
 
@@ -235,7 +276,7 @@ def analyze_face(input_data):
             return "<div style='color:red;'>⚠️ Error: No image provided.</div>", None
 
     # Resize image to reduce processing time
-    frame = cv2.resize(frame, (640, 480))  # Adjust resolution for Replit
+    frame = cv2.resize(frame, (640, 480))
     frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
     result = face_mesh.process(frame_rgb)
     if not result.multi_face_landmarks:
@@ -243,17 +284,13 @@ def analyze_face(input_data):
     landmarks = result.multi_face_landmarks[0].landmark
     features = extract_features(frame_rgb, landmarks)
     test_values = {}
-    r2_scores = {}
-
     for label in models:
         if label == "Hemoglobin":
             prediction = models[label].predict([features])[0]
             test_values[label] = prediction
-            r2_scores[label] = 0.385
         else:
             value = models[label].predict([[random.uniform(0.2, 0.5) for _ in range(7)]])[0]
             test_values[label] = value
-            r2_scores[label] = 0.0
 
     gray = cv2.cvtColor(frame_rgb, cv2.COLOR_RGB2GRAY)
     green_std = np.std(frame_rgb[:, :, 1]) / 255
@@ -269,33 +306,23 @@ def analyze_face(input_data):
     rr = int(12 + abs(heart_rate % 5 - 2))
 
     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", spo2, (95, 100)),
-                     ("Heart Rate", heart_rate, (60, 100)),
-                     ("Respiratory Rate", rr, (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", spo2, (95, 100)),
+                                           ("Heart Rate", heart_rate, (60, 100)),
+                                           ("Respiratory Rate", rr, (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>"
@@ -303,28 +330,27 @@ def analyze_face(input_data):
     _, buffer = cv2.imencode('.png', frame_rgb)
     profile_image_base64 = base64.b64encode(buffer).decode('utf-8')
 
-    # Generate PDF and return for download
     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, pdf_filename)
     
     if pdf_filepath:
-        # Copy the PDF to a temporary directory for Gradio to serve it
         temp_pdf_path = "/tmp/" + os.path.basename(pdf_filepath)
         shutil.copy(pdf_filepath, temp_pdf_path)
-        
-        # Pass pdf_filepath to build_health_card
-        health_card_html = build_health_card(
-            profile_image_base64, 
-            test_results,
-            summary, 
-            temp_pdf_path,  # Pass the PDF path
-            current_patient_details['name'],
-            current_patient_details['age'],
-            current_patient_details['gender'],
-            current_patient_details['id']
-        )
-        
-        return health_card_html, temp_pdf_path
+        if os.path.exists(temp_pdf_path) and os.access(temp_pdf_path, os.R_OK):
+            health_card_html = build_health_card(
+                profile_image_base64,
+                test_results,
+                summary,
+                temp_pdf_path,
+                current_patient_details['name'],
+                current_patient_details['age'],
+                current_patient_details['gender'],
+                current_patient_details['id']
+            )
+            return health_card_html, temp_pdf_path
+        else:
+            return "<div style='color:red;'>⚠️ Error: PDF file not accessible.</div>", None
+    return "<div style='color:red;'>⚠️ Error: Failed to generate PDF.</div>", None
 
 # Modified route_inputs function
 def route_inputs(mode, image, video, patient_name, patient_age, patient_gender, patient_id):
@@ -333,11 +359,10 @@ def route_inputs(mode, image, video, patient_name, patient_age, patient_gender,
     if mode == "Video" and video is None:
         return "<div style='color:red;'>⚠️ Error: No video provided.</div>", None
     
-    # Store patient details globally for use in analyze_face
     global current_patient_details
     current_patient_details = {
         'name': patient_name,
-        'age': patient_age, 
+        'age': patient_age,
         'gender': patient_gender,
         'id': patient_id
     }