Update app.py

app.py

@@ -25,40 +25,46 @@ def predict_ckd(
     platelet_count, polymorphs, rbc_count, rdw_sd,
     wbc_count, eosinophils, creatinine, urea, gfr
 ):
-    values = [
-        sex, age, haemoglobin, hematocrit, lymphocytes,
-        mch, mchc, mcv, monocytes, mpv, pct, pdw,
-        platelet_count, polymorphs, rbc_count, rdw_sd,
-        wbc_count, eosinophils, creatinine, urea, gfr
-    ]
-    features = pd.DataFrame([values], columns=FEATURE_ORDER)
-
-    pred = model.predict(features)[0]
-    proba = model.predict_proba(features)[0]
-
-    ckd_prob    = round(float(proba[1]) * 100, 2)
-    no_ckd_prob = round(float(proba[0]) * 100, 2)
-
-    if pred == 1:
-        label   = "⚠️ CKD Detected"
-        details = (
-            f"**Prediction: Chronic Kidney Disease (CKD)**\n\n"
-            f"- CKD Probability: **{ckd_prob}%**\n"
-            f"- No CKD Probability: {no_ckd_prob}%\n\n"
-            f"> ⚠️ This result suggests the presence of Chronic Kidney Disease. "
-            f"Please consult a qualified nephrologist for clinical evaluation."
-        )
-    else:
-        label   = "✅ No CKD Detected"
-        details = (
-            f"**Prediction: No Chronic Kidney Disease**\n\n"
-            f"- No CKD Probability: **{no_ckd_prob}%**\n"
-            f"- CKD Probability: {ckd_prob}%\n\n"
-            f"> ✅ Based on the provided blood parameters, no CKD pattern was detected. "
-            f"Regular health check-ups are still recommended."
-        )
-
-    return label, details
+    try:
+        values = [
+            sex, age, haemoglobin, hematocrit, lymphocytes,
+            mch, mchc, mcv, monocytes, mpv, pct, pdw,
+            platelet_count, polymorphs, rbc_count, rdw_sd,
+            wbc_count, eosinophils, creatinine, urea, gfr
+        ]
+        features = pd.DataFrame([values], columns=FEATURE_ORDER)
+
+        pred = model.predict(features)[0]
+        proba = model.predict_proba(features)[0]
+
+        ckd_prob    = round(float(proba[1]) * 100, 2)
+        no_ckd_prob = round(float(proba[0]) * 100, 2)
+
+        if pred == 1:
+            label   = "⚠️ CKD Detected"
+            details = (
+                f"**Prediction: Chronic Kidney Disease (CKD)**\n\n"
+                f"- CKD Probability: **{ckd_prob}%**\n"
+                f"- No CKD Probability: {no_ckd_prob}%\n\n"
+                f"> ⚠️ This result suggests the presence of Chronic Kidney Disease. "
+                f"Please consult a qualified nephrologist for clinical evaluation."
+            )
+        else:
+            label   = "✅ No CKD Detected"
+            details = (
+                f"**Prediction: No Chronic Kidney Disease**\n\n"
+                f"- No CKD Probability: **{no_ckd_prob}%**\n"
+                f"- CKD Probability: {ckd_prob}%\n\n"
+                f"> ✅ Based on the provided blood parameters, no CKD pattern was detected. "
+                f"Regular health check-ups are still recommended."
+            )
+
+        return label, details
+
+    except Exception as e:
+        import traceback
+        err = traceback.format_exc()
+        return "❌ Prediction Error", f"```\n{err}\n```"
 
 # ── UI ────────────────────────────────────────────────────────────────────────
 DESCRIPTION = """