@rohitaryal/whisk-api/dist/Project.js

Unofficial API for Whisk image generation.

import { ImageGenerationModel } from "./Constants.js";
import { Media } from "./Media.js";
import { request } from "./Utils.js";
import { Account } from "./Whisk.js";
export class Project {
    account;
    projectId;
    constructor(projectId, account) {
        if (typeof projectId !== "string" || !projectId.trim()) {
            throw new Error("project id is either invalid or missing");
        }
        if (!(account instanceof Account)) {
            throw new Error("account is invalid or missing");
        }
        this.projectId = projectId;
        this.account = account;
    }
    async generateImage(input) {
        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 (!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`);
        }
        const generationResponse = await request("https://aisandbox-pa.googleapis.com/v1/whisk:generateImage", {
            headers: { authorization: `Bearer ${await this.account.getToken()}` },
            body: JSON.stringify({
                "clientContext": {
                    "workflowId": this.projectId
                },
                "imageModelSettings": {
                    "imageModel": input.model,
                    "aspectRatio": input.aspectRatio
                },
                "seed": input.seed,
                "prompt": input.prompt,
                "mediaCategory": "MEDIA_CATEGORY_BOARD"
            })
        });
        const img = generationResponse.imagePanels[0].generatedImages[0];
        return new Media({
            seed: img.seed,
            prompt: img.prompt,
            workflowId: img.workflowId ?? generationResponse.workflowId,
            encodedMedia: img.encodedImage,
            mediaGenerationId: img.mediaGenerationId,
            aspectRatio: img.aspectRatio,
            mediaType: "IMAGE",
            model: img.imageModel,
            account: this.account
        });
    }
    /**
    * Deletes the project, clearance of your slop from the existance
    */
    async delete() {
        await request("https://labs.google/fx/api/trpc/media.deleteMedia", {
            headers: { cookie: this.account.getCookie() },
            body: JSON.stringify({
                "json": {
                    "parent": "userProject/", "names": [this.projectId]
                }
            }),
        });
    }
}