Update app.py

app.py

@@ -1,12 +1,6 @@
 # app.py — Spine Coder (Chatbot + Feedback + Session Logs) — Gradio 4.x
 # ------------------------------------------------------------------------------
-# - Modern soft theme + light custom CSS
-# - Two-column layout: Inputs | Results
-# - Examples use gr.Examples (click-to-fill works in Gradio 4.x)
-# - Suggestions render as a pandas DataFrame (no UI "Error" chips)
-# - Robust result-shape normalization (levels can be dict/list, flags list/dict)
-# - JSON output + warnings panel (+ optional DEBUG tracebacks)
-# - Feedback logger + one-click session export to /logs
+# Stable build: fixed annotations, robust normalization, error-safe, aesthetic UI
 # ------------------------------------------------------------------------------
 
 import os
@@ -66,22 +60,19 @@ def _coalesce_suggestions(result: Dict[str, Any]) -> Tuple[List[Dict[str, Any]],
     if not isinstance(result, dict):
         return [], {}
 
-    # rows
+    # ---------- rows ----------
     sugg = result.get("suggestions") or []
     rows: List[Dict[str, Any]] = []
     for s in sugg:
         if not isinstance(s, dict):
             continue
         conf_val = s.get("confidence")
-        if isinstance(conf_val, (int, float)):
-            conf_out = round(float(conf_val), 2)
-        else:
-            conf_out = conf_val or ""
-
+        conf_out = round(float(conf_val), 2) if isinstance(conf_val, (int, float)) else (conf_val or "")
         rows.append({
             "CPT": s.get("cpt", ""),
             "Modifier": s.get("modifier") or "",
-            "Modifiers": ", ".join(s.get("modifiers", [])) if isinstance(s.get("modifiers"), list) else (s.get("modifiers") or ""),
+            "Modifiers": ", ".join(s.get("modifiers", [])) if isinstance(s.get("modifiers"), list)
+                        else (s.get("modifiers") or ""),
             "Description": s.get("desc", ""),
             "Rationale": s.get("rationale", ""),
             "Confidence": conf_out,
@@ -91,9 +82,12 @@ def _coalesce_suggestions(result: Dict[str, Any]) -> Tuple[List[Dict[str, Any]],
             "Units": s.get("units", 1),
         })
 
-    # normalize levels (dict or list)
+    # ---------- normalize levels ----------
+    segs: List[str] = []
+    inters: List[str] = []
+    lvl_lat: str = ""
+
     levels_obj = result.get("levels")
-    segs: List[str], inters: List[str], lvl_lat = [], [], ""
     if isinstance(levels_obj, dict):
         segs = list(levels_obj.get("segments") or [])
         inters = list(levels_obj.get("interspaces") or [])
@@ -101,7 +95,7 @@ def _coalesce_suggestions(result: Dict[str, Any]) -> Tuple[List[Dict[str, Any]],
     elif isinstance(levels_obj, list):
         segs = [str(x) for x in levels_obj]
 
-    # normalize flags (list or dict)
+    # ---------- normalize flags ----------
     flags_obj = result.get("flags")
     if isinstance(flags_obj, list):
         flags_list = [str(x) for x in flags_obj]
@@ -110,6 +104,7 @@ def _coalesce_suggestions(result: Dict[str, Any]) -> Tuple[List[Dict[str, Any]],
     else:
         flags_list = []
 
+    # ---------- meta ----------
     meta = {
         "payer": result.get("payer", ""),
         "region": result.get("region", ""),
@@ -214,7 +209,6 @@ CUSTOM_CSS = """
 with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing") as demo:
     session_id = gr.State(new_session())
 
-    # Header
     with gr.Row():
         with gr.Column():
             gr.Markdown("### 🦴 Spine Coder — CPT Billing & Operative Note NLP")
@@ -224,7 +218,6 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
                 '<span class="footer-note">No PHI is stored; inputs are session-scoped and ephemeral.</span></div>'
             )
 
-    # Body
     with gr.Row():
         # Inputs
         with gr.Column(scale=5):
@@ -275,7 +268,6 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
                 json_out = gr.Code(language="json", value="", interactive=False)
             warn_md = gr.Markdown("")
 
-    # Events
     def _on_load():
         sid = new_session()
         return sid, sid
@@ -294,5 +286,4 @@ with gr.Blocks(theme=THEME, css=CUSTOM_CSS, title="Spine Coder — CPT Billing")
     export_btn.click(do_export, inputs=[session_id], outputs=[export_file])
 
 if __name__ == "__main__":
-    # Set share=True if you want a public link while testing locally.
     demo.launch()