-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
45 lines (32 loc) · 1.42 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
import streamlit as st
import openai
# Streamlit Community Cloudの「Secrets」からOpenAI API keyを取得
openai.api_key = st.secrets.OpenAIAPI.openai_api_key
# st.session_stateを使いメッセージのやりとりを保存
if "messages" not in st.session_state:
st.session_state["messages"] = [
{"role": "system", "content": st.secrets.AppSettings.chatbot_setting}
]
# チャットボットとやりとりする関数
def communicate():
messages = st.session_state["messages"]
user_message = {"role": "user", "content": st.session_state["user_input"]}
messages.append(user_message)
response = openai.ChatCompletion.create(
model="gpt-4.0-turbo",
messages=messages
)
bot_message = response["choices"][0]["message"]
messages.append(bot_message)
st.session_state["user_input"] = "" # 入力欄を消去
# ユーザーインターフェイスの構築
st.title("My AI Assistant")
st.write("ChatGPT APIを使ったチャットボットです。")
user_input = st.text_input("メッセージを入力してください。", key="user_input", on_change=communicate)
if st.session_state["messages"]:
messages = st.session_state["messages"]
for message in reversed(messages[1:]): # 直近のメッセージを上に
speaker = "🙂"
if message["role"]=="assistant":
speaker="🤖"
st.write(speaker + ": " + message["content"])