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


前言

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

答案是有!

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

一、Gradio是什么?

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

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

Gradio适用于:

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

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

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

1. 安装gradio库

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

2. 安装openai库(ChatGPT的python库)

pip install openai -i https://pypi.tuna.tsinghua.edu.cn/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)

运行后出现以下提示即为成功。
在这里插入图片描述
按Ctrl并鼠标点击http://127.0.0.1:7860 即可打开浏览器访问该网址。给ChatGPT提个问题,Submit后,她在运算后结果将显示在右侧文本框内。如下所示:

在这里插入图片描述


总结

Gradio 寥寥几行代码就能让原本抽象的算法瞬间变成可视化的交互应用,将原本干巴巴的结果形象地呈现出来,这对于智能算法的演示和结果验证非常有用。当然,Gradio 用法不止于此,ChatGPT更是可以运用丰富,各位可以自己探索。本文仅为抛砖引玉,期待更好的应用展现。

猜你喜欢

转载自blog.csdn.net/popboy29/article/details/130360979
今日推荐