ChatGPT：5. 使用OpenAI API创建自己的AI网站：4. flask框架网页界面上下载OpenAI API请求的图片

如果你还是一个OpenAI的小白，有OpenAI的账号，但想调用OpenAI的API搞一些有意思的事，那么这一系列的教程将仔细的为你讲解如何使用OpenAI的API制作属于自己的AI网站。博主只能利用下班时间更新，进度慢，请海涵🥳。

3. 🎖️网页界面上图片的下载

🐒回顾：上一节，我们已经把OpneAI返回的图片显示到网页界面了，本节我们将添加一个"下载"按钮，单击后，将文件保存到本地。跳转到上一节

🐥1. 先修改Html文件，添加一个button。

<!DOCTYPE html>
<html lang="en">
<head><meta charset="UTF-8"><title>Title</title>
</head>
<body>
<form action="/" method="post"><input type="text" name="desc" placeholder="请输入描述信息"><button>创作图画</button></form>
<!--显示图片-->
<div>{% if result %}<img src="{{ result }}" alt=""><div><a href="/download?img_url={{result|urlencode}}"><button>下载</button></a></div>{% endif %}
</div>
</body>
</html>

🐣2. 后端代码修改如下：

from flask import Flask, render_template, request, redirect, url_for, make_response
import openai
import requests
from uuid import uuid1 # 防止文件名重复，或者可以时间戳
api_key = "sk-e4F4owyBicdBYKA9RuuM*"  # 此处是自己的API Key
openai.api_key = api_keyapp = Flask(__name__)@app.route('/', methods=['GET', 'POST'])
def index():# 接收POST请求if request.method == 'POST':  # 接收表单处理desc = request.form['desc']print(desc)# 此处为OpenAI的API代码response = openai.Image.create(prompt=desc,n=1,size="512x512")image_url = response['data'][0]['url']print(image_url)return redirect(url_for('index', result=image_url))  # 跳转到当前界面下,并将image_url返回到页面中# 接收GET请求elif request.method == 'GET':  # 接收表单处理result = request.args.get("result")return render_template('index.html', result=result)@app.route('/download/')
def download():# 获取urlimg_url = request.args.get('img_url')  # 获取url地址res = requests.get(img_url)  # 得到一个响应if res.status_code != 200:  # 检查响应，成功时应该是200return "请求失败！"# 1. 根据url爬取照片,获取图片二进制文件res = requests.get(url)response = make_response(res.content)  # # 2. 保存文件，类型为图片response.headers['Content-Type'] = 'image/png'response.headers['Content-Disposition'] = f"attachment;filename={uuid1()}.png"return response@app.route('/debug')
def debug():return 'I am Demo.'if __name__ == '__main__':app.run(debug=True, port=8888)