File size: 2,431 Bytes
c4a0365 38c6288 c4a0365 38c6288 c4a0365 38c6288 c4a0365 38c6288 c4a0365 38c6288 c4a0365 |
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 |
import streamlit as st
from streamlit_chat import message
import os
import openai
#页面设置
st.set_page_config(
page_title="AI对话一问一答模式",
page_icon=":robot:"
)
st.header("🔥AI对话一问一答模式")
#检查账号登陆
def get_text1():
if 'openai_key' not in st.session_state:
input_text1 = st.text_input("📫请输入你的账号: ", key="input")
if st.button("确认登陆！", key="input3"):
st.session_state['openai_key'] = input_text1
return input_text1
else:
return st.session_state['openai_key']
openai_key = get_text1()
if openai_key:
openai.api_key = openai_key
st.write("")
# openAI code
def openai_create(prompt):
try:
response = openai.Completion.create(
model="text-davinci-003",
prompt=prompt,
temperature=0.5,
max_tokens=1024,
top_p=1,
frequency_penalty=0,
presence_penalty=0.6,
stop=[" Human:", " AI:"]
)
return response.choices[0].text
# except openai.error.UnauthorizedError:
# return "API Key不正确，请输入正确的API Key。"
# except openai.error.RateLimitError:
# Error="当前访问量过大，请过一段时间再试！开发者正在努力扩张服务能力，敬请期待！"
# return Error
except Exception as e:
return "你的账号填写有误，请刷新页面重新填写正确的账号！"
def chatgpt_clone(input):
output = openai_create(input)
return output
# Streamlit App
# st.set_page_config(
# page_title="AI对话单轮版",
# page_icon=":robot:"
# )
# st.header("🏅聊天")
if 'generated' not in st.session_state:
st.session_state['generated'] = []
if 'past' not in st.session_state:
st.session_state['past'] = []
def get_text():
input_text = st.text_input("📫你想说的: ", key="input1")
if st.button("发送", key="input2"):
return input_text
return None
user_input = get_text()
if user_input:
output = chatgpt_clone(user_input)
st.session_state.past.append(user_input)
st.session_state.generated.append(output)
if st.session_state['generated']:
for i in range(len(st.session_state['generated'])-1, -1, -1):
message(st.session_state["generated"][i], key=str(i))
message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
|