"""
Independent dual-grader review of already-inserted discharge-letter cases.
Loads stored letters + model outputs, assigns each case to exactly one of
Grader 1 / Grader 2, and saves grades without mixing the two raters.
"""
from __future__ import annotations
import json
import os
import re
import tempfile
from datetime import datetime, timezone
from pathlib import Path
import streamlit as st
from dotenv import load_dotenv
APP_DIR = Path(__file__).resolve().parent
def _detect_repo_dir() -> Path:
here = APP_DIR
for candidate in (here, here.parent):
if (candidate / "feedback_data.json").exists():
return candidate
if (candidate / "app.py").exists() and (candidate / "backend.py").exists():
return candidate
if here.name in {"pages", "grading_space"}:
return here.parent
return here
REPO_DIR = _detect_repo_dir()
load_dotenv(REPO_DIR / ".env")
HF_DATASET_REPO = "Vrda/im-error-check-data"
HF_CASES_FILE = "feedback_data.json"
HF_GRADES_FILE = "independent_grades.json"
HF_STATE_FILE = "grader_state.json"
LOCAL_GRADES_FILE = REPO_DIR / "independent_grades.json"
LOCAL_STATE_FILE = REPO_DIR / "grader_state.json"
LOCAL_FEEDBACK_FILE = REPO_DIR / "feedback_data.json"
def _case_id_map_path() -> Path:
candidates = [
APP_DIR / "case_id_map.json",
REPO_DIR / "grading_space" / "case_id_map.json",
REPO_DIR / "pages" / "case_id_map.json",
]
for path in candidates:
if path.exists():
return path
return candidates[0]
CASE_ID_MAP_FILE = _case_id_map_path()
GRADER_CHOICES = {
"Grader 1": "grader_1",
"Grader 2": "grader_2",
}
GRADER_LABELS = {value: key for key, value in GRADER_CHOICES.items()}
ERROR_JUDGMENT_OPTIONS = ["Correct", "Overcalled", "Unclear"]
USEFULNESS_OPTIONS = ["Not useful", "Somewhat", "Very useful"]
REVIEW_HELP_OPTIONS = ["Not at all", "Somewhat", "A lot"]
HARM_OPTIONS = ["No", "Possibly", "Yes"]
PREFERRED_MODEL_OPTIONS = [
"DeepSeek Reasoner",
"GPT-OSS-120B",
"About the same",
"Neither helpful",
]
SECTION_HEADERS = [
"Prezime i ime",
"MBOO",
"Adresa",
"Datum dolaska",
"Datum rođenja",
"Datum otpusta",
"Trijažna kategorija",
"Dijagnoze",
"Ostale dijagnoze",
"Podaci s trijaže",
"Vitalni znakovi",
"Razlog dolaska",
"Anamneza",
"Status",
"Dijagnostičke i konzilijarne pretrage",
"Terapija",
"Zaključak",
"Uputa izabranom",
"Ostale preporuke",
"Address",
"Date of arrival",
"Arrival date",
"Date of birth",
"Date of discharge",
"Discharge date",
"Triage category",
"Diagnoses",
"Triage data",
"Vital signs",
"Reason for arrival",
"Anamnesis",
"Therapy",
"Conclusion",
"Instructions for",
"Other recommendations",
]
# -------------------------------------------------------------------------
# Hub / local persistence
# -------------------------------------------------------------------------
def _hf_token() -> str | None:
return os.environ.get("HF_TOKEN") or os.environ.get("HUGGINGFACE_TOKEN")
def _download_hub_json(filename: str):
token = _hf_token()
if not token:
return None
try:
from huggingface_hub import HfApi
api = HfApi(token=token)
local_path = api.hf_hub_download(
repo_id=HF_DATASET_REPO,
filename=filename,
repo_type="dataset",
)
with open(local_path, "r", encoding="utf-8") as handle:
return json.load(handle)
except Exception:
return None
def _upload_hub_json(filename: str, data, message: str) -> str | None:
token = _hf_token()
if not token:
return "No HF token — saved locally only."
try:
from huggingface_hub import HfApi
api = HfApi(token=token)
with tempfile.NamedTemporaryFile(
mode="w", suffix=".json", delete=False, encoding="utf-8"
) as tmp:
json.dump(data, tmp, ensure_ascii=False, indent=2)
tmp_path = tmp.name
api.upload_file(
path_or_fileobj=tmp_path,
path_in_repo=filename,
repo_id=HF_DATASET_REPO,
repo_type="dataset",
commit_message=message,
)
os.unlink(tmp_path)
return None
except Exception as exc:
return str(exc)
def _read_local_json(path: Path, default):
if not path.exists():
return default
with open(path, "r", encoding="utf-8") as handle:
return json.load(handle)
def _write_local_json(path: Path, data):
path.parent.mkdir(parents=True, exist_ok=True)
with open(path, "w", encoding="utf-8") as handle:
json.dump(data, handle, ensure_ascii=False, indent=2)
def load_grades() -> list:
hub = _download_hub_json(HF_GRADES_FILE)
local = _read_local_json(LOCAL_GRADES_FILE, [])
if isinstance(hub, list) and len(hub) >= len(local):
return hub
return local if isinstance(local, list) else []
def load_state() -> dict:
hub = _download_hub_json(HF_STATE_FILE)
local = _read_local_json(LOCAL_STATE_FILE, {"assignments": {}})
if not isinstance(local, dict):
local = {"assignments": {}}
local.setdefault("assignments", {})
if isinstance(hub, dict) and isinstance(hub.get("assignments"), dict):
if len(hub["assignments"]) >= len(local["assignments"]):
return hub
return local
def save_grades(grades: list) -> str | None:
_write_local_json(LOCAL_GRADES_FILE, grades)
return _upload_hub_json(
HF_GRADES_FILE, grades, f"independent grade #{len(grades)}"
)
def save_state(state: dict) -> str | None:
_write_local_json(LOCAL_STATE_FILE, state)
return _upload_hub_json(HF_STATE_FILE, state, "grader assignment update")
# -------------------------------------------------------------------------
# Case loading
# -------------------------------------------------------------------------
def load_case_id_map() -> dict[str, str]:
mapping = _read_local_json(CASE_ID_MAP_FILE, [])
return {item["timestamp"]: item["case_id"] for item in mapping}
def _case_id_for(entry: dict, index: int, ts_map: dict[str, str]) -> str:
return ts_map.get(entry.get("timestamp")) or f"case_{index:03d}"
def load_cases() -> list[dict]:
raw = _download_hub_json(HF_CASES_FILE)
if not isinstance(raw, list) or not raw:
raw = _read_local_json(LOCAL_FEEDBACK_FILE, [])
ts_map = load_case_id_map()
cases = []
for index, entry in enumerate(raw, 1):
if not isinstance(entry, dict):
continue
cases.append(
{
"case_id": _case_id_for(entry, index, ts_map),
"timestamp": entry.get("timestamp", ""),
"clinical_input": entry.get("clinical_input") or "",
"translation": entry.get("translation") or "",
"medication_lookup_context": entry.get("medication_lookup_context") or "",
"model_a_output": entry.get("model_a_output") or "",
"model_b_output": entry.get("model_b_output") or "",
"model_a_latency": entry.get("model_a_latency"),
"model_b_latency": entry.get("model_b_latency"),
"both_models_complete": bool(entry.get("both_models_complete")),
}
)
cases.sort(key=lambda item: item["case_id"])
return cases
# -------------------------------------------------------------------------
# Assignment helpers
# -------------------------------------------------------------------------
def graded_case_ids(grades: list, grader_id: str) -> set[str]:
return {
item.get("case_id")
for item in grades
if item.get("grader_id") == grader_id and item.get("case_id")
}
def assigned_to(state: dict, case_id: str) -> str | None:
record = state.get("assignments", {}).get(case_id) or {}
return record.get("grader_id")
def available_case_ids(cases: list, state: dict, grades: list, grader_id: str) -> list[str]:
mine = graded_case_ids(grades, grader_id)
available = []
for case in cases:
case_id = case["case_id"]
owner = assigned_to(state, case_id)
if owner and owner != grader_id:
continue
if case_id in mine:
continue
available.append(case_id)
return available
def checkout_next_case(cases: list, grader_id: str) -> tuple[str | None, str | None]:
state = load_state()
grades = load_grades()
remaining = available_case_ids(cases, state, grades, grader_id)
if not remaining:
return None, "No unassigned cases left for this grader."
case_id = remaining[0]
owner = assigned_to(state, case_id)
if owner and owner != grader_id:
return None, f"{case_id} was just taken by the other grader. Try again."
state.setdefault("assignments", {})[case_id] = {
"grader_id": grader_id,
"assigned_at": datetime.now(timezone.utc).isoformat(),
"status": "in_progress",
}
warning = save_state(state)
return case_id, warning
def release_case(case_id: str, grader_id: str) -> str | None:
state = load_state()
record = state.get("assignments", {}).get(case_id)
if not record:
return None
if record.get("grader_id") != grader_id:
return "This case belongs to the other grader."
if record.get("status") == "completed":
return "Completed cases cannot be released."
state["assignments"].pop(case_id, None)
return save_state(state)
def mark_case_completed(case_id: str, grader_id: str) -> str | None:
state = load_state()
state.setdefault("assignments", {})[case_id] = {
"grader_id": grader_id,
"assigned_at": (state.get("assignments", {}).get(case_id) or {}).get(
"assigned_at", datetime.now(timezone.utc).isoformat()
),
"completed_at": datetime.now(timezone.utc).isoformat(),
"status": "completed",
}
return save_state(state)
# -------------------------------------------------------------------------
# Display helpers
# -------------------------------------------------------------------------
def parse_model_json(raw: str) -> dict:
text = (raw or "").strip()
if text.startswith("```"):
first_newline = text.index("\n")
last_fence = text.rfind("```")
text = text[first_newline + 1 : last_fence].strip()
try:
return json.loads(text)
except json.JSONDecodeError:
start = text.find("{")
end = text.rfind("}") + 1
if start != -1 and end > start:
return json.loads(text[start:end])
raise
def parse_model_result(raw: str) -> dict:
if not (raw or "").strip():
return {
"success": False,
"error_message": "No stored output for this model.",
"errors": [],
"suggestions": [],
"summary": {},
}
try:
parsed = parse_model_json(raw)
except Exception as exc:
return {
"success": False,
"error_message": f"Could not parse stored output: {exc}",
"errors": [],
"suggestions": [],
"summary": {},
}
summary = parsed.get("summary")
if not isinstance(summary, dict):
summary = {"overall_assessment": summary or ""}
return {
"success": True,
"errors": parsed.get("errors") or [],
"suggestions": parsed.get("suggestions") or [],
"summary": summary,
}
def format_letter(text: str) -> str:
if not text.strip():
return "_No discharge letter stored for this case._"
cleaned = text.replace("\r\n", "\n").replace("\r", "\n")
cleaned = re.sub(r"[ \t]+\n", "\n", cleaned)
cleaned = re.sub(r"\n{3,}", "\n\n", cleaned).strip()
header_pattern = re.compile(
r"^(" + "|".join(re.escape(item) for item in SECTION_HEADERS) + r")\b",
re.IGNORECASE,
)
lines = []
for line in cleaned.split("\n"):
stripped = line.strip()
if stripped and header_pattern.match(stripped):
lines.append(f"\n### {stripped}\n")
else:
lines.append(stripped)
return "\n".join(lines).strip()
def extract_diagnosis(text: str) -> str:
match = re.search(
r"(?:Dijagnoze|Diagnoses)\s*\n+([^\n]+)",
text or "",
flags=re.IGNORECASE,
)
if match:
return match.group(1).strip()
return "Diagnosis not listed"
def extract_arrival(text: str) -> str:
match = re.search(
r"(?:Datum dolaska|Date of arrival|Arrival date):\s*([^\n]+)",
text or "",
flags=re.IGNORECASE,
)
if match:
return match.group(1).strip()
return ""
SEVERITY_LABELS = {
"critical": "🔴 Critical",
"high": "🟠 High",
"medium": "🟡 Medium",
"low": "🟢 Low",
}
CATEGORY_LABELS = {
"medication_error": "Medication",
"diagnostic_error": "Diagnostic",
"dosing_error": "Dosing",
"documentation_error": "Documentation",
"lab_interpretation_error": "Lab Interpretation",
"contraindication": "Contraindication",
"omission": "Omission",
"disposition_safety": "Disposition Safety",
"other": "Other",
"documentation_quality": "Documentation Quality",
"clinical_workflow": "Clinical Workflow",
"patient_safety": "Patient Safety",
"continuity_of_care": "Continuity of Care",
"patient_education": "Patient Education",
"followup_quality": "Follow-up Quality",
"medication_safety": "Medication Safety",
"completeness": "Completeness",
}
PRIORITY_LABELS = {"high": "🔴 High", "medium": "🟡 Medium", "low": "🟢 Low"}
SAFETY_RISK_LABELS = {"low": "🟢 Low", "moderate": "🟡 Moderate", "high": "🔴 High"}
DOC_QUALITY_LABELS = {
"excellent": "Excellent",
"good": "Good",
"adequate": "Adequate",
"needs_improvement": "Needs Improvement",
"poor": "Poor",
}
def render_model_output(result: dict, latency):
if not result["success"]:
st.error(result["error_message"])
return
if latency is not None:
st.caption(f"Original response time: {latency}s")
summary = result["summary"]
if summary.get("overall_assessment"):
parts = [f"**Summary:** {summary['overall_assessment']}"]
if summary.get("safety_risk"):
parts.append(
"**Safety risk:** "
+ SAFETY_RISK_LABELS.get(summary["safety_risk"], summary["safety_risk"])
)
if summary.get("documentation_quality"):
parts.append(
"**Documentation quality:** "
+ DOC_QUALITY_LABELS.get(
summary["documentation_quality"], summary["documentation_quality"]
)
)
st.markdown(" | ".join(parts))
if result["errors"]:
for index, err in enumerate(result["errors"], 1):
sev = SEVERITY_LABELS.get(err.get("severity", ""), err.get("severity", ""))
cat = CATEGORY_LABELS.get(err.get("category", ""), err.get("category", ""))
quote = (
f'
Quote: "{err.get("quote")}"' if err.get("quote") else ""
)
why = ""
if err.get("why_it_matters"):
why = f"
Why it matters: {err['why_it_matters']}"
st.markdown(
f'