mirror of
https://github.com/vastxie/99AI.git
synced 2025-09-18 17:56:36 +08:00
82 lines
3.8 KiB
JavaScript
82 lines
3.8 KiB
JavaScript
"use strict";
|
|
var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
|
|
var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
|
|
if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
|
|
else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
|
|
return c > 3 && r && Object.defineProperty(target, key, r), r;
|
|
};
|
|
var __metadata = (this && this.__metadata) || function (k, v) {
|
|
if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
|
|
};
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.DrawService = void 0;
|
|
const common_1 = require("@nestjs/common");
|
|
const axios_1 = require("axios");
|
|
const uuid = require("uuid");
|
|
const upload_service_1 = require("../upload/upload.service");
|
|
let DrawService = class DrawService {
|
|
constructor(uploadService) {
|
|
this.uploadService = uploadService;
|
|
}
|
|
async onModuleInit() {
|
|
var _a;
|
|
this.apiHost = (_a = process.env.API_HOST) !== null && _a !== void 0 ? _a : 'https://api.stability.ai';
|
|
this.apiKey = process.env.STABILITY_API_KEY;
|
|
if (!this.apiKey) {
|
|
this.apiKey = '*********';
|
|
}
|
|
this.Authorization = `Bearer ${this.apiKey}`;
|
|
}
|
|
async getEngines() {
|
|
var _a, _b;
|
|
const url = `${this.apiHost}/v1/engines/list`;
|
|
const res = await (0, axios_1.default)(url, {
|
|
method: 'GET',
|
|
headers: { Authorization: this.Authorization },
|
|
});
|
|
if (res.status === 401) {
|
|
console.log(`stability api key is invalid, ${(_a = res === null || res === void 0 ? void 0 : res.data) === null || _a === void 0 ? void 0 : _a.message}`);
|
|
}
|
|
if (res.status !== 200) {
|
|
console.log(`${res.status} ${(_b = res === null || res === void 0 ? void 0 : res.data) === null || _b === void 0 ? void 0 : _b.message}}`);
|
|
throw new common_1.HttpException('获取列表失败', common_1.HttpStatus.BAD_REQUEST);
|
|
}
|
|
return res.data;
|
|
}
|
|
async drawTextToImage(body) {
|
|
const { engineId = 'stable-diffusion-768-v2-1' } = body;
|
|
const headers = {
|
|
'Content-Type': 'application/json',
|
|
Accept: 'application/json',
|
|
Authorization: this.Authorization,
|
|
};
|
|
const url = `${this.apiHost}/v1/generation/${engineId}/text-to-image`;
|
|
try {
|
|
const response = await axios_1.default.post(url, body, { headers });
|
|
if (response.status !== 200) {
|
|
throw new common_1.HttpException('绘制失败', common_1.HttpStatus.BAD_REQUEST);
|
|
}
|
|
const resImageBasetask = [];
|
|
for (const item of response.data.artifacts) {
|
|
const filename = uuid.v4().slice(0, 10) + '.png';
|
|
const buffer = Buffer.from(item.base64, 'base64');
|
|
resImageBasetask.push(this.uploadService.uploadFile({ filename, buffer }));
|
|
}
|
|
const urls = await Promise.all(resImageBasetask);
|
|
return urls;
|
|
}
|
|
catch (error) {
|
|
if (!(error === null || error === void 0 ? void 0 : error.response)) {
|
|
throw new common_1.HttpException('绘制失败', common_1.HttpStatus.BAD_REQUEST);
|
|
}
|
|
const { status, data } = error.response;
|
|
throw new common_1.HttpException(data.message, status);
|
|
}
|
|
}
|
|
};
|
|
DrawService = __decorate([
|
|
(0, common_1.Injectable)(),
|
|
__metadata("design:paramtypes", [upload_service_1.UploadService])
|
|
], DrawService);
|
|
exports.DrawService = DrawService;
|