我用Streamlit+ChatGPT轻松实现Web聊天

阿里云国内75折回扣微信号：monov8

阿里云国际，腾讯云国际，低至75折。AWS 93折免费开户实名账号代冲值优惠多多微信号：monov8 飞机：@monov6

Streamlit是时下比较热门的一个基于Python的Web应用程序框架,它可以在几分钟内将数据转化为可共享的Web应用程序无需前端开发经验使用纯Python代码实现简单且高效。ChatGPT是目前非常火的OpenAI公司开发的聊天机器人模型它无所不知就像一本大百科全书它可以帮你做很多繁杂的日常工作比如可以代你写文章代你做excel表格甚至代你写代码。今天我们要将两者结合起来开发一个基于web的应用聊天小程序。

注册Openai账号申请api_key

由于ChatGPT平台并没有对中国开放因此我们无法按照正常流程通过OpenAI官网来注册OpenAI账号但是可以通过一些变通的方法(你懂的)或者请国外的亲朋好友代你注册账号和申请api_key,一旦我们拿到了api_key就不再受网络的限制可以正常通过api来和ChatGPT机器人聊天了。

安装OpenAPI和Streamlit包

我们需要在python环境中安装openai和streamlit的第三方python包可以通过在命令行窗口中安装这些包

pip install openai
pip install streamlit
pip install streamlit_chat

聊天机器人API接口参数说明

model模型名词
prompt您对机器人提出的问题
temperature温度参数该参数控制生成文本的随机性级别。较高的温度参数会导致更多变化且可能不太连贯的响应而较低的t温度参数会产生更可预测且可能更连贯的响应。
max_tokens应答语句的长度

创建聊天Python代码文件

我们需要创建一个用于聊天的streamlit的代码文件 chat_bot.py:

# chat_bot.py

import openai
import streamlit as st
from streamlit_chat import message

#申请的api_key
openai.api_key = "xxxxxxxxxxxxxxxxx" 
def generate_response(prompt):
    completion=openai.Completion.create(
        model='text-davinci-003',
        prompt=prompt,
        max_tokens=1024,
        temperature=0.6
    )
    message=completion.choices[0].text
    return message

st.markdown("#### 我是ChatGPT聊天机器人,我可以回答您的任何问题")
if 'generated' not in st.session_state:
    st.session_state['generated'] = []
if 'past' not in st.session_state:
    st.session_state['past'] = []
user_input=st.text_input("请输入您的问题:",key='input')
if user_input:
    output=generate_response(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')