Update app.py

app.py

@@ -48,7 +48,7 @@ from PIL import Image
 
 # ---- Friendly names (UI) ----
 MODEL_NAME_TBNET = "TBNet (CNN model)"
-MODEL_NAME_RADIO = "nvidia/C-RADIOv4-SO400M (visual model)"
+MODEL_NAME_RADIO = "Nvidia/C-RADIOv4-SO400M (visual model)"
 
 # ---- Default TB/Lung weights (HF-friendly relative paths) ----
 DEFAULT_TB_WEIGHTS = "weights/best.pt"
@@ -102,14 +102,14 @@ recommend **CBNAAT / GeneXpert**, sputum studies, and/or **CT chest** regardless
 # Friendly labels still map to GREEN/YELLOW/RED logic
 REPORT_LABELS = {
     "GREEN": {
-        "title": "LOW TB LIKELIHOOD",
+        "title": "LOW TB LIKELIHOOD / Pulmonary T.B not detected by A.I",
         "summary": (
             f"✅ **{MODEL_NAME_TBNET}** did not find patterns that strongly suggest pulmonary tuberculosis.\n\n"
             "**What to do next:** If symptoms or TB risk factors are present, please seek clinician/radiologist review."
         ),
     },
     "YELLOW": {
-        "title": "INDETERMINATE — REVIEW RECOMMENDED",
+        "title": "INDETERMINATE — REVIEW RECOMMENDED BY A RADIOLOGIST",
         "summary": (
             f"⚠️ **{MODEL_NAME_TBNET}** result is **not definitive**.\n\n"
             "**Common reasons:** image quality limitations, non-standard/cropped view, or non-focal attention.\n\n"
@@ -1288,78 +1288,120 @@ def build_ui():
 
             gr.Markdown(
                 f"""
-<div class="card">
-  <div style="font-size:16px; font-weight:900; margin-bottom:8px;">What this Space does</div>
-  <div style="opacity:0.92;">
-    This app analyzes chest X-rays for <b>TB-like patterns</b> and shows results from:
-    <ul>
-      <li><b>{MODEL_NAME_TBNET}</b> with <b>Grad-CAM</b> explainability</li>
-      <li><b>Auto Lung Mask</b> (Lung U-Net) + a <b>fail-safe</b> to prevent unsafe scoring on bad masks</li>
-      <li>Optional <b>{MODEL_NAME_RADIO}</b> with <b>RAW / MASKED</b> scoring and heatmaps</li>
-      <li>A final <b>consensus</b>: ✅ LOW / ⚠️ INDET / ⚠️ SCREEN+ / 🚩 TB+</li>
-    </ul>
+<style>
+  .wrap {{ max-width: 980px; margin: 0 auto; }}
+  .hero {{
+    padding: 14px 16px;
+    border-radius: 14px;
+    background: rgba(255,255,255,0.04);
+    border: 1px solid rgba(255,255,255,0.10);
+  }}
+  .hero h2 {{ margin: 0 0 6px 0; font-size: 18px; font-weight: 900; }}
+  .hero p {{ margin: 0; opacity: 0.9; font-size: 13px; line-height: 1.35; }}
+  .chips {{ margin-top: 10px; display: flex; flex-wrap: wrap; gap: 8px; }}
+  .chip {{
+    font-size: 12px; padding: 6px 10px; border-radius: 999px;
+    background: rgba(255,255,255,0.06);
+    border: 1px solid rgba(255,255,255,0.10);
+    opacity: 0.95;
+  }}
+  .warn {{
+    margin-top: 12px;
+    padding: 10px 12px;
+    border-left: 5px solid #f59e0b;
+    border-radius: 12px;
+    background: rgba(245,158,11,0.10);
+    font-size: 12.5px;
+    line-height: 1.35;
+  }}
+  details {{
+    margin-top: 12px;
+    padding: 10px 12px;
+    border-radius: 12px;
+    background: rgba(255,255,255,0.03);
+    border: 1px solid rgba(255,255,255,0.08);
+  }}
+  summary {{ cursor: pointer; font-weight: 800; font-size: 13px; }}
+  .muted {{ opacity: 0.85; font-size: 12.5px; line-height: 1.35; margin-top: 6px; }}
+  ul, ol {{ margin: 8px 0 0 18px; }}
+  li {{ margin: 4px 0; }}
+  .footer {{ margin-top: 10px; opacity: 0.70; font-size: 12px; }}
+</style>
+
+<div class="wrap">
+
+  <div class="hero">
+    <h2>TB X-ray screening assistant (research / decision support)</h2>
+    <p>
+      Upload chest X-rays to get an AI screening score, heatmaps, and a simple consensus output.
+      Designed to be safer on imperfect inputs (phone photos / WhatsApp / screenshots).
+    </p>
+
+    <div class="chips">
+      <span class="chip"><b>{MODEL_NAME_TBNET}</b> + Grad-CAM</span>
+      <span class="chip">Auto Lung Mask + fail-safe</span>
+      <span class="chip"><b>{MODEL_NAME_RADIO}</b> RAW / MASKED</span>
+      <span class="chip">Consensus: ✅ LOW · ⚠️ INDET · ⚠️ SCREEN+ · 🚩 TB+</span>
+    </div>
   </div>
-</div>
-
-<div class="warnbox">
-  <b>Clinical disclaimer:</b> Decision support only (not diagnostic). TB can be subtle (including miliary TB).
-  If TB is clinically suspected, pursue microbiology (CBNAAT/GeneXpert, sputum) and/or CT chest regardless of AI output.
-</div>
-
-<div class="card">
-  <div style="font-size:16px; font-weight:900; margin-bottom:8px;">Special feature: Phone / WhatsApp Mode</div>
-
-  Many users upload:
-  <ul>
-    <li><b>WhatsApp-forwarded X-rays</b> (compressed, low contrast)</li>
-    <li><b>Phone photos</b> of printed films or monitor screens (borders, glare, blur)</li>
-    <li><b>Screenshots</b> with large margins / UI elements</li>
-  </ul>
-
-  <b>Phone / WhatsApp Mode</b> is designed for these cases. When enabled, it applies:
-  <ul>
-    <li><b>Safe border crop</b> (reduces margins / screenshot framing)</li>
-    <li><b>Conditional CLAHE</b> (boosts local contrast when underexposed / low-detail)</li>
-    <li><b>Quality warnings</b> (blur, over/underexposure, heavy borders) to flag reduced reliability</li>
-  </ul>
-
-  <blockquote>
-    <b>Tip:</b> Enable Phone/WhatsApp Mode if your image is a phone photo, WhatsApp-forwarded, or has big borders / low contrast.<br/>
-    Keep it <b>OFF</b> for clean digital exports to avoid unnecessary preprocessing.
-  </blockquote>
-</div>
 
-<div class="card">
-  <div style="font-size:16px; font-weight:900; margin-bottom:8px;">Explainability & reliability</div>
-  <ul>
-    <li><b>Grad-CAM</b> (TBNet) highlights regions that influenced the TB score.</li>
-    <li><b>RADIO heatmaps</b> show where the visual model is focusing (RAW and sometimes MASKED).</li>
-    <li><b>Fail-safe</b>: if lung segmentation looks unreliable, TBNet scoring is disabled (shown as indeterminate).</li>
-    <li><b>Quality scoring</b> warns when results may be less reliable (blur, compression, non-standard view).</li>
-  </ul>
-</div>
-
-<div class="card">
-  <div style="font-size:16px; font-weight:900; margin-bottom:8px;">How to use</div>
-  <ol>
-    <li>Click <b>Continue</b> to open the interface.</li>
-    <li>Upload one or multiple X-ray images.</li>
-    <li>If your images come from <b>WhatsApp / phone camera / screenshots</b>, enable <b>Phone/WhatsApp Mode</b>.</li>
-    <li>(Optional) Enable <b>{MODEL_NAME_RADIO}</b> for a second independent model + heatmaps.</li>
-    <li>Click <b>Run Analysis</b>.</li>
-  </ol>
-</div>
+  <div class="warn">
+    <b>Clinical disclaimer:</b> Not diagnostic. TB can be subtle (incl. miliary TB).
+    If TB is clinically suspected, pursue microbiology (CBNAAT/GeneXpert, sputum) and/or CT chest
+    <b>regardless of AI output</b>.
+  </div>
 
-<div class="card">
-  <div style="font-size:16px; font-weight:900; margin-bottom:8px;">Privacy / processing note (HF Spaces)</div>
-  <div style="opacity:0.92;">
-    Images are processed by this Space runtime. Avoid uploading personally identifiable medical data.
-    Use anonymized images when possible.
+  <details>
+    <summary>Phone / WhatsApp Mode — when should I use it?</summary>
+    <div class="muted">
+      Use this for <b>WhatsApp-forwarded</b> images, <b>phone photos</b> of films/screens, or <b>screenshots</b> with borders.
+      It applies:
+      <ul>
+        <li><b>Safe border crop</b> (only when borders are detected)</li>
+        <li><b>Conditional CLAHE</b> (only when underexposed / low detail)</li>
+        <li><b>Quality warnings</b> (blur, over/underexposure, heavy borders)</li>
+      </ul>
+      <div class="muted"><b>Tip:</b> Keep it <b>OFF</b> for clean digital exports.</div>
+    </div>
+  </details>
+
+  <details>
+    <summary>Explainability & safety checks</summary>
+    <div class="muted">
+      <ul>
+        <li><b>Grad-CAM</b> highlights regions influencing TBNet.</li>
+        <li><b>RADIO heatmaps</b> show model attention (RAW and sometimes MASKED).</li>
+        <li><b>Fail-safe:</b> if lung masking looks unreliable, scoring is disabled → indeterminate.</li>
+        <li><b>Quality score</b> warns when reliability may be reduced.</li>
+      </ul>
+    </div>
+  </details>
+
+  <details>
+    <summary>How to use</summary>
+    <div class="muted">
+      <ol>
+        <li>Click <b>Continue</b> to open the interface.</li>
+        <li>Upload one or more X-ray images.</li>
+        <li>Enable <b>Phone/WhatsApp Mode</b> if your images are phone/WhatsApp/screenshot.</li>
+        <li>(Optional) Enable <b>{MODEL_NAME_RADIO}</b> for a second model + heatmaps.</li>
+        <li>Click <b>Run Analysis</b>.</li>
+      </ol>
+    </div>
+  </details>
+
+  <details>
+    <summary>Privacy / processing (HF Spaces)</summary>
+    <div class="muted">
+      Images are processed by the Space runtime. Avoid uploading personally identifiable medical data.
+      Prefer anonymized images when possible.
+    </div>
+  </details>
+
+  <div class="footer">
+    Device policy: <b>{DEVICE}</b> (FORCE_CPU={FORCE_CPU})
   </div>
-</div>
 
-<div class="subtitle">
-  Device policy: <b>{DEVICE}</b> (FORCE_CPU={FORCE_CPU})
 </div>
 """
             )