import os
import streamlit as st
from pipeline import run_pipeline
st.set_page_config(page_title="AirNova Support | Guardrailed Assistant", page_icon="✈️", layout="wide")
# ---------------------------------------------------------------------------
# Theme: deep red / near-black, minimal, high-contrast -- airline-brand feel
# ---------------------------------------------------------------------------
st.markdown(
"""
""",
unsafe_allow_html=True,
)
if "messages" not in st.session_state:
st.session_state.messages = []
if "audit_log" not in st.session_state:
st.session_state.audit_log = []
with st.sidebar:
st.markdown("### ✈️ AirNova Assistant")
st.caption("Deterministic Rule-Enforced Support Agent — demo")
guardrails_on = st.toggle("Guardrails enabled", value=True)
st.markdown(
f''
f'{"GUARDRAILS ON" if guardrails_on else "GUARDRAILS OFF — unsafe demo mode"}',
unsafe_allow_html=True,
)
st.divider()
llm_mode = st.radio(
"Response generator",
options=["mock", "real"],
format_func=lambda m: "Mock (offline, deterministic)" if m == "mock" else "Real LLM (Claude, needs ANTHROPIC_API_KEY)",
)
if llm_mode == "real" and not os.environ.get("ANTHROPIC_API_KEY"):
st.warning("No ANTHROPIC_API_KEY set — will fall back to mock.")
st.divider()
st.caption(
"Try asking: *'My father passed away and I already flew, can I get a refund?'* "
"with guardrails off, then on, to see the difference."
)
if st.button("Clear conversation"):
st.session_state.messages = []
st.session_state.audit_log = []
st.rerun()
st.divider()
st.markdown("#### 📋 Audit log")
if not st.session_state.audit_log:
st.caption("No commitment-bearing responses yet.")
for entry in reversed(st.session_state.audit_log[-8:]):
css = "nova-blocked" if entry["blocked"] else "nova-passed"
icon = "🛑 BLOCKED" if entry["blocked"] else "✅ VERIFIED"
st.markdown(
f'{icon}
'
f'{entry["reason"]}
',
unsafe_allow_html=True,
)
st.markdown('AirNova Customer Support
', unsafe_allow_html=True)
st.markdown(
'Every commitment-bearing reply is cross-checked against an immutable, '
'legally-authorized policy database before it reaches you.
',
unsafe_allow_html=True,
)
st.write("")
for msg in st.session_state.messages:
with st.chat_message(msg["role"]):
st.markdown(msg["content"])
if msg.get("meta"):
st.caption(msg["meta"])
if prompt := st.chat_input("Ask about refunds, cancellations, baggage, delays..."):
st.session_state.messages.append({"role": "user", "content": prompt})
with st.chat_message("user"):
st.markdown(prompt)
result = run_pipeline(prompt, guardrails_enabled=guardrails_on, llm_mode=llm_mode)
with st.chat_message("assistant"):
st.markdown(result.final_response)
if result.blocked:
st.error(f"⚠️ Original draft blocked by guardrails: {result.verification.reason}", icon="🛑")
with st.expander("See the unverified LLM draft that was blocked"):
st.code(result.draft_response)
elif result.verification and result.verification.authorized:
st.success(f"✅ Verified against {', '.join(result.verification.citations)}", icon="✅")
elif not result.guardrails_enabled and result.commitment_detected:
st.warning(
"Guardrails are OFF. This commitment-bearing response was sent to the customer "
"with no verification against the policy database — this is the Air Canada failure mode.",
icon="⚠️",
)
meta = None
if result.commitment_detected:
meta = f"Flagged terms: {', '.join(result.flagged_terms)}"
st.session_state.messages.append({"role": "assistant", "content": result.final_response, "meta": meta})
if result.commitment_detected and result.guardrails_enabled:
st.session_state.audit_log.append({
"blocked": result.blocked,
"reason": result.verification.reason if result.verification else "",
})
st.rerun()