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

前言

都说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,Image
openai.api_key = "sk-xxx"  # 更换为自己的密钥


def ChatGPT_AI_DrawBot(input):
    if input:
        response = openai.Image.create(
          prompt=input,
          n=1,
          size="1024x1024"
        )
        image_url = response['data'][0]['url']
        return image_url

inputs = Textbox(lines=7, label="请输入你的问题")
outputs = Image(label="ChatGPT AI结果")

gr.Interface(fn=ChatGPT_AI_DrawBot, inputs=inputs, outputs=outputs, title="ChatGPT AI助理",
             description="我是您的AI助理，输入您的绘图关键字，我将为您进行关键字的AI绘图",
             theme=gr.themes.Default()).launch(share=True)

运行后出现以下提示即为成功。

按Ctrl并鼠标点击http://127.0.0.1:7860 即可打开浏览器访问该网址。

给ChatGPT写入一段话：

在一片巨大的森林里，数十棵巨大的百年橡树，金色的秋天，浩瀚的晨光，穿过树枝的天空，一棵长着翡翠叶子的石头树，地面上有巨大的水晶和钻石，所有这些都有多种颜色，淡蓝色的雾，金色的部分，细节，真实感，照片

Submit后，ChatGPT运算后会将生成的图片展示在右侧。点击图片右上角即可下载该图片（直接下载会保存为image.txt，修改扩展名为jpg后即可显示为图片）。如下所示：

最终生成的1024*1024像素图片：

总结

这里是ChatGPT 3.5引擎生成的图片，仅作一个展示，抛砖引玉。期待大家更好的作品！