app.py

from langchain_openai import ChatOpenAI
from langchain.schema import AIMessage, HumanMessage, SystemMessage
import gradio as gr

SYSTEM_PROMPT = """
당신의 이름은 오은영입니다.
당신은 초등학생을 위한 챗봇입니다.
당신은 항상 존댓말을 사용하며 친절하고 따뜻한 말투로 대화합니다.
당신은 학생들의 감정을 존중하고, 언제나 부드럽고 공감하는 태도로 응답해야 합니다.
당신의 말투는 부드럽고 배려심 있으며, 학생들이 안심하고 대화할 수 있도록 돕습니다.

당신의 역할은 다음과 같습니다.

학생들이 물어보는 모든 질문에 친절하고 이해하기 쉽게 답변합니다.

학생들이 사용하는 극단적이거나 부정적인 언어를 바른 말로 고쳐줍니다.

학생들이 건강하고 긍정적인 언어 습관을 기를 수 있도록 올바른 표현을 알려줍니다.

학생들이 사용한 부적절한 표현은 부드럽고 단호하게 짚어주며, 대체할 수 있는 바른 말을 제시합니다.

학생들의 눈높이에 맞는 말과 예시를 사용하여, 말의 의미를 쉽게 이해할 수 있도록 설명합니다.

학생이 위로가 필요하거나 감정 표현을 할 때, 따뜻하고 공감하는 말로 마음을 어루만져 줍니다.

당신은 절대 학생을 꾸짖거나 비난하지 않으며, 언제나 존중하는 태도로 대화합니다.

당신은 초등학생의 정서와 언어 수준에 맞는 방식으로만 말합니다. 어려운 단어는 쓰지 않으며, 쓰게 될 경우 반드시 쉽게 설명합니다.

당신은 학생의 말에 귀 기울이고, 진심을 담아 대답합니다.

당신은 학생들이 편하게 다가올 수 있도록 다정하고 따뜻한 존재입니다.

지금부터 당신은 오은영이 되어 이 기준에 따라 대화합니다.

"""

model = ChatOpenAI(model="gpt-4o")


def chatbot_predict(message: str, history: list[dict[str, str]]):

    langchain_messages = []

    # 1. 시스템 프롬프트 추가
    if SYSTEM_PROMPT and SYSTEM_PROMPT.strip():
        langchain_messages.append(SystemMessage(content=SYSTEM_PROMPT))

    # 2. 이전 대화 기록 추가
    for entry in history:
        role = entry['role']
        content = entry['content']
        if role == "user":
            langchain_messages.append(HumanMessage(content=content))
        elif role == "assistant":
            langchain_messages.append(AIMessage(content=content))

    # 3. 현재 사용자 메시지 추가
    langchain_messages.append(HumanMessage(content=message))

    # 모델 호출
    response = model.invoke(langchain_messages)
    return response.content

# --- Gradio 인터페이스 ---
# 예시 질문 (시스템 프롬프트에 따라 간단히 변경 가능)
examples = ["What is your name?"]
if SYSTEM_PROMPT and "한국어" in SYSTEM_PROMPT:
    examples = ["안녕? 네 이름은 뭐니?"]

chat_interface = gr.ChatInterface(
    fn=chatbot_predict,
    title="AI 챗봇 (시스템 프롬프트 설정 가능)",
    description=f"",
    examples=examples,
    # type="messages"는 history를 [{'role': 'user', 'content': '...'}] 형식으로 받기 위해 필요합니다.
    # 코드2의 형식을 따릅니다.
    type="messages"
)

if __name__ == "__main__":
    print(f"Gradio 앱 실행 중... (시스템 프롬프트: \"{SYSTEM_PROMPT or '없음'}\")")
    chat_interface.launch()