-
Notifications
You must be signed in to change notification settings - Fork 126
/
Copy pathGenerate an AI image.js
75 lines (69 loc) · 2.58 KB
/
Generate an AI image.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/*
activation_example:!dalle minty donut
regex:!dalle
flags:gmi
*/
var prompt = current.text.replace(/!dalle/gmi, "").trim().substring(0, 1000);
var dalleReq = new sn_ws.RESTMessageV2();
dalleReq.setEndpoint('https://api.openai.com/v1/images/generations');
dalleReq.setHttpMethod("POST");
dalleReq.setRequestHeader("Authorization", "Bearer " + gs.getProperty("openai.key"));
dalleReq.setRequestHeader('Content-Type', "application/json");
dalleReq.setRequestHeader('User-Agent', "ServiceNow");
dalleReq.setRequestHeader("Accept", "*/*");
var body = {
"prompt": prompt,
"n": 1,
"size": "1024x1024",
quality: "standard",
model: "dall-e-3",
"response_format": "url",
"user": current.user.name.toString()
};
dalleReq.setRequestBody(JSON.stringify(body));
var dalleResponse = dalleReq.execute();
var dalleResponseStatusCode = dalleResponse.getStatusCode();
var dalleResponseBody = JSON.parse(dalleResponse.getBody());
var failureMessage = "Image Generation failed. If this is your first error please try again, otherwise contact an admin";
if (dalleResponseStatusCode != 200) {
new x_snc_slackerbot.Slacker().send_chat(current, failureMessage);
if (gs.getProperty("generateaiimage.debug.enable", "true") == "true") {
gs.error("Error: " + dalleResponseStatusCode + " \n" + dalleResponseBody.error.message + "\n" + "Prompt: " + prompt);
}
} else {
//Cloudinary
var cloudinary = new sn_ws.RESTMessageV2();
cloudinary.setEndpoint('https://api.cloudinary.com/v1_1/dxllc568e/image/upload');
cloudinary.setHttpMethod('POST');
cloudinary.setRequestHeader('Content-Type', "application/json");
cloudinary.setRequestHeader('User-Agent', "ServiceNow");
cloudinary.setRequestHeader("Accept", "*/*");
var cloudinaryBody = {
"file": dalleResponseBody.data[0].url,
"upload_preset": gs.getProperty("cloudinary_upload_preset"),
"api_key": gs.getProperty("cloudinary_api_key"),
"timestamp": Math.floor(new Date().getTime() / 1000),
};
cloudinary.setRequestBody(JSON.stringify(cloudinaryBody));
var cloudinaryResponse = cloudinary.execute();
var cloudinaryResponseBody = JSON.parse(cloudinaryResponse.getBody());
var message = {
"blocks": [
{
"type": "image",
"title": {
"type": "plain_text",
"text": prompt
},
"block_id": "image4",
"image_url": cloudinaryResponseBody.url,
"alt_text": prompt
}
]
}
if (cloudinaryResponseBody.url) {
new x_snc_slackerbot.Slacker().send_chat(current, message);
} else {
new x_snc_slackerbot.Slacker().send_chat(current, failureMessage);
}
}