Python 调用API在本地生成图像

最新推荐文章于 2025-04-27 23:00:00 发布

原创最新推荐文章于 2025-04-27 23:00:00 发布 · 517 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

概述

本程序利用硅基流动目前的免费 stabilityai/stable-diffusion-2-1 模型API，生成图像并下载到本地，用户可以通过输入描述性提示词来获取相应的图像。使用Python编写，依赖于requests库进行HTTP请求，并使用os库处理文件系统操作。

功能

用户输入：程序提示用户输入描述图像的提示词，用户输入时要添加单引号。
API请求：使用用户输入的提示词构建请求负载，并向指定的API发送POST请求以生成图像。
响应处理：解析API的响应，提取生成的图像URL。
文件保存：将生成的图像下载并保存到本地文件系统，文件名基于用户输入的前10个字符命名。
目录管理：在保存图像之前，检查并创建必要的目录。

代码结构
以下是应用程序的主要代码结构：

# -*- coding: utf-8 -*-
import requests
import os

url = "https://api.siliconflow.cn/v1/images/generations"

# 提示用户输入描述图片的提示词
try:
    user_input = input("\n请输入描述图片的提示词，例如：'blue sky': ").strip()  # 用户输入不需要引号
    user_prompt = "'{}'".format(user_input)  # 自动为输入添加单引号
except Exception as e:
    print("输入提示词时发生错误: {}".format(e))
    exit(1)  # 退出程序

payload = {
    "model": "deepseek-ai/Janus-Pro-7B",
    "prompt": user_prompt,  # 使用用户输入的提示词
    "seed": 4999999999
}
headers = {
    "Authorization": "Bearer <token>",  #<token> 替换成你在硅基流动里的 API-KEY
    "Content-Type": "application/json"
}

response = requests.request("POST", url, json=payload, headers=headers)

# 解析响应以获取图像URL
response_json = response.json()  # 使用 json() 方法解析响应
image_url = response_json['data'][0]['url']  # 从 data 数组中获取 URL

# 定义保存图像的本地文件名
local_filename = os.path.join("images", "{}.png".format(user_input[:10]))  # 使用用户输入的前10个字符命名

# 创建图像目录（如果不存在）
if not os.path.exists(os.path.dirname(local_filename)):
    os.makedirs(os.path.dirname(local_filename))

# 下载图像并保存到本地
with requests.get(image_url, stream=True) as img_response:
    img_response.raise_for_status()  # 检查请求错误
    with open(local_filename, 'wb') as f:
        for chunk in img_response.iter_content(chunk_size=8192):
            f.write(chunk)

print("Image saved to {local_filename}")