Spaces:

ysharma
/

OPF-Document-PII-Explorer

Running on Zero

App Files Files Community

ysharma HF Staff commited on 16 days ago

Commit

47b2347

verified ·

1 Parent(s): a4f68e5

Update app_v6.py

Browse files

Files changed (1) hide show

app_v6.py +8 -34

app_v6.py CHANGED Viewed

@@ -1,33 +1,7 @@
 """
-PII Reveal - Document Privacy Explorer (v6)
 ============================================
-Fixes regressions reported against v5.
-1. Scan got slower in v5 (225s vs ~120s in v4)
-   The v5 change moved the Viterbi decoder loop to CPU because I assumed
-   the dominant cost on T4 was kernel-launch overhead. That was wrong.
-   PyTorch's per-op dispatch cost on CPU (~50-200µs) is much higher than
-   a CUDA kernel launch, so a 100k-iteration loop with 4-5 ops per iter
-   regressed ~90-100s. Viterbi goes back on GPU (v4 behaviour). The
-   remaining inference cost on T4 is fundamental — bf16 attention
-   matmuls are emulated on Turing and there's no code fix for that, only
-   a hardware change to L4 / A10 / A100.
-2. Redacted PDF export took "several minutes" on large files
-   The v5 implementation called page.search_for(needle) for every
-   unique PII string on every page, even when the string didn't appear
-   on the page. For a 100-page doc with 200 unique PII strings that's
-   20k search calls.
-   v6 does a cheap `needle in page_text` prefilter per page (one
-   page.get_text() call) before issuing any search_for, skips
-   apply_redactions on pages with no matches, and saves with
-   garbage=1 instead of 4 (the aggressive stream-recompression setting
-   is slow and brings little benefit after redaction).
-3. Retained from v5: light-theme refresh, PDF redaction endpoint, UI.
-   Retained code-level perf win: single torch.cat instead of the
-   unbind-then-stack roundtrip in predict_text.
 """
 # ── stdlib ───────────────────────────────────────────────────────
@@ -809,7 +783,7 @@ FRONTEND_HTML = r"""<!DOCTYPE html>
 <head>
 <meta charset="UTF-8">
 <meta name="viewport" content="width=device-width,initial-scale=1">
-<title>PII Reveal — Inspector</title>
 <link rel="preconnect" href="https://fonts.googleapis.com">
 <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
 <link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&family=JetBrains+Mono:wght@400;500&family=Source+Serif+4:opsz,wght@8..60,400;8..60,500;8..60,600&display=swap" rel="stylesheet">
@@ -1072,7 +1046,7 @@ button{font:inherit;color:inherit;background:transparent;border:0;cursor:pointer
             <circle cx="8.5" cy="8.5" r="3.2" stroke="var(--block-background-fill)" stroke-width="1.4" fill="none"/>
             <line x1="11.2" y1="11.2" x2="14.2" y2="14.2" stroke="var(--block-background-fill)" stroke-width="1.4" stroke-linecap="round"/>
           </svg>
-          <span class="u-brand-name">PII Reveal<span class="sub">/ inspector</span></span>
         </div>
         <h1 class="u-title">See what your documents are leaking.</h1>
         <p class="u-sub">Find every PII span in a PDF, DOC or DOCX — names, accounts, secrets and five other entity types — then export a fully redacted copy.</p>
@@ -1100,10 +1074,10 @@ button{font:inherit;color:inherit;background:transparent;border:0;cursor:pointer
         </div>
         <div class="u-meta">
-          <span>openai privacy filter</span>
           <span>128k ctx</span>
-          <span>bfloat16</span>
           <span>apache 2.0</span>
         </div>
       </div>
@@ -1136,7 +1110,7 @@ button{font:inherit;color:inherit;background:transparent;border:0;cursor:pointer
 <!-- ============ results view ============ -->
 <div id="results-view">
   <div class="shell">
-    <div class="pr-app" aria-label="PII Reveal inspector">
       <div class="pr-top">
         <div class="pr-logo">
@@ -1145,7 +1119,7 @@ button{font:inherit;color:inherit;background:transparent;border:0;cursor:pointer
             <circle cx="8.5" cy="8.5" r="3.2" stroke="var(--block-background-fill)" stroke-width="1.4" fill="none"/>
             <line x1="11.2" y1="11.2" x2="14.2" y2="14.2" stroke="var(--block-background-fill)" stroke-width="1.4" stroke-linecap="round"/>
           </svg>
-          <span class="pr-name">PII Reveal<span class="pr-name-sub">/ inspector</span></span>
         </div>
         <span class="pr-file-chip" id="file-chip"></span>
         <span class="pr-status" id="scan-status"><span class="pr-status-dot"></span>Scan complete</span>

 """
 ============================================
+PII Explorer - Document Privacy Explorer / Playground
+============================================
 """
 # ── stdlib ───────────────────────────────────────────────────────
 <head>
 <meta charset="UTF-8">
 <meta name="viewport" content="width=device-width,initial-scale=1">
+<title>PII Explorer — Playground</title>
 <link rel="preconnect" href="https://fonts.googleapis.com">
 <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
 <link href="https://fonts.googleapis.com/css2?family=Inter:wght@400;500;600;700&family=JetBrains+Mono:wght@400;500&family=Source+Serif+4:opsz,wght@8..60,400;8..60,500;8..60,600&display=swap" rel="stylesheet">
             <circle cx="8.5" cy="8.5" r="3.2" stroke="var(--block-background-fill)" stroke-width="1.4" fill="none"/>
             <line x1="11.2" y1="11.2" x2="14.2" y2="14.2" stroke="var(--block-background-fill)" stroke-width="1.4" stroke-linecap="round"/>
           </svg>
+          <span class="u-brand-name">PII Explorer<span class="sub">/ Playground</span></span>
         </div>
         <h1 class="u-title">See what your documents are leaking.</h1>
         <p class="u-sub">Find every PII span in a PDF, DOC or DOCX — names, accounts, secrets and five other entity types — then export a fully redacted copy.</p>
         </div>
         <div class="u-meta">
+          <span><b>OpenAI Privacy Filter</b></span>
           <span>128k ctx</span>
           <span>apache 2.0</span>
+          <span><b>gr.Server</b></span>
         </div>
       </div>
 <!-- ============ results view ============ -->
 <div id="results-view">
   <div class="shell">
+    <div class="pr-app" aria-label="PII Explorer Playground">
       <div class="pr-top">
         <div class="pr-logo">
             <circle cx="8.5" cy="8.5" r="3.2" stroke="var(--block-background-fill)" stroke-width="1.4" fill="none"/>
             <line x1="11.2" y1="11.2" x2="14.2" y2="14.2" stroke="var(--block-background-fill)" stroke-width="1.4" stroke-linecap="round"/>
           </svg>
+          <span class="pr-name">PII Explorer<span class="pr-name-sub">/ Playground</span></span>
         </div>
         <span class="pr-file-chip" id="file-chip"></span>
         <span class="pr-status" id="scan-status"><span class="pr-status-dot"></span>Scan complete</span>