@rohitaryal/whisk-api 3.0.1 → 3.1.0

package/dist/Whisk.d.ts

@@ -1,5 +1,6 @@
 import { Media } from "./Media.js";
 import { Project } from "./Project.js";
+import { PromptConfig } from "./Types.js";
 export declare class Account {
     private cookie;
     private authToken?;
@@ -43,4 +44,9 @@ export declare class Whisk {
     * @param projectName Name of the project
     */
     newProject(projectName?: string): Promise<Project>;
+    /**
+     * Uses imagefx's api to generate image.
+     * Advantage here is it can generate multiple images in single request
+     */
+    generateImage(input: string | PromptConfig, count?: number): Promise<Media>;
 }

package/dist/Whisk.js

@@ -1,3 +1,4 @@
+import { ImageGenerationModel } from "./Constants.js";
 import { Media } from "./Media.js";
 import { Project } from "./Project.js";
 import { request } from "./Utils.js";
@@ -162,4 +163,66 @@ export class Whisk {
         });
         return new Project(projectInfo.workflowId, this.account);
     }
+    /**
+     * Uses imagefx's api to generate image.
+     * Advantage here is it can generate multiple images in single request
+     */
+    async generateImage(input, count = 1) {
+        if (typeof input === "string") {
+            input = {
+                seed: 0,
+                prompt: input,
+                model: "IMAGEN_3_5",
+                aspectRatio: "IMAGE_ASPECT_RATIO_LANDSCAPE",
+            };
+        }
+        if (!input.seed)
+            input.seed = 0;
+        if (!count)
+            count = 1;
+        if (!input.model)
+            input.model = "IMAGEN_3_5";
+        if (!input.aspectRatio)
+            input.aspectRatio = "IMAGE_ASPECT_RATIO_LANDSCAPE";
+        if (!input.prompt?.trim?.())
+            throw new Error("prompt is required");
+        if (!Object.values(ImageGenerationModel).includes(input.model)) {
+            throw new Error(`'${input.model}': invalid image generation model provided for imagefx`);
+        }
+        if (count < 1 || count > 8) {
+            throw new Error(`'${count}': image generation count must be between 1 and 8 (1 <= count <= 8)`);
+        }
+        const generationResponse = await request("https://aisandbox-pa.googleapis.com/v1:runImageFx", {
+            headers: { "authorization": `Bearer ${await this.account.getToken()}` },
+            body: JSON.stringify({
+                "userInput": {
+                    "candidatesCount": count,
+                    "prompts": [input.prompt],
+                    "seed": input.seed
+                },
+                "clientContext": {
+                    "sessionId": ";1768371666629",
+                    "tool": "IMAGE_FX"
+                },
+                "modelInput": {
+                    "modelNameType": input.model
+                },
+                "aspectRatio": input.aspectRatio
+            }),
+            method: "POST",
+        });
+        return generationResponse.imagePanels[0].generatedImages.map((img) => {
+            return new Media({
+                seed: img.seed,
+                prompt: img.prompt,
+                workflowId: img.workflowId,
+                encodedMedia: img.encodedImage,
+                mediaGenerationId: img.mediaGenerationId,
+                aspectRatio: img.aspectRatio,
+                mediaType: "IMAGE",
+                model: img.modelNameType,
+                account: this.account
+            });
+        });
+    }
 }

package/dist/index.js

@@ -1,5 +1,4 @@
 #!/usr/bin/env node
-// Re-export your public API
 export { Whisk } from "./Whisk.js";
 export { Media } from "./Media.js";
 export { Project } from "./Project.js";

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@rohitaryal/whisk-api",
-    "version": "3.0.1",
+    "version": "3.1.0",
     "type": "module",
     "main": "./dist/index.js",
     "types": "./dist/index.d.ts",