Update app.py

app.py

@@ -137,6 +137,7 @@ def analyze_face(image):
     features = extract_features(frame_rgb, landmarks)
     test_values = {}
     r2_scores = {}
+
     for label in models:
         if label == "Hemoglobin":
             prediction = models[label].predict([features])[0]
@@ -146,6 +147,7 @@ def analyze_face(image):
             value = models[label].predict([[random.uniform(0.2, 0.5) for _ in range(7)]])[0]
             test_values[label] = value
             r2_scores[label] = 0.0  # simulate other 7D inputs
+
     gray = cv2.cvtColor(frame_rgb, cv2.COLOR_RGB2GRAY)
     green_std = np.std(frame_rgb[:, :, 1]) / 255
     brightness_std = np.std(gray) / 255
@@ -158,17 +160,32 @@ def analyze_face(image):
     spo2_features = [heart_rate, brightness_variation, skin_tone_index]
     spo2 = spo2_model.predict([spo2_features])[0]
     rr = int(12 + abs(heart_rate % 5 - 2))
-    html_output = "".join([  
-        f'<div style="font-size:14px;color:#888;margin-bottom:10px;">Hemoglobin R² Score: {r2_scores.get("Hemoglobin", "NA"):.2f}</div>',
-        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))]),
-        build_table("🧬 Iron Panel", [("Iron", test_values["Iron"], (60, 170)), ("Ferritin", test_values["Ferritin"], (30, 300)), ("TIBC", test_values["TIBC"], (250, 400))]),
-        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))]),
-        build_table("🧪 Electrolytes", [("Sodium", test_values["Sodium"], (135, 145)), ("Potassium", test_values["Potassium"], (3.5, 5.1))]),
-        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))])
-    ])
+
+    # Prepare the test results
+    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))])
+    }
+
     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>"
-    
-    health_card_html = build_health_card("profile_image_placeholder_base64", html_output, summary)
+
+    # Generate Health Card HTML
+    health_card_html = build_health_card("profile_image_placeholder_base64", test_results, summary)
     return health_card_html
 
 # Create Gradio interface