Update app.py

app.py

@@ -1,5 +1,5 @@
 # Face Detection-Based AI Automation of Lab Tests
-# Redesigned UI using Gradio Blocks + HTML Cards
+# Redesigned UI using Gradio Blocks + HTML Cards (Fixed .then() error)
 
 import gradio as gr
 import cv2
@@ -28,20 +28,20 @@ def estimate_spo2_rr(heart_rate):
 def get_risk_color(value, normal_range):
     low, high = normal_range
     if value < low:
-        return ("🔻 LOW", "#FFCCCC")  # Red background
+        return ("🔻 LOW", "#FFCCCC")
     elif value > high:
-        return ("🔺 HIGH", "#FFE680")  # Yellow background
+        return ("🔺 HIGH", "#FFE680")
     else:
-        return ("✅ Normal", "#CCFFCC")  # Green background
+        return ("✅ Normal", "#CCFFCC")
 
 def analyze_face(image):
     if image is None:
-        return [], None
+        return "<div style='color:red;'>⚠️ Error: No image provided.</div>", None
 
     frame_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     result = face_mesh.process(frame_rgb)
     if not result.multi_face_landmarks:
-        return [["Face not detected", "#FFDDDD"]], None
+        return "<div style='color:red;'>⚠️ Error: Face not detected.</div>", None
 
     landmarks = result.multi_face_landmarks[0].landmark
     heart_rate = estimate_heart_rate(frame_rgb, landmarks)
@@ -62,36 +62,17 @@ def analyze_face(image):
         html += '</div>'
         return html
 
-    cards = [
-        section("🩸 Hematology", [
-            ("Hemoglobin", hb, (13.5, 17.5)),
-            ("WBC Count", wbc, (4.0, 11.0)),
-            ("Platelets", platelets, (150, 450))
-        ]),
-        section("🧬 Iron & Liver Panel", [
-            ("Iron", iron, (60, 170)),
-            ("Ferritin", ferritin, (30, 300)),
-            ("TIBC", tibc, (250, 400)),
-            ("Bilirubin", bilirubin, (0.3, 1.2))
-        ]),
-        section("🧪 Kidney, Thyroid & Stress", [
-            ("Creatinine", creatinine, (0.6, 1.2)),
-            ("TSH", tsh, (0.4, 4.0)),
-            ("Cortisol", cortisol, (5, 25))
-        ]),
-        section("🧁 Metabolic Panel", [
-            ("Fasting Blood Sugar", fbs, (70, 110)),
-            ("HbA1c", hba1c, (4.0, 5.7))
-        ]),
-        section("❤️ Vital Signs", [
-            ("SpO2", spo2, (95, 100)),
-            ("Heart Rate", heart_rate, (60, 100)),
-            ("Respiratory Rate", rr, (12, 20))
-        ])
-    ]
-    return cards, frame_rgb
-
-# Gradio App Layout (Custom UI with Cards)
+    report = "".join([
+        section("🩸 Hematology", [("Hemoglobin", hb, (13.5, 17.5)), ("WBC Count", wbc, (4.0, 11.0)), ("Platelets", platelets, (150, 450))]),
+        section("🧬 Iron & Liver Panel", [("Iron", iron, (60, 170)), ("Ferritin", ferritin, (30, 300)), ("TIBC", tibc, (250, 400)), ("Bilirubin", bilirubin, (0.3, 1.2))]),
+        section("🧪 Kidney, Thyroid & Stress", [("Creatinine", creatinine, (0.6, 1.2)), ("TSH", tsh, (0.4, 4.0)), ("Cortisol", cortisol, (5, 25))]),
+        section("🧁 Metabolic Panel", [("Fasting Blood Sugar", fbs, (70, 110)), ("HbA1c", hba1c, (4.0, 5.7))]),
+        section("❤️ Vital Signs", [("SpO2", spo2, (95, 100)), ("Heart Rate", heart_rate, (60, 100)), ("Respiratory Rate", rr, (12, 20))])
+    ])
+
+    return report, frame_rgb
+
+# Gradio App Layout (Fixed HTML output handling)
 demo = gr.Blocks()
 with demo:
     gr.Markdown("""
@@ -107,17 +88,10 @@ with demo:
             result_html = gr.HTML(label="🧪 Visual Diagnostic Cards")
             result_image = gr.Image(label="📷 Face Scan Annotated")
 
-    def format_html(cards):
-        return "".join(cards)
-
     submit_btn.click(
         fn=analyze_face,
         inputs=image_input,
         outputs=[result_html, result_image]
-    ).then(
-        fn=format_html,
-        inputs=None,
-        outputs=result_html
     )
 
 demo.launch()