actions/effect/GenerativeBackgroundReplace.ts

import { Action } from "../../internal/Action.js";
import { Qualifier } from "../../internal/qualifier/Qualifier.js";
import { IGenerativeBackgroundReplaceModel } from "../../internal/models/IEffectActionModel.js";

/**
 * @description Uses generative AI to replace background of your image with something else.
 * @extends SDK.Action
 * @memberOf Actions.Effect
 * @see Visit {@link Actions.Effect|Effect} for an example
 */
class GenerativeBackgroundReplace extends Action {
  private _prompt: string;

  constructor() {
    super();
    this._actionModel.actionType = "generativeBackgroundReplace";
  }

  prompt(value: string): GenerativeBackgroundReplace {
    this._prompt = value;
    this._actionModel.prompt = value;

    return this;
  }

  protected prepareQualifiers(): void {
    if (!this._prompt) {
      this.addQualifier(new Qualifier("e", "gen_background_replace"));
    } else {
      this.addQualifier(
        new Qualifier("e", `gen_background_replace:prompt_${this._prompt}`)
      );
    }
  }

  static fromJson(
    actionModel: IGenerativeBackgroundReplaceModel
  ): GenerativeBackgroundReplace {
    const { prompt } = actionModel;
    const result = new this();

    return result.prompt(prompt);
  }
}

export { GenerativeBackgroundReplace };