프로젝트 1 — 여행지 소개 체인

import pandas as pd

# 가상의 여행지 8곳
spots = pd.read_csv("travel_spots.csv")
print(spots[["여행지", "어울리는 손님", "추천 계절", "예산"]].to_string(index=False))

# 손님 조건: 가족, 여름, 한 사람 1박 15만 원 이하
GUEST = {"손님": "가족", "계절": "여름", "예산": 15}
picked = spots[
    (spots["어울리는 손님"] == GUEST["손님"])
    & (spots["추천 계절"] == GUEST["계절"])
    & (spots["예산"] <= GUEST["예산"])
].sort_values("예산")            # 예산이 낮은 차례로
print()
print("조건에 맞는 곳:", len(picked), "곳")
print(picked[["여행지", "지역", "예산"]].to_string(index=False))

from langchain_core.prompts import ChatPromptTemplate

SYSTEM = """너는 구름다리여행의 여행지 소개 작가다.
아래 여행지 정보로 소개글을 쓴다.
- 꼭 세 문장으로 쓴다.
- 첫 문장은 여행지 이름으로 시작한다.
- 둘째 문장은 특징을 손님에게 맞춰 풀어 쓴다.
- 마지막 문장에 추천 계절과 한 사람 1박 예산을 적는다.
- 정보에 없는 사실(거리, 시설, 가격)은 지어내지 않는다."""

HUMAN = """여행지: {name}
지역: {region}
특징: {feature}
어울리는 손님: {guest}
추천 계절: {season}
예산: 한 사람 1박 {budget}만 원"""

prompt = ChatPromptTemplate.from_messages([("system", SYSTEM), ("human", HUMAN)])

# 표의 행 하나를 빈칸 이름의 사전으로 바꾼다
def to_input(row):
    return {"name": row["여행지"], "region": row["지역"], "feature": row["특징"],
            "guest": row["어울리는 손님"], "season": row["추천 계절"], "budget": row["예산"]}

# iterrows(): 표의 행을 (행 번호, 행) 짝으로 하나씩 꺼낸다
inputs = [to_input(row) for _, row in picked.iterrows()]
print("빈칸:", prompt.input_variables)
print(prompt.invoke(inputs[0]).to_messages()[1].content)

import os

from dotenv import find_dotenv, load_dotenv
from langchain_core.output_parsers import StrOutputParser
from langchain_openai import ChatOpenAI

load_dotenv(find_dotenv(usecwd=True))
MODEL = os.getenv("OPENAI_MODEL", "gpt-4.1-mini")
# 소개글은 조금 길 수 있어 최대 300 토큰
model_0 = ChatOpenAI(model=MODEL, temperature=0, max_tokens=300, timeout=30)

chain_0 = prompt | model_0 | StrOutputParser()
# 고른 여행지 두 곳의 소개글을 한 번에 만든다
intros = chain_0.batch(inputs)


def count_sentences(text):
    """마침표로 나눈 조각 가운데 빈 것을 뺀 수로 문장 수를 어림한다."""
    return len([s for s in text.split(".") if s.strip()])


for item, text in zip(inputs, intros):
    print(f"[{item['name']}] 문장 {count_sentences(text)}개")
    print(text)

# 온도만 다른 모델을 하나 더 만든다. 프롬프트와 파서는 그대로 둔다
model_1 = ChatOpenAI(model=MODEL, temperature=1, max_tokens=300, timeout=30)
chain_1 = prompt | model_1 | StrOutputParser()

rows = []
for temp, chain in [(0, chain_0), (1, chain_1)]:
    for trial in [1, 2]:
        text = chain.invoke(inputs[0])          # 첫째 여행지로 두 번씩
        rows.append({"온도": temp, "시도": trial, "문장 수": count_sentences(text), "소개글": text})
        print(f"온도 {temp}, {trial}번째: {text}")

result = pd.DataFrame(rows)
# 같은 온도의 두 소개글이 글자 하나까지 같은지 견준다
for temp in [0, 1]:
    pair = result[result["온도"] == temp]["소개글"].tolist()
    print(f"온도 {temp}: 두 번이 같은가?", pair[0] == pair[1])

result.to_csv("out_ch03_spot_intro.csv", index=False, encoding="utf-8-sig")
print("저장: out_ch03_spot_intro.csv")