실습 2 — 대화가 쌓이는 채팅 화면

from pathlib import Path

APP_CODE = r'''
import streamlit as st
from langchain_core.language_models.fake_chat_models import FakeListChatModel
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder

st.title("구름다리여행 여행 비서")
st.caption("연습용 가짜 모델이 정해 둔 답을 차례로 말합니다.")


# ---------- 챗봇 로직(주방): 화면과 상관없는 부분 ----------
@st.cache_resource
def make_chat_chain():
    """4차시의 대화 체인. 서버가 켜질 때 한 번만 만들어 모든 손님이 함께 쓴다"""
    prompt = ChatPromptTemplate.from_messages([
        ("system", "너는 구름다리여행의 친절한 여행 비서다. 두 문장 이내로 답한다."),
        MessagesPlaceholder("history"),
        ("human", "{question}"),
    ])
    model = FakeListChatModel(responses=[
        "좋습니다. 며칠 일정으로 생각하세요?",
        "2박 3일이면 동쪽 해안 일정을 추천합니다.",
        "아이와 함께라면 첫날은 숙소 근처에서 쉬는 일정을 권합니다.",
    ])
    return prompt | model | StrOutputParser()


def answer(chain, messages, question):
    """지금까지의 기록과 새 질문을 받아 답 글을 돌려준다"""
    return chain.invoke({"history": messages, "question": question})


# ---------- 화면(홀): 질문을 받고 말풍선을 그린다 ----------
chain = make_chat_chain()

# 다시 돌아도 남는 메모장에 대화 기록을 둔다(4차시의 기록 상자 자리)
if "messages" not in st.session_state:
    st.session_state.messages = []

# 지금까지의 대화를 말풍선으로 다시 그린다
for m in st.session_state.messages:
    with st.chat_message(m["role"]):
        st.markdown(m["content"])

if question := st.chat_input("여행에 대해 물어보세요"):
    with st.chat_message("user"):
        st.markdown(question)
    # 기록(이번 질문 전까지)과 새 질문을 챗봇 로직에 넘긴다
    reply = answer(chain, st.session_state.messages, question)
    with st.chat_message("assistant"):
        st.markdown(reply)
    # 질문과 답을 둘 다 기록에 넣는다
    st.session_state.messages.extend([
        {"role": "user", "content": question},
        {"role": "assistant", "content": reply},
    ])
'''

Path("out_ch07_chat.py").write_text(APP_CODE.lstrip(), encoding="utf-8")
print("저장: out_ch07_chat.py,", len(APP_CODE.strip().splitlines()), "줄")

import streamlit as st
from streamlit.testing.v1 import AppTest

# 앞 실습의 캐시가 섞이지 않게 비운다(서버를 새로 켠 것과 같다)
st.cache_resource.clear()

at = AppTest.from_file("out_ch07_chat.py").run()
print("제목:", at.title[0].value, "/ 말풍선:", len(at.chat_message), "개")
print("입력창의 안내 글:", at.chat_input[0].placeholder)

# 질문 하나를 보낸다
at.chat_input[0].set_value("다음 달에 제주도 가요").run()
for m in at.chat_message:
    print(f"  [{m.name}] {m.markdown[0].value}")

for q in ["2박 3일이요", "아이 둘이랑 가요"]:
    at.chat_input[0].set_value(q).run()

print("말풍선:", len(at.chat_message), "개")
for m in at.chat_message:
    print(f"  [{m.name}] {m.markdown[0].value}")

# 메모장에 쌓인 기록을 앱 밖에서 들여다본다
history = at.session_state["messages"]
print("기록:", len(history), "개 / 역할:", [m["role"] for m in history])

# 같은 앱을 새로 열면 새 손님(새 브라우저 탭)이다
guest = AppTest.from_file("out_ch07_chat.py").run()
print("새 손님의 말풍선:", len(guest.chat_message), "개")

guest.chat_input[0].set_value("부산 가고 싶어요").run()
print("새 손님의 기록:", len(guest.session_state["messages"]), "개 / 첫 손님의 기록:", len(at.session_state["messages"]), "개")
print("새 손님이 받은 답:", guest.chat_message[-1].markdown[0].value)