6.2.3 OpenAI API로 요청 및 응답 처리하기
화면을 완성했으니 OpenAI API로 파이썬 프로그램과 AI 모델을 연동해 고정된 답변이 아닌 진짜 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)")
# (10) OpenAI 클라이언트 생성
client = OpenAI(api_key=openai_api_key)