Require review for low confidence medicine matches

gradio_pharmacopilot_demo.py

@@ -53,6 +53,7 @@ MODEL_ID = os.getenv("PHARMACOPILOT_MODEL_ID", "openbmb/MiniCPM-V-4_5")
 LIVE_GPU_OCR = os.getenv("PHARMACOPILOT_LIVE_GPU_OCR", "1").lower() not in {"0", "false", "no"}
 DEMO_OCR_TEXT = "Neuoxen"
 DEMO_PROMPT = "Read the handwritten medicine name in the image. Return only the text."
+ACCEPTANCE_THRESHOLD = int(os.getenv("PHARMACOPILOT_ACCEPTANCE_THRESHOLD", "75"))
 OCR_MODEL = None
 OCR_TOKENIZER = None
 
@@ -277,7 +278,7 @@ def pipeline_html(stage: int = 0) -> str:
         ("Prescription", "captured"),
         ("MiniCPM OCR", "completed"),
         ("Retrieval Engine", "searched"),
-        ("Nemotron Nano 9B", "validated"),
+        ("Validation", "checked"),
         ("Medicine Found", "ready"),
     ]
     cards = []
@@ -310,30 +311,51 @@ def medicine_details_html(
     display_name: str,
     confidence: int,
 ) -> str:
+    accepted = confidence >= ACCEPTANCE_THRESHOLD
+    medicine_label = display_name if accepted else "Needs pharmacist review"
+    generic_label = medicine.get("name", "Unknown") if accepted else f"Suggestion: {medicine.get('name', 'Unknown')}"
+    strength_label = first_strength(medicine.get("strength", "")) if accepted else "Not confirmed"
+    manufacturer_label = (medicine.get("manufacturer") or "Not listed") if accepted else "Not confirmed"
+    category_label = medicine.get("category", "General") if accepted else "Not confirmed"
+    price_label = "PKR 145" if accepted else "Not confirmed"
+    validation_label = "Pending review" if not accepted else "Accepted by retrieval threshold"
+    inventory_label = (
+        f"Shelf {inventory['shelf']}, row {inventory['row']}"
+        if accepted
+        else "Hidden until a confident medicine match is available"
+    )
     return f"""
     <div class="result-card">
       <h3>Prescription Details</h3>
       <dl class="details">
-        <dt>Medicine</dt><dd>{display_name}</dd>
-        <dt>Generic</dt><dd>{medicine.get('name', 'Unknown')}</dd>
-        <dt>Strength</dt><dd>{first_strength(medicine.get('strength', ''))}</dd>
-        <dt>Manufacturer</dt><dd>{medicine.get('manufacturer') or 'Not listed'}</dd>
+        <dt>Medicine</dt><dd>{medicine_label}</dd>
+        <dt>Generic</dt><dd>{generic_label}</dd>
+        <dt>Strength</dt><dd>{strength_label}</dd>
+        <dt>Manufacturer</dt><dd>{manufacturer_label}</dd>
         <dt>Confidence</dt><dd>{confidence}%</dd>
-        <dt>Category</dt><dd>{medicine.get('category', 'General')}</dd>
-        <dt>Price</dt><dd>PKR 145</dd>
+        <dt>Category</dt><dd>{category_label}</dd>
+        <dt>Price</dt><dd>{price_label}</dd>
       </dl>
       <div class="explain">
         <h4>AI Explanation</h4>
         <p><b>OCR detected:</b> "{ocr_text}"</p>
-        <p><b>Retrieved:</b> {display_name}</p>
-        <p><b>Validated using:</b> Nemotron Nano 9B</p>
-        <p><b>Inventory:</b> Shelf {inventory['shelf']}, row {inventory['row']}</p>
+        <p><b>Retrieved:</b> {display_name} ({medicine.get('name', 'Unknown')})</p>
+        <p><b>Validation:</b> {validation_label}</p>
+        <p><b>Inventory:</b> {inventory_label}</p>
       </div>
     </div>
     """
 
 
-def package_status_html(inventory: dict[str, Any]) -> str:
+def package_status_html(inventory: dict[str, Any], accepted: bool = True) -> str:
+    if not accepted:
+        return """
+        <div class="stock-card">
+          <div><span>Available Stock</span><strong>-</strong></div>
+          <div><span>Status</span><strong class="bad">Needs Review</strong></div>
+          <div><span>Shelf</span><strong>Hidden</strong></div>
+        </div>
+        """
     status = "In Stock" if inventory.get("quantity", 0) > 0 else "Out of Stock"
     dot_class = "ok" if inventory.get("quantity", 0) > 0 else "bad"
     return f"""
@@ -360,11 +382,12 @@ def candidates_html(candidates: list[dict[str, Any]]) -> str:
     """
 
 
-def ocr_compare_html(medicine: dict[str, Any], ocr_text: str, display_name: str) -> str:
+def ocr_compare_html(medicine: dict[str, Any], ocr_text: str, display_name: str, confidence: int) -> str:
+    corrected = display_name if confidence >= ACCEPTANCE_THRESHOLD else f"Needs review: {display_name}"
     return f"""
     <div class="compare-grid">
       <div><span>OCR Output</span><strong>{ocr_text}</strong></div>
-      <div><span>AI Corrected</span><strong>{display_name}</strong></div>
+      <div><span>AI Corrected</span><strong>{corrected}</strong></div>
       <div><span>Canonical</span><strong>{medicine['name']}</strong></div>
     </div>
     """
@@ -433,21 +456,23 @@ def analyze_prescription(image, progress=gr.Progress()):
 
     for pct, label in [
         (0.70, "Retrieval search over medicine aliases"),
-        (0.88, "Nemotron validation"),
-        (1.00, "Product identified"),
+        (0.88, "Validation check"),
+        (1.00, "Result prepared"),
     ]:
         progress(pct, desc=label)
         time.sleep(0.25)
 
     medicine, candidates, display_name, confidence = find_medicine_from_ocr(ocr_text)
+    accepted = confidence >= ACCEPTANCE_THRESHOLD
     inventory = get_inventory(medicine)
     image_path = resolve_asset_path(medicine.get("image_path"))
-    package_image = str(image_path) if image_path else None
+    package_image = str(image_path) if image_path and accepted else None
 
     state = {
         "medicine_id": medicine["id"],
         "medicine_name": medicine["name"],
         "display_name": display_name,
+        "accepted": accepted,
         "shelf": inventory["shelf"],
         "row": inventory["row"],
     }
@@ -458,12 +483,12 @@ def analyze_prescription(image, progress=gr.Progress()):
         pipeline_html(5),
         medicine_details_html(medicine, inventory, ocr_text, display_name, confidence),
         package_image,
-        package_status_html(inventory),
+        package_status_html(inventory, accepted),
         confidence_gauge(confidence),
         candidates_html(candidates),
-        ocr_compare_html(medicine, ocr_text, display_name),
-        gr.update(visible=True),
+        ocr_compare_html(medicine, ocr_text, display_name, confidence),
         gr.update(visible=True),
+        gr.update(visible=True, interactive=accepted),
         state,
     )