haoqi7/question

1import streamlit as st2from streamlit_chat import message3import os4import openai5 6#页面设置7st.set_page_config(8    page_title="AI对话一问一答模式",9    page_icon=":robot:"10)11st.header("🔥AI对话一问一答模式")12 13#检查账号登陆14def get_text1():15    if 'openai_key' not in st.session_state:16        input_text1 = st.text_input("📫请输入你的账号: ", key="input")17        if st.button("确认登陆！", key="input3"):18            st.session_state['openai_key'] = input_text119            return input_text120    else:21        return st.session_state['openai_key']22 23openai_key = get_text1()24if openai_key:25    openai.api_key = openai_key26    st.write("")27 28# openAI code29def openai_create(prompt):30    try:31 32        response = openai.Completion.create(33        model="text-davinci-003",34        prompt=prompt,35        temperature=0.5,36        max_tokens=1024,37        top_p=1,38        frequency_penalty=0,39        presence_penalty=0.6,40        stop=[" Human:", " AI:"]41        )42 43        return response.choices[0].text44 45    # except openai.error.UnauthorizedError:46    #     return "API Key不正确，请输入正确的API Key。"47    # except openai.error.RateLimitError:48    #     Error="当前访问量过大，请过一段时间再试！开发者正在努力扩张服务能力，敬请期待！"49    #     return Error50 51    except Exception as e:52        return "你的账号填写有误，请刷新页面重新填写正确的账号！"53 54 55def chatgpt_clone(input):56    output = openai_create(input)57    return output58 59#  Streamlit App60# st.set_page_config(61#     page_title="AI对话单轮版",62#     page_icon=":robot:"63# )64 65 66 67 68# st.header("🏅聊天")69 70if 'generated' not in st.session_state:71    st.session_state['generated'] = []72 73if 'past' not in st.session_state:74    st.session_state['past'] = []75 76 77def get_text():78    input_text = st.text_input("📫你想说的: ", key="input1")79    if st.button("发送", key="input2"):80        return input_text81    return None82 83 84user_input = get_text()85 86if user_input:87     88    output = chatgpt_clone(user_input)89    st.session_state.past.append(user_input)90    st.session_state.generated.append(output)91 92if st.session_state['generated']:93 94    for i in range(len(st.session_state['generated'])-1, -1, -1):95        message(st.session_state["generated"][i], key=str(i))96        message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')97