[Components] letzai #16541

Sources
 - New Image
 - New Image Edit

Actions
 - Create New Image
 - Get Image Information
 - Create Image Edit

Author: luancazarine

components/letzai/actions/create-image-edit/create-image-edit.mjs

@@ -1,11 +1,11 @@
+import { parseObject } from "../../common/utils.mjs";
 import letzai from "../../letzai.app.mjs";
-import { axios } from "@pipedream/platform";
 
 export default {
   key: "letzai-create-image-edit",
   name: "Create Image Edit",
-  description: "Creates an image edit task that modifies an existing image using inpainting or outpainting in LetzAI. [See the documentation](https://letz.ai/docs/api)",
-  version: "0.0.{{ts}}",
+  description: "Creates an image edit task that modifies an existing image using inpainting or outpainting in LetzAI. [See the documentation](https://api.letz.ai/doc#/image-edit/image_edit_create)",
+  version: "0.0.1",
   type: "action",
   props: {
     letzai,
@@ -14,6 +14,7 @@ export default {
         letzai,
         "mode",
       ],
+      reloadProps: true,
     },
     originalImageCompletionId: {
       propDefinition: [
@@ -79,22 +80,27 @@ export default {
       optional: true,
     },
   },
+  async additionalProps(props) {
+    if (this.mode === "in") {
+      props.mask.optional = false;
+    }
+    return {};
+  },
   async run({ $ }) {
-    const data = {
-      mode: this.mode,
-      original_image_completion_id: this.originalImageCompletionId,
-      image_url: this.imageUrl,
-      prompt: this.prompt,
-      mask: this.mask,
-      width: this.width,
-      height: this.height,
-      image_completions_count: this.imageCompletionsCount,
-      settings: this.settings,
-      webhook_url: this.webhookUrl,
-    };
+    const {
+      letzai,
+      settings,
+      ...data
+    } = this;
 
-    const response = await this.letzai.createImageEditTask(data);
-    $.export("$summary", `Image edit task created successfully with ID: ${response.id}`);
+    const response = await letzai.createImageEditTask({
+      $,
+      data: {
+        ...data,
+        settings: settings && parseObject(settings),
+      },
+    });
+    $.export("$summary", `Image edit task created successfully with request ID: ${response.id}`);
     return response;
   },
 };

components/letzai/actions/create-new-image/create-new-image.mjs

@@ -1,14 +1,18 @@
 import letzai from "../../letzai.app.mjs";
-import { axios } from "@pipedream/platform";
 
 export default {
   key: "letzai-create-new-image",
   name: "Create New Image",
-  description: "Creates a new image generation task from a text prompt in LetzAI. [See the documentation](https://letz.ai/docs/api#createImages)",
-  version: "0.0.{{ts}}",
+  description: "Creates a new image generation task from a text prompt in LetzAI. [See the documentation](https://api.letz.ai/doc#/images/images_create)",
+  version: "0.0.1",
   type: "action",
   props: {
     letzai,
+    info: {
+      type: "alert",
+      alertType: "info",
+      content: "**Note:** You can monitor the generation status using the Action \"Get Image Information\".",
+    },
     prompt: {
       propDefinition: [
         letzai,
@@ -20,57 +24,61 @@ export default {
         letzai,
         "width",
       ],
+      optional: true,
     },
     height: {
       propDefinition: [
         letzai,
         "height",
       ],
+      optional: true,
     },
     quality: {
       propDefinition: [
         letzai,
         "quality",
       ],
+      optional: true,
     },
     creativity: {
       propDefinition: [
         letzai,
         "creativity",
       ],
+      optional: true,
     },
     hasWatermark: {
       propDefinition: [
         letzai,
         "hasWatermark",
       ],
+      optional: true,
     },
     systemVersion: {
       propDefinition: [
         letzai,
         "systemVersion",
       ],
+      optional: true,
     },
     mode: {
       propDefinition: [
         letzai,
         "generationMode",
       ],
+      optional: true,
     },
   },
   async run({ $ }) {
-    const data = {
-      prompt: this.prompt,
-      width: this.width,
-      height: this.height,
-      quality: this.quality,
-      creativity: this.creativity,
-      hasWatermark: this.hasWatermark,
-      systemVersion: this.systemVersion,
-      mode: this.mode,
-    };
+    const {
+      letzai,
+      ...data
+    } = this;
 
-    const response = await this.letzai.createImageGenerationTask(data);
+    const response = await letzai.createImage({
+      $,
+      data,
+    });
     $.export("$summary", `Created image with ID: ${response.id}`);
     return response;
   },

components/letzai/actions/get-image-information/get-image-information.mjs

@@ -1,11 +1,10 @@
 import letzai from "../../letzai.app.mjs";
-import { axios } from "@pipedream/platform";
 
 export default {
   key: "letzai-get-image-information",
   name: "Get Image Information",
-  description: "Retrieves information about a specific image by ID. [See the documentation](https://api.letz.ai/docs/api)",
-  version: "0.0.{{ts}}",
+  description: "Retrieves information about a specific image by ID. [See the documentation](https://api.letz.ai/doc#/images/images_get)",
+  version: "0.0.1",
   type: "action",
   props: {
     letzai,
@@ -17,7 +16,10 @@ export default {
     },
   },
   async run({ $ }) {
-    const response = await this.letzai.retrieveImageInfo(this.imageId);
+    const response = await this.letzai.retrieveImageInfo({
+      $,
+      imageId: this.imageId,
+    });
     $.export("$summary", `Successfully retrieved information for image ID: ${this.imageId}`);
     return response;
   },

components/letzai/common/constants.mjs

@@ -0,0 +1,34 @@
+export const LIMIT = 50;
+
+export const GENERATION_MODE_OPTIONS = [
+  {
+    label: "Default",
+    value: "default",
+  },
+  {
+    label: "Sigma",
+    value: "sigma",
+  },
+];
+
+export const SYSTEM_VERSION_OPTIONS = [
+  {
+    label: "Version 2",
+    value: 2,
+  },
+  {
+    label: "Version 3",
+    value: 3,
+  },
+];
+
+export const MODE_OPTIONS = [
+  {
+    label: "Inpainting",
+    value: "in",
+  },
+  {
+    label: "Outpainting",
+    value: "out",
+  },
+];

components/letzai/common/utils.mjs

@@ -0,0 +1,24 @@
+export const parseObject = (obj) => {
+  if (!obj) return undefined;
+
+  if (Array.isArray(obj)) {
+    return obj.map((item) => {
+      if (typeof item === "string") {
+        try {
+          return JSON.parse(item);
+        } catch (e) {
+          return item;
+        }
+      }
+      return item;
+    });
+  }
+  if (typeof obj === "string") {
+    try {
+      return JSON.parse(obj);
+    } catch (e) {
+      return obj;
+    }
+  }
+  return obj;
+};