Fixed API error

api.py

@@ -20,7 +20,7 @@ def _load_image(image):
     return image
 
 
-def check_liveness(image) -> dict:
+def check_liveness(image):
     img = _load_image(image)
     if img is None:
         return {"error": "No image provided"}
@@ -43,3 +43,5 @@ def check_liveness(image) -> dict:
         return {"error": "Request timed out"}
     except requests.exceptions.RequestException as e:
         return {"error": f"API request failed: {str(e)}"}
+    except ValueError as e:
+        return {"error": f"Invalid response from server (not JSON): {str(e)}"}

ui.py

@@ -3,38 +3,74 @@ from api import check_liveness
 
 EXAMPLE_IMAGES = [[f"assets/{i}.jpg"] for i in range(1, 4)]
 
+THRESHOLD = 0.5
 
-def _format_result_html(result: dict) -> str:
-    if not result or "error" in result:
-        return ""
 
-    liveness_text = result.get("liveness result", result.get("liveness_result", "")).lower()
-    is_genuine = "genuine" in liveness_text or "live" in liveness_text
-    score = result.get("score", result.get("probability", 0))
-    if isinstance(score, str):
-        try:
-            score = float(score)
-        except (ValueError, TypeError):
-            score = 0
+def _normalize(d: dict) -> dict:
+    return {k.strip().replace(" ", "_"): v for k, v in d.items()}
+
 
+def _format_result_html(results: list) -> str:
+    if not results or (isinstance(results, dict) and "error" in results):
+        return ""
+
+    if isinstance(results, dict):
+        results = [results]
+
+    max_prob = 0
+    max_attack = ""
+    rows = ""
+    for item in results:
+        r = _normalize(item)
+        attack = r.get("attack_method", "Unknown")
+        prob = r.get("liveness_probability", 0)
+        score = r.get("liveness_score", 0)
+        quality = r.get("quality_score", 0)
+        state = r.get("state", "")
+
+        if isinstance(prob, str):
+            try:
+                prob = float(prob)
+            except (ValueError, TypeError):
+                prob = 0
+
+        if prob > max_prob:
+            max_prob = prob
+            max_attack = attack
+
+        pct = round(max(0, min(prob, 1)) * 100)
+        bar_color = "#059669" if prob < THRESHOLD else "#dc2626"
+
+        rows += f"""
+        <tr>
+            <td style="padding:8px 12px;font-weight:600;color:#334155;border-bottom:1px solid #f1f5f9;">
+                {attack}
+            </td>
+            <td style="padding:8px 12px;border-bottom:1px solid #f1f5f9;">
+                <div style="display:flex;align-items:center;gap:8px;">
+                    <div style="flex:1;background:#e2e8f0;border-radius:999px;height:8px;overflow:hidden;">
+                        <div style="width:{pct}%;height:100%;background:{bar_color};border-radius:999px;"></div>
+                    </div>
+                    <span style="font-weight:700;color:{bar_color};min-width:40px;text-align:right;">{pct}%</span>
+                </div>
+            </td>
+            <td style="padding:8px 12px;color:#64748b;border-bottom:1px solid #f1f5f9;text-align:center;">
+                {state}
+            </td>
+        </tr>"""
+
+    is_genuine = max_prob < THRESHOLD
     if is_genuine:
         badge_text = "✅  GENUINE / LIVE"
         badge_color = "#059669"
         bg_color = "#ecfdf5"
         border_color = "#a7f3d0"
     else:
-        badge_text = "❌  SPOOF / FAKE"
+        badge_text = f"❌  SPOOF / FAKE — {max_attack}"
         badge_color = "#dc2626"
         bg_color = "#fef2f2"
         border_color = "#fecaca"
 
-    detail_rows = "".join(
-        f'<tr><td style="padding:6px 12px;font-weight:600;color:#64748b;'
-        f'border-bottom:1px solid #f1f5f9;">{k}</td>'
-        f'<td style="padding:6px 12px;color:#334155;border-bottom:1px solid #f1f5f9;">{v}</td></tr>'
-        for k, v in result.items()
-    )
-
     return f"""
     <div style="padding:12px 0;">
         <div style="background:{bg_color};border:2px solid {border_color};
@@ -49,7 +85,16 @@ def _format_result_html(result: dict) -> str:
             </div>
             <div style="margin-top:16px;background:white;border:1px solid #e2e8f0;border-radius:12px;overflow:hidden;">
                 <table style="width:100%;border-collapse:collapse;">
-                    {detail_rows}
+                    <thead>
+                        <tr style="background:#f8fafc;">
+                            <th style="padding:8px 12px;text-align:left;font-weight:600;color:#64748b;">Attack Method</th>
+                            <th style="padding:8px 12px;text-align:left;font-weight:600;color:#64748b;">Probability</th>
+                            <th style="padding:8px 12px;text-align:center;font-weight:600;color:#64748b;">Status</th>
+                        </tr>
+                    </thead>
+                    <tbody>
+                        {rows}
+                    </tbody>
                 </table>
             </div>
         </div>
@@ -66,7 +111,7 @@ def process_image(image):
         )
 
     result = check_liveness(image)
-    if "error" in result:
+    if isinstance(result, dict) and "error" in result:
         return (
             f'<div style="padding:20px;text-align:center;color:#dc2626;">{result["error"]}</div>',
             result,