Spaces:
Runtime error
Runtime error
Create app.py
Browse files
app.py
ADDED
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# chat_bot.py
|
2 |
+
|
3 |
+
import openai
|
4 |
+
import streamlit as st
|
5 |
+
from streamlit_chat import message
|
6 |
+
|
7 |
+
#申请的api_key
|
8 |
+
openai.api_key = "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"
|
9 |
+
def generate_response(prompt):
|
10 |
+
completion = openai.Completion.create(
|
11 |
+
model="text-davinci-003", #'gpt-3.5-turbo',
|
12 |
+
prompt=prompt,
|
13 |
+
max_tokens=1024,
|
14 |
+
temperature=0.6
|
15 |
+
)
|
16 |
+
message=completion.choices[0].text
|
17 |
+
return message
|
18 |
+
|
19 |
+
st.markdown("#### 我是ChatGPT聊天机器人,我可以回答您的任何问题!")
|
20 |
+
if 'generated' not in st.session_state:
|
21 |
+
st.session_state['generated'] = []
|
22 |
+
if 'past' not in st.session_state:
|
23 |
+
st.session_state['past'] = []
|
24 |
+
user_input=st.text_input("请输入您的问题:",key='input')
|
25 |
+
if user_input:
|
26 |
+
output=generate_response(user_input)
|
27 |
+
st.session_state['past'].append(user_input)
|
28 |
+
st.session_state['generated'].append(output)
|
29 |
+
if st.session_state['generated']:
|
30 |
+
for i in range(len(st.session_state['generated'])-1, -1, -1):
|
31 |
+
message(st.session_state["generated"][i], key=str(i))
|
32 |
+
message(st.session_state['past'][i],
|
33 |
+
is_user=True,
|
34 |
+
key=str(i)+'_user')
|