더북(TheBook)

6.2.4 과거 대화 내역 관리하기

친근한 AI 챗봇이 거의 완성됐지만 아직은 다음 두 가지가 부족합니다.

• 반말로 이야기하지 않고 이모티콘도 보여주지 않습니다.

• 대화가 연속적으로 이어지지 않고 최근 대화 내용만 표시됩니다.

그럼 코드를 보완해 완성해봅시다.

ch06_chatbot.py

import streamlit as st
from openai import OpenAI # (9) openai 패키지에서 OpenAI 클래스 불러오기

# (1) main() 함수 선언
def main():
    #(2) 메인 화면 구성
    st.set_page_config(layout="wide")
    st.title("친근한 AI 챗봇")
    st.caption("스트림릿과 OpenAI API를 활용한 간단한 챗봇")
    #(3) 사이드바 구성
    with st.sidebar:
        st.subheader("OpenAI API Key 설정")
        #(4) 입력 위젯 유형 설정(비밀번호)
        openai_api_key = st.text_input("OpenAI API Key", type="password")
        st.write("[OpenAI API Key 받기](https://platform.openai.com/account/api-keys)")
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.