update

Signed-off-by: bitliu <[email protected]>

app.py

@@ -519,6 +519,35 @@ def main():
             components.html(
                 create_highlighted_html_simple(result["text"], entities), height=150
             )
+        elif result["type"] == "toolcall_verifier":
+            unauthorized = result["unauthorized"]
+
+            if unauthorized:
+                st.error(f"⚠️ BLOCKED: Unauthorized tool call detected!")
+                st.markdown(f"**Flagged tokens:** {[t for t, _ in unauthorized[:10]]}")
+                st.markdown(f"**Total unauthorized tokens:** {len(unauthorized)}")
+            else:
+                st.success("✅ Tool call authorized")
+
+            st.markdown("### Input Format")
+            st.code(result["input_text"], language="text")
+
+            st.markdown("### Token-Level Classification")
+            # Create a simple table view
+            token_label_pairs = list(zip(result["tokens"], result["labels"]))
+            # Show first 50 tokens to avoid overwhelming the UI
+            display_tokens = token_label_pairs[:50]
+
+            for i in range(0, len(display_tokens), 5):
+                cols = st.columns(5)
+                for j, col in enumerate(cols):
+                    if i + j < len(display_tokens):
+                        token, label = display_tokens[i + j]
+                        color = "🔴" if label == "UNAUTHORIZED" else "🟢"
+                        col.markdown(f"{color} `{token}`")
+
+            if len(token_label_pairs) > 50:
+                st.info(f"Showing first 50 of {len(token_label_pairs)} tokens")
 
         # Raw Prediction Data expander
         with st.expander("🔬 Raw Prediction Data"):