【Python小技巧】使用Gradio构建基于ChatGPT的 Web 应用（附源码）

编程知识更新时间:2023-05-02 18:11:38

文章目录

前言
一、Gradio是什么？
二、使用Gradio构建基于ChatGPT的 Web 应用
- 1. 安装gradio库
- 2. 安装openai库（ChatGPT的python库）
- 3. Web 应用示例（源代码）
总结

前言

随着人工智能的不断发展，各种智能算法越来越普遍，但是这些算法结果通常显示在cmd命令窗口里。有没有一种方法可以动态展示，更具需要计算后动态展现？

答案是有！

下面让我了解一下Gradio库，只需寥寥几行代码就可以展现出ChatGPT的对话窗口。经过包装，在本地就可以通过Python访问openai并将结果展现。

一、Gradio是什么？

Gradio是一个开源的 Python 库，用于构建机器学习和数据科学演示和 Web 应用。

官网：https://www.gradio.app/

Gradio适用于:

演示客户/合作者/用户/学生的机器学习模型。
部署通过自动共享链接快速创建模型，并获得模型性能反馈。
排除故障使用内置的操作和解释工具，在开发过程中以交互方式处理您的模型。

案例展示：https://huggingface.co/spaces/hysts/insightface-SCRFD
安装包：https://pypi/project/gradio/
Github：https://github/gradio-app/gradio

二、使用Gradio构建基于ChatGPT的 Web 应用

1. 安装gradio库

pip install gradio -i https://pypi.tuna.tsinghua.edu/simple

2. 安装openai库（ChatGPT的python库）

pip install openai -i https://pypi.tuna.tsinghua.edu/simple

请提前申请好账号，并生成密钥。

3. Web 应用示例（源代码）

注：openai.api_key 需更换为自己的密钥

import openai
import gradio as gr
from gradio.components import Textbox
openai.api_key = "sk-xxx"  # 更换为自己的密钥

messages = [
    {"role": "system", "content": "You are a helpful and kind AI Assistant."},
]

def ChatGPT_Bot(input):
    if input:
        messages.append({"role": "user", "content": input})
        chat = openai.ChatCompletion.create(
            model="gpt-3.5-turbo", messages=messages
        )
        reply = chat.choices[0].message.content
        messages.append({"role": "assistant", "content": reply})
        return reply

inputs = Textbox(lines=7, label="请输入你的问题")
outputs = Textbox(lines=7, label="来自ChatGPT的回答")

gr.Interface(fn=ChatGPT_Bot, inputs=inputs, outputs=outputs, title="ChatGPT AI助理",
             description="我是您的AI助理，您可以问任何你想知道的问题",
             theme=gr.themes.Default()).launch(share=True)