Light review UI in the ai-toolstack brand, plus small accuracy fixes

config/settings.yaml

@@ -32,7 +32,7 @@ pii:
     - UK_NIN          # National Insurance Number
     - CREDIT_CARD
     - IBAN_CODE
-    - LOCATION        # postcodes / addresses
+    - LOCATION        # place names only; UK postcodes are NOT caught
     - IP_ADDRESS
     - URL
 
@@ -50,7 +50,9 @@ debug:
   # When on, the API writes artifacts inside the session folder, so they are
   # removed with the session (DELETE /api/session/{id} or the TTL cleanup).
   enabled: false
-  # Root folder for debug runs. Each execution creates a timestamped sub-folder.
+  # Used only by the command-line runner (src/main.py): root folder for its debug
+  # runs, one timestamped sub-folder each. The API ignores it and writes debug
+  # output inside the session folder instead.
   output_dir: "./output/debug"
   # Save the raw Markdown produced by docling for each PDF.
   save_markdown: true

src/api.py

@@ -259,8 +259,8 @@ async def process_documents(files: list[UploadFile] = File(...)):
 
     Returns a ``session_id`` which the UI uses for all subsequent requests.
 
-    Note: This endpoint is synchronous and may take 30–90 seconds depending
-    on Groq API response times.
+    Note: This endpoint is synchronous. A full four-PDF pack takes about two
+    to three minutes on Groq's free tier, mostly waiting on rate limits.
     """
     if not files:
         raise HTTPException(status_code=400, detail="No files uploaded.")

src/privacy.py

@@ -3,7 +3,11 @@ privacy.py — PII detection and masking via Microsoft Presidio.
 
 Entities masked before any text is sent to the LLM:
   PERSON, PHONE_NUMBER, EMAIL_ADDRESS, UK_NHS, UK_NIN,
-  CREDIT_CARD, IBAN_CODE, DATE_TIME (opt-in), LOCATION
+  CREDIT_CARD, IBAN_CODE, LOCATION, IP_ADDRESS, URL, DATE_TIME (opt-in)
+
+Not masked: UK postcodes (Presidio has no UK postcode recogniser; LOCATION is
+a named-entity model that catches place names, not postcodes) and often street
+lines. Dates of birth are not masked unless DATE_TIME is enabled.
 
 Usage
 -----
@@ -37,7 +41,7 @@ UK_MOTOR_ENTITIES: list[str] = [
     "UK_NIN",         # National Insurance Number
     "CREDIT_CARD",
     "IBAN_CODE",
-    "LOCATION",       # postcodes / addresses
+    "LOCATION",       # place names only; UK postcodes are NOT caught
     "IP_ADDRESS",
     "URL",
 ]

src/prompts.py

@@ -11,8 +11,8 @@ Usage
     registry = PromptRegistry(config_path="custom.yaml")
 
     system_prompt = registry.get(DocumentType.SCHEDULE)
-    print(registry.active_version)   # → "v1"
-    print(registry.available_versions)  # → ["v1"]
+    print(registry.active_version)   # → "v2" (the active_version in config/prompts.yaml)
+    print(registry.available_versions)  # → ["v1", "v2"]
 """
 from __future__ import annotations
 

ui/src/Brand.tsx

@@ -1,16 +1,17 @@
 /** ai·toolstack brand primitives shared by every PolicyTrace screen. */
 
 export function Mark({ size = 24 }: { size?: number }) {
+  // Three layers of a system; the middle one ends in an amber dot: a value traced to its source.
   return (
-    <svg width={size} height={size} viewBox="0 0 56 56" aria-hidden="true" focusable="false">
-      <rect x="0.5" y="0.5" width="55" height="55" rx="12" fill="#1F2940" stroke="#2C3752" />
-      <rect x="12" y="14" width="32" height="6" rx="2" fill="#E0A946" />
-      <rect x="16" y="25" width="28" height="6" rx="2" fill="#ECEAE4" />
-      <rect x="12" y="36" width="32" height="6" rx="2" fill="#56617A" />
+    <svg width={size} height={size} viewBox="0 0 32 32" aria-hidden="true" focusable="false">
+      <rect width="32" height="32" rx="7" fill="#1F2940" />
+      <rect x="7" y="7.5" width="18" height="3" rx="1.5" fill="#8A93A8" />
+      <rect x="7" y="14.5" width="11" height="3" rx="1.5" fill="#F6F5F1" />
+      <circle cx="23.5" cy="16" r="2.4" fill="#E0A946" />
+      <rect x="7" y="21.5" width="18" height="3" rx="1.5" fill="#8A93A8" />
     </svg>
   )
 }
-
 export function Lockup({ product = 'PolicyTrace' }: { product?: string }) {
   return (
     <div className="flex items-center gap-3 min-w-0">
@@ -21,9 +22,9 @@ export function Lockup({ product = 'PolicyTrace' }: { product?: string }) {
         className="flex items-center gap-2.5 text-ink-100"
         aria-label="ai·toolstack home"
       >
-        <Mark size={26} />
-        <span className="text-[17px] font-semibold tracking-[-0.035em] hidden sm:inline">
-          <span className="font-mono font-medium text-ink-300">ai·</span>toolstack
+        <Mark size={24} />
+        <span className="text-[17px] font-semibold tracking-[-0.02em] hidden sm:inline">
+          ai<span className="text-signal-deep">·</span>toolstack
         </span>
       </a>
       <span className="text-ink-600 select-none hidden sm:inline" aria-hidden="true">/</span>

ui/src/FieldRow.tsx

@@ -72,7 +72,7 @@ export function FieldRow({ entry, sessionId, isActive, review, onClick }: Props)
                   if (e.key === 'Escape') setEditing(false)
                 }}
               />
-              <button onClick={handleSaveOverride} className="text-[12px] font-medium px-3 rounded-md bg-ink text-ink-100 hover:bg-ink-700">Save</button>
+              <button onClick={handleSaveOverride} className="text-[12px] font-medium px-3 rounded-md bg-ink-100 text-white hover:bg-black">Save</button>
               <button onClick={() => setEditing(false)} className="text-[12px] font-medium px-3 rounded-md border border-paper-rule text-paper-ink hover:bg-paper">Cancel</button>
             </div>
           ) : (

ui/src/PDFPane.tsx

@@ -111,11 +111,11 @@ export function PDFPane({ sessionId }: Props) {
               aria-selected={on}
               onClick={() => setActivePdf(f)}
               className={`relative shrink-0 inline-flex items-center gap-2 h-9 px-3 rounded-md text-[13px] transition-colors ${
-                on ? 'bg-ink-700 text-ink-100' : 'text-ink-300 hover:text-ink-100 hover:bg-ink-700/60'
+                on ? 'bg-white text-ink-100 shadow-sheet' : 'text-ink-300 hover:text-ink-100 hover:bg-white/70'
               }`}
             >
               <span className="truncate max-w-[14rem]">{f.replace(/\.pdf$/i, '').replace(/ - Demo$/i, '')}</span>
-              <span className={`font-mono text-[11px] ${on ? 'text-signal' : 'text-ink-400'}`}>{count}</span>
+              <span className={`font-mono text-[11px] ${on ? 'text-signal-deep' : 'text-ink-400'}`}>{count}</span>
               {on && <span className="absolute left-3 right-3 -bottom-[9px] h-[2px] bg-signal rounded-full" aria-hidden="true" />}
             </button>
           )
@@ -125,7 +125,7 @@ export function PDFPane({ sessionId }: Props) {
       {/* PDF scroll area */}
       <div
         ref={containerRef}
-        className="flex-1 overflow-y-auto pdf-scroll-container scroll-ink grid-bg bg-ink-900 p-4 sm:p-6 space-y-6"
+        className="flex-1 overflow-y-auto pdf-scroll-container scroll-ink wash p-4 sm:p-6 space-y-6"
       >
         {pdfUrl ? (
           <Document
@@ -156,7 +156,7 @@ export function PDFPane({ sessionId }: Props) {
                   // the canvas dimensions exactly (inline-block can shrink-wrap)
                   style={{ position: 'relative', width: containerWidth }}
                   className={`rounded-sm overflow-hidden transition-shadow ${
-                    hasActive ? 'ring-2 ring-signal ring-offset-4 ring-offset-ink-900' : ''
+                    hasActive ? 'ring-2 ring-signal ring-offset-4 ring-offset-[#F3F2EF]' : ''
                   }`}
                 >
                   <Page
@@ -225,7 +225,7 @@ function LoadingPlaceholder() {
 
 function ErrorPlaceholder() {
   return (
-    <div className="flex items-center justify-center p-12 text-[#F2A35E] text-sm font-mono">
+    <div className="flex items-center justify-center p-12 text-state-flag text-sm font-mono">
       Failed to load PDF.
     </div>
   )

ui/src/ReviewDashboard.tsx

@@ -66,9 +66,9 @@ export function ReviewDashboard({ sessionId }: Props) {
 }
 
 const TONES = {
-  pass: 'text-[#6FCF97]',
-  override: 'text-[#A9B8E0]',
-  flag: 'text-[#F2A35E]',
+  pass: 'text-state-pass',
+  override: 'text-state-override',
+  flag: 'text-state-flag',
 } as const
 
 function Stat({ label, value, tone }: { label: string; value: number | string; tone?: keyof typeof TONES }) {

ui/src/SessionPage.tsx

@@ -46,7 +46,7 @@ export function SessionPage() {
     return (
       <div className="min-h-screen flex items-center justify-center bg-ink grid-bg">
         <div className="text-center space-y-3">
-          <svg className="animate-spin h-8 w-8 mx-auto text-signal" viewBox="0 0 24 24" fill="none">
+          <svg className="animate-spin h-8 w-8 mx-auto text-signal-deep" viewBox="0 0 24 24" fill="none">
             <circle className="opacity-25" cx="12" cy="12" r="10" stroke="currentColor" strokeWidth="4" />
             <path className="opacity-75" fill="currentColor" d="M4 12a8 8 0 018-8v8z" />
           </svg>
@@ -60,12 +60,12 @@ export function SessionPage() {
     return (
       <div className="min-h-screen flex items-center justify-center bg-ink grid-bg">
         <div className="text-center space-y-4 max-w-sm">
-          <p role="alert" className="text-sm text-[#F2B79A] bg-state-flag/10 border border-state-flag/40 rounded-xl px-4 py-3">
+          <p role="alert" className="text-sm text-state-flag bg-state-flag-tint border border-state-flag/40 rounded-xl px-4 py-3">
             {error}
           </p>
           <button
             onClick={() => navigate('/')}
-            className="text-sm font-medium text-signal hover:text-signal-hover"
+            className="text-sm font-medium text-signal-deep hover:text-ink-100"
           >
             ← Back to upload
           </button>

ui/src/UploadPage.tsx

@@ -6,13 +6,13 @@ import type { SamplePack } from './types'
 import { ExternalIcon, LINKS, Lockup, Spinner } from './Brand'
 
 const PIPELINE = [
-  ['01', 'parse', 'Docling layout + text'],
-  ['02', 'pii_mask', 'Presidio, before any model call'],
-  ['03', 'classify', 'route each document'],
-  ['04', 'extract', 'typed record + verbatim citations'],
-  ['05', 'arbitrate', 'source-authority rules'],
-  ['06', 'provenance', 'match citations to PDF geometry'],
-  ['07', 'review', 'you verify, flag or override'],
+  ['01', 'parse', 'Docling: text and page positions'],
+  ['02', 'mask', 'Presidio: names out before any model call'],
+  ['03', 'classify', 'which document is which'],
+  ['04', 'extract', 'typed record + the exact phrase read'],
+  ['05', 'merge', 'a written rule picks the winner'],
+  ['06', 'trace', 'each value back to its PDF line'],
+  ['07', 'review', 'you verify, override or flag'],
 ] as const
 
 export function UploadPage() {
@@ -111,27 +111,27 @@ export function UploadPage() {
 
           {/* Left: what this is */}
           <section className="space-y-7 lg:pt-4">
-            <p className="eyebrow text-signal">Reference system · Document AI</p>
-            <h1 className="text-[40px] sm:text-[52px] leading-[1.0] font-semibold tracking-[-0.045em] text-ink-100">
-              Reviewable extraction for motor insurance packs.
+            <p className="inline-flex font-mono text-[12px] uppercase tracking-[0.06em] text-[#2F5FA8] bg-[#E8EFFA] rounded-full px-2.5 py-1">Reference system · Document AI</p>
+            <h1 className="text-[34px] sm:text-[44px] leading-[1.08] font-semibold tracking-[-0.03em] text-ink-100">
+              A policy pack in. One checked record out.
             </h1>
             <p className="text-[17px] leading-relaxed text-ink-300 max-w-[34rem]">
-              Upload a Schedule, Certificate, Statement of Fact and Policy Booklet. PolicyTrace builds one typed
-              Golden Record, resolves disagreements between documents, and links every field back to the exact
-              place it came from in the PDF.
+              Upload the Schedule, Certificate, Statement of Fact and policy booklet. PolicyTrace pulls the facts a
+              claims handler needs into one typed record, settles disagreements between documents with a written rule,
+              and links every value to the line in the PDF it came from.
             </p>
 
-            <ol className="rounded-xl border border-ink-600 bg-ink-800/90 font-mono text-[13px] overflow-hidden" aria-label="Pipeline">
+            <ol className="rounded-xl border border-ink-700 bg-white font-mono text-[13px] overflow-hidden" aria-label="Pipeline">
               {PIPELINE.map(([n, name, desc], i) => {
                 const state = !loading ? 'idle' : i < step ? 'done' : i === step ? 'active' : 'idle'
                 return (
                   <li key={n} className="flex items-center gap-4 px-4 py-2.5 border-b border-dashed border-ink-700 last:border-b-0">
                     <span className="w-6 text-ink-400">{n}</span>
-                    <span className={`w-24 ${state === 'active' ? 'text-signal' : 'text-ink-100'}`}>{name}</span>
+                    <span className={`w-24 ${state === 'active' ? 'text-signal-deep' : 'text-ink-100'}`}>{name}</span>
                     <span className="flex-1 text-ink-400 truncate">{desc}</span>
                     <span className="w-14 text-right">
-                      {state === 'done' && <span className="text-[#6FCF97]">ok</span>}
-                      {state === 'active' && <span className="inline-flex text-signal"><Spinner className="h-3.5 w-3.5" /></span>}
+                      {state === 'done' && <span className="text-state-pass">ok</span>}
+                      {state === 'active' && <span className="inline-flex text-signal-deep"><Spinner className="h-3.5 w-3.5" /></span>}
                     </span>
                   </li>
                 )
@@ -140,7 +140,7 @@ export function UploadPage() {
           </section>
 
           {/* Right: upload card */}
-          <section className="rounded-2xl border border-ink-600 bg-ink-800 p-5 sm:p-6 shadow-page" aria-label="Upload documents">
+          <section className="rounded-2xl border border-ink-700 bg-white p-5 sm:p-6 shadow-page" aria-label="Upload documents">
             <div
               role="button"
               tabIndex={0}
@@ -151,7 +151,7 @@ export function UploadPage() {
               onDrop={(e) => { e.preventDefault(); setDragOver(false); handleFiles(e.dataTransfer.files) }}
               onClick={() => inputRef.current?.click()}
               className={`rounded-xl border border-dashed px-6 py-10 text-center cursor-pointer transition-colors ${
-                dragOver ? 'border-signal bg-signal/10' : 'border-ink-500 hover:border-ink-400 bg-ink-900/40'
+                dragOver ? 'border-signal bg-signal-tint' : 'border-ink-500 hover:border-ink-400 bg-ink-800'
               }`}
             >
               <svg className={`mx-auto mb-4 h-9 w-9 ${dragOver ? 'text-signal' : 'text-ink-400'}`} fill="none" viewBox="0 0 24 24"
@@ -159,7 +159,7 @@ export function UploadPage() {
                 <path strokeLinecap="round" strokeLinejoin="round" d="M12 16V4m0 0-4 4m4-4 4 4M4 16v2a2 2 0 0 0 2 2h12a2 2 0 0 0 2-2v-2" />
               </svg>
               <p className="text-[15px] font-medium text-ink-100">
-                Drop PDFs here or <span className="text-signal">browse</span>
+                Drop PDFs here or <span className="text-signal-deep underline decoration-signal underline-offset-2">browse</span>
               </p>
               <p className="mt-1.5 font-mono text-[12px] text-ink-400">Schedule · Certificate · Statement of Fact · Booklet</p>
               <input ref={inputRef} type="file" accept=".pdf" multiple className="hidden"
@@ -167,7 +167,7 @@ export function UploadPage() {
             </div>
 
             {samplePack?.files.length ? (
-              <div className="mt-4 rounded-xl border border-ink-600 bg-ink-900/50 p-4">
+              <div className="mt-4 rounded-xl border border-ink-700 bg-ink-800 p-4">
                 <div className="flex flex-col gap-3 sm:flex-row sm:items-center sm:justify-between">
                   <div>
                     <p className="text-sm font-medium text-ink-100">No documents to hand?</p>
@@ -185,7 +185,7 @@ export function UploadPage() {
             {files.length > 0 && (
               <ul className="mt-4 space-y-2" aria-label="Selected files">
                 {files.map((f) => (
-                  <li key={f.name} className="flex items-center justify-between gap-3 rounded-lg border border-ink-600 bg-ink-900/40 px-3.5 py-2.5">
+                  <li key={f.name} className="flex items-center justify-between gap-3 rounded-lg border border-ink-700 bg-ink-800 px-3.5 py-2.5">
                     <div className="flex items-center gap-3 min-w-0">
                       <span className="shrink-0 rounded font-mono text-[10px] font-medium px-1.5 py-0.5 bg-ink-700 text-ink-300">PDF</span>
                       <span className="text-sm text-ink-200 truncate">{f.name}</span>
@@ -200,13 +200,13 @@ export function UploadPage() {
             )}
 
             {error && (
-              <div role="alert" className="mt-4 rounded-lg border border-state-flag/40 bg-state-flag/10 px-3.5 py-3 text-sm text-[#F2B79A]">
+              <div role="alert" className="mt-4 rounded-lg border border-state-flag/40 bg-state-flag/10 px-3.5 py-3 text-sm text-state-flag">
                 {error}
               </div>
             )}
 
             <button onClick={handleSubmit} disabled={!files.length || loading} className="btn-signal w-full mt-5 text-[15px]">
-              {loading ? <><Spinner /> Extracting · usually 30–90 s</> : <>Extract &amp; review{files.length ? ` ${files.length} file${files.length > 1 ? 's' : ''}` : ''}</>}
+              {loading ? <><Spinner /> Extracting · usually 2–3 min</> : <>Extract &amp; review{files.length ? ` ${files.length} file${files.length > 1 ? 's' : ''}` : ''}</>}
             </button>
 
             <p className="mt-4 text-[12px] leading-relaxed text-ink-400">
@@ -217,7 +217,7 @@ export function UploadPage() {
       </main>
 
       <footer className="border-t border-ink-700 px-5 sm:px-8 py-4 flex flex-wrap gap-x-6 gap-y-1 justify-between font-mono text-[12px] text-ink-400">
-        <span>PolicyTrace · a reference system from <a className="text-ink-300 hover:text-signal" href="https://www.ai-toolstack.com/" target="_blank" rel="noopener noreferrer">ai·toolstack</a></span>
+        <span>PolicyTrace · a reference system from <a className="text-ink-300 hover:text-signal-deep" href="https://www.ai-toolstack.com/" target="_blank" rel="noopener noreferrer">ai·toolstack</a></span>
         <span>Docling · Presidio · Groq · Instructor · FastAPI · React</span>
       </footer>
     </div>

ui/src/index.css

@@ -22,7 +22,7 @@
     @apply bg-ink text-ink-100 font-sans antialiased;
     text-rendering: optimizeLegibility;
   }
-  ::selection { background: rgba(224, 169, 70, 0.35); }
+  ::selection { background: rgba(224, 169, 70, 0.30); }
   :focus-visible { outline: 2px solid #E0A946; outline-offset: 2px; border-radius: 6px; }
 }
 
@@ -35,14 +35,11 @@
       disabled:opacity-50 disabled:cursor-not-allowed;
     min-height: 44px;
   }
-  .btn-signal { @apply btn bg-signal text-ink hover:bg-signal-hover; }
-  .btn-ghost { @apply btn border border-ink-600 text-ink-100 hover:bg-ink-800; }
-  .grid-bg {
-    background-image:
-      linear-gradient(rgba(44, 55, 82, 0.35) 1px, transparent 1px),
-      linear-gradient(90deg, rgba(44, 55, 82, 0.35) 1px, transparent 1px);
-    background-size: 56px 56px;
-  }
+  .btn { border-radius: 999px; }
+  .btn-signal { @apply btn bg-ink-100 text-white hover:bg-black; }
+  .btn-ghost { @apply btn border border-ink-500 bg-white text-ink-100 hover:border-ink-100; }
+  .grid-bg { background-color: #F6F5F1; }
+  .wash { background-color: #F3F2EF; }
 }
 
 /* ── react-pdf ──────────────────────────────────────────────────── */
@@ -52,7 +49,7 @@
 .pdf-scroll-container { scroll-behavior: smooth; }
 
 /* Thin scrollbars that suit both panes */
-.scroll-ink { scrollbar-color: #2C3752 transparent; scrollbar-width: thin; }
+.scroll-ink { scrollbar-color: #D9D6CF transparent; scrollbar-width: thin; }
 .scroll-paper { scrollbar-color: #D3D0C6 transparent; scrollbar-width: thin; }
 
 @media (prefers-reduced-motion: reduce) {

ui/tailwind.config.js

@@ -1,5 +1,5 @@
 /** @type {import('tailwindcss').Config} */
-// ai·toolstack design tokens (mirrors the ai-toolstack.com theme)
+// ai·toolstack design tokens (mirrors the ai-toolstack.com light theme)
 export default {
   content: [
     "./index.html",
@@ -9,16 +9,18 @@ export default {
     extend: {
       colors: {
         ink: {
-          DEFAULT: '#121826', // midnight slate ground
-          900: '#121826',
-          800: '#182033',     // surface
-          700: '#1F2940',     // raised
-          600: '#2C3752',     // rule
-          500: '#3A4766',
-          400: '#7E88A2',     // dim text (AA on ink)
-          300: '#9CA6BA',     // muted text
-          200: '#CDD3DE',     // body text on ink
-          100: '#ECEAE4',     // bone
+          // Light scheme matching ai-toolstack.com 2.5. Names kept so classes still read
+          // "ground → surface → raised → rule → text"; values run light to dark.
+          DEFAULT: '#FFFFFF', // page ground
+          900: '#FFFFFF',
+          800: '#FAFAF8',     // surface (bars, cards)
+          700: '#ECEAE5',     // raised / rule
+          600: '#E2E0DA',     // stronger rule
+          500: '#D9D6CF',
+          400: '#7A8190',     // dim text
+          300: '#5B6270',     // muted text
+          200: '#4A5160',     // body text
+          100: '#1A1D24',     // ink
         },
         paper: {
           DEFAULT: '#F6F5F1',
@@ -48,7 +50,7 @@ export default {
       },
       boxShadow: {
         sheet: '0 1px 0 rgba(18,24,38,0.04), 0 8px 24px -16px rgba(18,24,38,0.25)',
-        page: '0 24px 60px -30px rgba(0,0,0,0.65)',
+        page: '0 1px 2px rgba(26,29,36,0.06), 0 12px 32px -18px rgba(26,29,36,0.28)',
       },
     },
   },