Streamlit Chat UI 핵심 패턴

기본 구조

import streamlit as st

# 1. 페이지 설정
st.set_page_config(page_title="제조 AI 어시스턴트", page_icon="")

# 2. 세션 초기화
if "messages" not in st.session_state:
    st.session_state.messages = []

# 3. 이전 대화 이력 표시
for msg in st.session_state.messages:
    with st.chat_message(msg["role"]):
        st.markdown(msg["content"])

# 4. 사용자 입력
if prompt := st.chat_input("설비에 대해 물어보세요..."):
    # 사용자 메시지 추가
    st.session_state.messages.append({"role": "user", "content": prompt})
    with st.chat_message("user"):
        st.markdown(prompt)

    # Agent 응답
    with st.chat_message("assistant"):
        response = agent.run(prompt)
        st.markdown(response["answer"])

    # 응답 저장
    st.session_state.messages.append(
        {"role": "assistant", "content": response["answer"]}
    )

도구 사용 로그 표시

# Agent 응답 시 도구 사용 정보 표시
with st.chat_message("assistant"):
    # 도구 사용 로그 (접을 수 있는 영역)
    if response.get("tool_calls"):
        with st.expander("사용된 도구", expanded=False):
            for tc in response["tool_calls"]:
                st.markdown(
                    f"- **{tc['name']}**"
                    f"(`{tc['args']}`)"
                )

    # 실제 답변
    st.markdown(response["answer"])

    # KG 시각화 (있으면)
    if response.get("kg_data"):
        with st.expander("설비 관계도", expanded=True):
            # pyvis 그래프 표시
            display_kg_graph(response["kg_data"])

사이드바 활용

# 사이드바에 시스템 정보 표시
with st.sidebar:
    st.title("제조 AI 어시스턴트")
    st.caption("v0.1.0 MVP")

    st.divider()

    # 시스템 상태
    st.subheader("시스템 상태")

    rag_status = "연결됨" if rag_service else "미연결"
    kg_status = "연결됨" if kg_service.is_connected else "미연결"

    st.markdown(f"- RAG: {rag_status}")
    st.markdown(f"- KG: {kg_status}")
    st.markdown(f"- LLM: {settings.llm_model}")

    st.divider()

    # 대화 초기화 버튼
    if st.button("대화 초기화", type="secondary"):
        st.session_state.messages = []
        st.rerun()

    # 통계
    st.subheader("이번 세션 통계")
    st.metric("총 질문 수", st.session_state.get("total_queries", 0))

스트리밍 응답 (선택사항)

# 스트리밍으로 실시간 타이핑 효과
with st.chat_message("assistant"):
    message_placeholder = st.empty()
    full_response = ""

    for chunk in agent.run_stream(prompt):
        if chunk.get("node") == "agent":
            # Agent 노드의 출력을 실시간 표시
            output = chunk["output"]
            if "messages" in output:
                last_msg = output["messages"][-1]
                if hasattr(last_msg, "content") and last_msg.content:
                    full_response = last_msg.content
                    message_placeholder.markdown(full_response + "...")

    message_placeholder.markdown(full_response)