【AIGC 应用之线稿生图】约稿插画师的得利助手！

npm install tencentcloud-sdk-nodejs-aiart

// Depends on tencentcloud-sdk-nodejs version 4.0.3 or higher
import tencentcloud from "tencentcloud-sdk-nodejs-aiart"
const AiartClient = tencentcloud.aiart.v20221229.Client;
// 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
// 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
// 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
const clientConfig = {
    credential: {
        secretId: "AKIDjIcgU1HI2VhcHxxxxxxxxxxxxxxxxxxxxx",
        secretKey: "KOBhwykFagX8UaS5Sxxxxxxxxxxxxxxxx",
    },
    region: "ap-shanghai",
    profile: {
        httpProfile: {
            endpoint: "aiart.tencentcloudapi.com",//接口请求域名
        },
    },
};
// 实例化要请求产品的client对象,clientProfile是可选的
const client = new AiartClient(clientConfig);

/**
 * 线稿生图请求
 * @param {string} prompt - 用于线稿生图的文本描述。建议格式：线稿中的主体对象+画面场景+配色/材质/元素/风格等
 * @param {string} inputImage - 线稿图 Base64 数据。图片限制：黑白线稿图片，512<单边分辨率<5000，小于 6MB……
 * @returns {SketchToImageResponse} 返回结果
 * @throws {Error} 如果请求失败或解析失败，抛出错误
 */
export async function sketchToImage(prompt, inputImage) {
    const params = {
        "Prompt": prompt,
        "InputImage": inputImage,
        "RspImgType": "url"
    };
    try {
        const response = await client.SketchToImage(params)
        if (!response.RequestId) {
            throw new Error(`HTTP error!`);
        }
        return response;
    } catch (error) {
        console.error(`Error fetching`, error);
        throw error;
    }
}
// 采用ES6写法，需在package.json文件中加入 "type": "module"

// saveImageFromUrl.js
import https from 'https';
import fs from 'fs';
/**
 * 从指定的URL获取图片并保存到本地文件
 * @param {string} url - 图片的URL
 * @param {string} outputFilePath - 保存图片的本地文件路径
 * @returns {Promise<void>}
 */
export async function saveImageFromUrl(url, outputFilePath) {
    return new Promise((resolve, reject) => {
        const file = fs.createWriteStream(outputFilePath);
        https.get(url, (response) => {
            if (response.statusCode !== 200) {
                reject(new Error(`Failed to get image, status code: ${response.statusCode}`));
                return;
            }
            response.pipe(file);
            file.on('finish', () => {
                file.close(resolve);  // 文件写入完成并关闭
            });
        }).on('error', (err) => {
            fs.unlink(outputFilePath, () => {});  // 删除文件
            reject(err);
        });
    });
}

import { sketchToImage } from "./tencentCloudApi.js"
import { readFileSync } from "fs"
import { saveImageFromUrl } from './saveImageFromUrl.js';

// 本文以测试为例，将目标图片已存放至项目根目录，命名为test.jpg
var filePath = "test.jpg";
var base64Str = readFileSync(filePath, 'base64');
// 本文以测试为例，固定输入了如下提示语。建议格式：线稿中的主体对象+画面场景+配色/材质/元素/风格等
const prompt = '几艘帆船在海上航行，晴空万里，深蓝色的大海，白色的浪花翻滚，海鸥，蓝天白云，色彩绚丽，高清，绘本，童话';

const data = await sketchToImage(prompt, base64Str);
console.log(data)

const url = data.ResultImage
const timestamp = new Date().getTime().toString();
const outputFilePath = timestamp + '.jpg';

saveImageFromUrl(url, outputFilePath)
    .then(() => {
        console.log(`Image saved to ${outputFilePath}`);
    })
    .catch((err) => {
        console.error('Error saving image:', err);
    });

node index.js

{
  ResultImage: 'https://aiart-1258344699.cos.ap-guangzhou.myqcloud.com/sketch_to_img%2F1258618245%2Fe6da2563-668a-46f0-a4df-7a75a31129fa-1258618245.jpg?q-sign-algorithm=sha1&q-ak=AKIDpRovliU1IJ5ctufBSVIq8AwTlnZ5MN8d&q-sign-time=1726134391%3B1726137991&q-key-time=1726134391%3B1726137991&q-header-list=host&q-url-param-list=&q-signature=d484398ea6950ddccf6e9ca974c44ab336cedf0f',
  RequestId: 'e6da2563-668a-46f0-a4df-7a75a31129fa'
}
Image saved to 1726134392625.jpg