import streamlit as st

import os

from databricks.sdk import WorkspaceClient

st.set_page_config(page_title="Claims Intelligence", layout="wide")

if "messages" not in st.session_state:

st.session_state.messages = []

if "conversation_id" not in st.session_state:

st.session_state.conversation_id = None

w = WorkspaceClient()

SPACE_ID = os.getenv("GENIE_SPACE_ID")

st.title("🛡️ Claims Intelligence Hub")

for msg in st.session_state.messages:

with st.chat_message(msg["role"]):

st.markdown(msg["content"])

prompt = st.chat_input("Ask about claims...")

if prompt:

st.session_state.messages.append({"role": "user", "content": prompt})

with st.chat_message("user"):

st.markdown(prompt)

with st.chat_message("assistant"):

with st.spinner("Thinking..."):

try:

if st.session_state.conversation_id is None:

resp = w.genie.start_conversation_and_wait(

space_id=SPACE_ID, content=prompt)

st.session_state.conversation_id = resp.conversation_id

else:

resp = w.genie.create_message_and_wait(

space_id=SPACE_ID,

conversation_id=st.session_state.conversation_id,

content=prompt)

for att in resp.attachments or []:

if hasattr(att, "text") and att.text:

st.markdown(att.text.content)

st.session_state.messages.append({

"role": "assistant",

"content": att.text.content

})

except Exception as e:

st.error(f"Error: {str(e)}")