aimaswx commited on
Commit
eff5e28
·
1 Parent(s): 2729aa6

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +34 -0
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')