Update app.py

app.py

@@ -43,6 +43,10 @@ def _sha256(path: str) -> str:
         return "unknown"
 
 def _load_core_from_file(path: str):
+    """
+    Load spine_coder_core.py directly from a given path (bypass module cache).
+    Register in sys.modules BEFORE exec to enable imports inside the core.
+    """
     path = __os.path.abspath(path)
     if not __os.path.exists(path):
         return None
@@ -55,8 +59,10 @@ def _load_core_from_file(path: str):
     return mod
 
 def _force_import_core():
+    """Order: purge caches → SPINE_CORE_PATH → local files → package modules."""
     _purge_spine_modules()
 
+    # 1) Explicit override
     forced = __os.environ.get("SPINE_CORE_PATH")
     if forced and __os.path.exists(forced):
         mod = _load_core_from_file(forced)
@@ -64,6 +70,7 @@ def _force_import_core():
             print("[CORE] forced path:", forced, "sha:", _sha256(forced))
             return mod
 
+    # 2) Likely file paths near app
     for rel in [
         "spine_coder_core.py",
         "spine_coder/spine_coder_core.py",
@@ -76,6 +83,7 @@ def _force_import_core():
                 print("[CORE] loaded file:", p, "sha:", _sha256(p))
                 return mod
 
+    # 3) Package modules
     for modname in [
         "spine_coder.spine_coder.spine_coder_core",
         "spine_coder.spine_coder_core",
@@ -100,7 +108,7 @@ try:
 except Exception:
     _active_src = "unknown"
 
-# ---- One-time startup probe ---------------------------------------------------
+# ---- One-time startup probe (prints to Space logs) ----------------------------
 try:
     _probe_note = "Discectomies at C4–C5, C5–C6, and C6–C7. Interbody cages and anterior plate spanning C4–C7."
     _probe = suggest_with_cpt_billing(_probe_note, payer="Medicare", top_k=5)
@@ -219,6 +227,7 @@ def _core_path() -> str:
     except Exception:
         return "unknown"
 
+# Display columns (per-row modifiers are deliberately not shown here)
 SUGG_COLS = [
     "CPT", "Description", "Rationale",
     "Confidence", "Primary", "Category", "Laterality", "Units"
@@ -227,6 +236,7 @@ EMPTY_SUGG_DF = pd.DataFrame(columns=SUGG_COLS)
 EMPTY_MODS_DF = pd.DataFrame([{"modifier": "—", "reason": ""}])
 
 def _coalesce_rows(result: Dict[str, Any]) -> Tuple[pd.DataFrame, Dict[str, Any]]:
+    """Build the suggestions table (no modifier columns) and meta badges."""
     if not isinstance(result, dict):
         return EMPTY_SUGG_DF, {}
 
@@ -239,6 +249,7 @@ def _coalesce_rows(result: Dict[str, Any]) -> Tuple[pd.DataFrame, Dict[str, Any]
             continue
         mods = s.get("modifiers", []) or []
 
+        # Derive row laterality
         row_lat = (s.get("laterality") or "").strip().lower()
         if not row_lat:
             if isinstance(mods, list) and "LT" in mods:
@@ -262,7 +273,7 @@ def _coalesce_rows(result: Dict[str, Any]) -> Tuple[pd.DataFrame, Dict[str, Any]
             "Units": s.get("units", 1),
         })
 
-    # Levels
+    # Normalize levels
     segs: List[str] = []
     inters_list: List[str] = []
     lvl_lat = ""
@@ -274,7 +285,7 @@ def _coalesce_rows(result: Dict[str, Any]) -> Tuple[pd.DataFrame, Dict[str, Any]
     elif isinstance(levels_obj, list):
         segs = [str(x) for x in levels_obj]
 
-    # Flags
+    # Normalize flags
     flags_obj = result.get("flags")
     if isinstance(flags_obj, list):
         flags_list = [str(x) for x in flags_obj]
@@ -346,7 +357,7 @@ def run_inference(note: str, payer: str, top_k: int, session_id: str):
     try:
         result = suggest_with_cpt_billing(note=note, payer=payer, top_k=top_k)
         if DEBUG:
-            print("[DEBUG] build/region/laterality/flags:",
+            print("[DEBUG] build/region/laterality/flags]:",
                   result.get("build"), result.get("region"),
                   result.get("laterality"), result.get("flags"))
     except Exception as e:
@@ -406,7 +417,7 @@ def run_probe(session_id: str) -> Tuple[str, str]:
         tests = [
             ("Implicit TLIF",
              "Left facetectomy L4–L5 with PEEK interbody cage and pedicle screws; rods secured.",
-             ["22633"],
+             ["22633"],  # expect
             ),
             ("ACDF chain w/ plate",
              "Discectomies at C4–C5, C5–C6, and C6–C7. Interbody cages and anterior plate spanning C4–C7.",
@@ -417,10 +428,12 @@ def run_probe(session_id: str) -> Tuple[str, str]:
              ["00000"],
             ),
         ]
+        # Run once to get build
         probe = suggest_with_cpt_billing(tests[1][1], payer="Medicare", top_k=10)
         build = probe.get("build","")
         info_lines.append(f"**Core path:** `{core_path}`  \n**Build:** `{build}`")
 
+        # Execute each test
         for label, note, expect_codes in tests:
             res = suggest_with_cpt_billing(note, payer="Medicare", top_k=10)
             codes = [s.get("cpt") for s in (res.get("suggestions") or [])]
@@ -442,6 +455,7 @@ EXAMPLES = [
     ["Posterior cervical foraminotomy right C6–C7; no fusion or instrumentation.", "Medicare", 10],
     ["ALIF L5–S1 with structural allograft; non-segmental instrumentation placed.", "Medicare", 10],
     ["Removal of posterior segmental instrumentation T10–L2; no new hardware placed.", "Medicare", 10],
+    # Case-modifier smoke tests:
     ["TLIF L4–L5 was initiated but aborted midway due to neuromonitoring changes.", "Medicare", 10],  # -53
     ["Bilateral decompression and foraminotomy at L4–L5 and L5–S1.", "Medicare", 10],                # -50
     ["Assistant surgeon present; resident not available.", "Medicare", 10],                           # -82
@@ -464,27 +478,77 @@ THEME = gr.themes.Soft(
 CUSTOM_CSS = """
 :root { --radius-lg: 16px; }
 .gradio-container { font-family: ui-sans-serif, system-ui, -apple-system; }
-.header-card { border-radius: 18px; padding: 18px; border: 1px solid #1f2937;
-  background: linear-gradient(180deg,#0f172a,#0b1220); color: #e5e7eb; }
-.badge-row code { margin-right: 8px; border-radius: 12px; padding: 2px 8px;
-  background: #111827; color: #e5e7eb; }
+
+/* Header card */
+.header-card {
+  border-radius: 18px;
+  padding: 18px;
+  border: 1px solid #1f2937;
+  background: linear-gradient(180deg,#0f172a,#0b1220);
+  color: #e5e7eb;
+}
+.badge-row code {
+  margin-right: 8px;
+  border-radius: 12px;
+  padding: 2px 8px;
+  background: #111827;
+  color: #e5e7eb;
+}
+
+/* Table container */
 .table-wrap { max-height: 520px; overflow: auto; }
-#suggestions_table .dataframe { font-size: 15px; width: 100% !important;
-  table-layout: auto !important; border-collapse: collapse; }
-#suggestions_table .dataframe th, #suggestions_table .dataframe td {
-  white-space: normal; word-wrap: break-word; text-align: left;
-  vertical-align: top; padding: 10px 12px; }
+
+/* Suggestions table target */
+#suggestions_table .dataframe {
+  font-size: 15px;
+  width: 100% !important;
+  table-layout: auto !important;
+  border-collapse: collapse;
+}
+#suggestions_table .dataframe th,
+#suggestions_table .dataframe td {
+  white-space: normal;
+  word-wrap: break-word;
+  text-align: left;
+  vertical-align: top;
+  padding: 10px 12px;
+}
+
+/* Column sizing (1-indexed):
+   1=CPT, 2=Description, 3=Rationale, 4=Confidence,
+   5=Primary, 6=Category, 7=Laterality, 8=Units  */
+
+/* CPT — wider & no wrap, monospace, centered */
 #suggestions_table .dataframe th:nth-child(1),
 #suggestions_table .dataframe td:nth-child(1) {
-  min-width: 120px; max-width: 140px; white-space: nowrap !important;
+  min-width: 120px;
+  max-width: 140px;
+  white-space: nowrap !important;
   font-family: ui-monospace, SFMono-Regular, Menlo, Monaco, Consolas, "Liberation Mono", monospace;
-  text-align: center; }
+  text-align: center;
+}
+
+/* Description — roomy */
 #suggestions_table .dataframe th:nth-child(2),
-#suggestions_table .dataframe td:nth-child(2) { min-width: 360px; max-width: 560px; }
+#suggestions_table .dataframe td:nth-child(2) {
+  min-width: 360px;
+  max-width: 560px;
+}
+
+/* Rationale — roomy */
 #suggestions_table .dataframe th:nth-child(3),
-#suggestions_table .dataframe td:nth-child(3) { min-width: 320px; max-width: 520px; }
+#suggestions_table .dataframe td:nth-child(3) {
+  min-width: 320px;
+  max-width: 520px;
+}
+
+/* Category — a bit wider */
 #suggestions_table .dataframe th:nth-child(6),
-#suggestions_table .dataframe td:nth-child(6) { min-width: 180px; }
+#suggestions_table .dataframe td:nth-child(6) {
+  min-width: 180px;
+}
+
+/* Keep tiny columns compact */
 #suggestions_table .dataframe th:nth-child(4),
 #suggestions_table .dataframe td:nth-child(4),
 #suggestions_table .dataframe th:nth-child(5),
@@ -492,7 +556,10 @@ CUSTOM_CSS = """
 #suggestions_table .dataframe th:nth-child(7),
 #suggestions_table .dataframe td:nth-child(7),
 #suggestions_table .dataframe th:nth-child(8),
-#suggestions_table .dataframe td:nth-child(8) { min-width: 90px; }
+#suggestions_table .dataframe td:nth-child(8) {
+  min-width: 90px;
+}
+
 .footer-note { color:#94a3b8; font-size:12px; }
 """
 
@@ -510,6 +577,7 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
             )
 
     with gr.Row():
+        # Inputs
         with gr.Column(scale=5):
             note_in = gr.Textbox(
                 label="Operative Note",
@@ -525,7 +593,11 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
                 clear_btn = gr.Button("Clear")
 
             with gr.Accordion("Quick Examples", open=False):
-                gr.Examples(examples=EXAMPLES, inputs=[note_in, payer_dd, topk], label="Click a row to load an example")
+                gr.Examples(
+                    examples=EXAMPLES,
+                    inputs=[note_in, payer_dd, topk],
+                    label="Click a row to load an example"
+                )
 
             with gr.Accordion("Feedback", open=False):
                 fb_choice = gr.Radio(choices=["👍", "👎"], label="Was this helpful?")
@@ -542,10 +614,12 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
                 probe_btn = gr.Button("Run Probe (core path + 3 tests)")
                 probe_md = gr.Markdown("")
 
+        # Results
         with gr.Column(scale=7):
             gr.Markdown("#### Results")
             summary_md = gr.Markdown("—", elem_classes=["badge-row"])
 
+            # Suggestions table (no modifier columns)
             table = gr.Dataframe(
                 value=EMPTY_SUGG_DF,
                 label="CPT Suggestions",
@@ -556,6 +630,7 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
                 elem_id="suggestions_table",
             )
 
+            # Case-level modifiers table
             gr.Markdown("### Case Modifiers (visit-level)")
             case_mods_table = gr.Dataframe(
                 value=EMPTY_MODS_DF.copy(),
@@ -582,7 +657,10 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
     run_btn.click(run_inference, inputs=run_inputs, outputs=run_outputs)
     note_in.submit(run_inference, inputs=run_inputs, outputs=run_outputs)
 
-    clear_btn.click(on_clear, outputs=[note_in, payer_dd, topk, table, case_mods_table, summary_md, json_out, warn_md, session_id])
+    clear_btn.click(
+        on_clear,
+        outputs=[note_in, payer_dd, topk, table, case_mods_table, summary_md, json_out, warn_md, session_id]
+    )
 
     fb_submit.click(record_feedback, inputs=[session_id, fb_choice, fb_text], outputs=fb_status)
     export_btn.click(do_export, inputs=[session_id], outputs=[export_file])
@@ -590,4 +668,5 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
     probe_btn.click(run_probe, inputs=[session_id], outputs=[probe_md, warn_md])
 
 if __name__ == "__main__":
+    # If running locally: set server_name to 0.0.0.0 for external access; Space ignores.
     demo.launch()