Create app.py

app.py

@@ -0,0 +1,83 @@
+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"현재 시스템 프롬프트: \"{SYSTEM_PROMPT or '없음'}\"",
+    examples=examples,
+    # type="messages"는 history를 [{'role': 'user', 'content': '...'}] 형식으로 받기 위해 필요합니다.
+    # 코드2의 형식을 따릅니다.
+    type="messages"
+)
+
+if __name__ == "__main__":
+    print(f"Gradio 앱 실행 중... (시스템 프롬프트: \"{SYSTEM_PROMPT or '없음'}\")")
+    chat_interface.launch()