aboutsummaryrefslogtreecommitdiff
path: root/src/server/ApiManagers/FireflyManager.ts
diff options
context:
space:
mode:
Diffstat (limited to 'src/server/ApiManagers/FireflyManager.ts')
-rw-r--r--src/server/ApiManagers/FireflyManager.ts53
1 files changed, 47 insertions, 6 deletions
diff --git a/src/server/ApiManagers/FireflyManager.ts b/src/server/ApiManagers/FireflyManager.ts
index 5e3ba1f83..e10e43704 100644
--- a/src/server/ApiManagers/FireflyManager.ts
+++ b/src/server/ApiManagers/FireflyManager.ts
@@ -16,7 +16,7 @@ export default class FireflyManager extends ApiManager {
});
askFirefly = (prompt: string = 'a realistic illustration of a cat coding') => {
const fetched = this.getBearerToken().then(response =>
- response.json().then((data: { access_token: string }) =>
+ (response as Response).json().then((data: { access_token: string }) =>
fetch('https://firefly-api.adobe.io/v3/images/generate', {
method: 'POST',
headers: [
@@ -36,16 +36,57 @@ export default class FireflyManager extends ApiManager {
);
return fetched;
};
+ askFireflyText = (testshotpng: Blob) => {
+ const fetched = this.getBearerToken().then(response =>
+ (response as Response).json().then((data: { access_token: string }) => {
+ return fetch('https://sensei.adobe.io/services/v2/predict', {
+ method: 'POST',
+ headers: [
+ ['Prefer', 'respond-async, wait=59'],
+ ['x-api-key', process.env._CLIENT_FIREFLY_CLIENT_ID ?? ''],
+ ['content-type', 'multipart/form-data'],
+ ['Authorization', `Bearer ${data.access_token}`],
+ ],
+ body: ((form: FormData) => {
+ form.append('file', testshotpng);
+ form.append(
+ 'contentAnalyzerRequests',
+ JSON.stringify({
+ 'sensei:name': 'Feature:cintel-object-detection:Service-b9ace8b348b6433e9e7d82371aa16690',
+ })
+ );
+ return form;
+ })(new FormData()),
+ }).then(response2 =>
+ response2
+ .json()
+ .then(json => {
+ console.log(json);
+ return '';
+ })
+ .catch(error => {
+ console.error('Error:', error);
+ return '';
+ })
+ );
+ })
+ );
+ return fetched;
+ };
protected initialize(register: Registration): void {
register({
method: Method.POST,
subscription: '/queryFireflyImage',
secureHandler: ({ req, res }) =>
- this.askFirefly(req.body.prompt).then(fire =>
- DashUploadUtils.UploadImage(JSON.parse(fire).url).then(info => {
- if (info instanceof Error) _invalid(res, info.message);
- else _success(res, info.accessPaths.agnostic.client);
- })
+ fetch('http://localhost:1050/files/images/testshot.png').then(json =>
+ json.blob().then(file =>
+ this.askFireflyText(file).then(fire =>
+ DashUploadUtils.UploadImage(JSON.parse(fire).url).then(info => {
+ if (info instanceof Error) _invalid(res, info.message);
+ else _success(res, info.accessPaths.agnostic.client);
+ })
+ )
+ )
),
});
}