실습 4 — 진짜 모델을 붙인 채팅 화면

from pathlib import Path

APP_CODE = r'''
import os

import streamlit as st
from dotenv import find_dotenv, load_dotenv
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain_openai import ChatOpenAI


def get_api_key():
    """클라우드에서는 Secrets 칸, 내 PC 에서는 .env 에서 키를 읽는다"""
    try:
        return st.secrets["OPENAI_API_KEY"]
    except (FileNotFoundError, KeyError):
        load_dotenv(find_dotenv(usecwd=True))
        return os.getenv("OPENAI_API_KEY")


@st.cache_resource
def make_openai_chain(api_key, model_name):
    """진짜 모델의 대화 체인. 키와 모델 이름이 같으면 한 번만 만든다"""
    prompt = ChatPromptTemplate.from_messages([
        ("system", "너는 구름다리여행의 친절한 여행 비서다. 손님이 앞에서 말한 조건을 기억해 두 문장 이내로 답한다."),
        MessagesPlaceholder("history"),
        ("human", "{question}"),
    ])
    # api_key: 환경 변수 대신 읽어 온 키를 직접 넘긴다
    model = ChatOpenAI(model=model_name, api_key=api_key, temperature=0)
    return prompt | model | StrOutputParser()


st.title("구름다리여행 여행 비서")
api_key = get_api_key()
MODEL = os.getenv("OPENAI_MODEL", "gpt-4.1-mini")
with st.sidebar:
    st.header("설정")
    st.caption(f"모델: {MODEL}")
    st.caption("최근 10개 메시지만 모델에 보냅니다.")

if not api_key:
    # st.error: 빨간 안내 상자. st.stop: 이 줄 아래를 실행하지 않고 멈춘다
    st.error("API 키가 없습니다. .env 나 Secrets 에 OPENAI_API_KEY 를 넣어 주세요.")
    st.stop()

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)
    chain = make_openai_chain(api_key, MODEL)
    # 4차시의 Window 처럼 최근 10개만 보낸다
    history = st.session_state.messages[-10:]
    with st.chat_message("assistant"):
        reply = st.write_stream(chain.stream({"history": history, "question": question}))
    st.session_state.messages.extend([
        {"role": "user", "content": question},
        {"role": "assistant", "content": reply},
    ])
'''

Path("out_ch07_openai_chat.py").write_text(APP_CODE.lstrip(), encoding="utf-8")
print("저장: out_ch07_openai_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_openai_chat.py").run()
print("제목:", at.title[0].value)
print("사이드바:", [c.value for c in at.sidebar.caption])
# 키가 없으면 빨간 안내 상자가 하나 생긴다
print("오류 안내:", [e.value for e in at.error])

for q in ["다음 달에 제주도로 2박 3일 가요. 아이 둘이랑 가요.", "지금까지 말한 조건을 정리해 주세요."]:
    # 진짜 모델은 답에 몇 초가 걸리므로 기다리는 시간을 60초로 늘린다
    at.chat_input[0].set_value(q).run(timeout=60)

for m in at.chat_message:
    print(f"[{m.name}] {m.markdown[0].value}")

history = at.session_state["messages"]
print("기록:", len(history), "개")
print("오류:", len(at.exception), "개")