@m5kdev/backend 0.1.0

package/src/modules/ai/ideogram/ideogram.constants.ts

@@ -0,0 +1,170 @@
+export const ideogramStyleTypes = ["AUTO", "GENERAL", "REALISTIC", "DESIGN", "FICTION"] as const;
+
+export const ideogramStylePresets = [
+  "80S_ILLUSTRATION",
+  "90S_NOSTALGIA",
+  "ABSTRACT_ORGANIC",
+  "ANALOG_NOSTALGIA",
+  "ART_BRUT",
+  "ART_DECO",
+  "ART_POSTER",
+  "AURA",
+  "AVANT_GARDE",
+  "BAUHAUS",
+  "BLUEPRINT",
+  "BLURRY_MOTION",
+  "BRIGHT_ART",
+  "C4D_CARTOON",
+  "CHILDRENS_BOOK",
+  "COLLAGE",
+  "COLORING_BOOK_I",
+  "COLORING_BOOK_II",
+  "CUBISM",
+  "DARK_AURA",
+  "DOODLE",
+  "DOUBLE_EXPOSURE",
+  "DRAMATIC_CINEMA",
+  "EDITORIAL",
+  "EMOTIONAL_MINIMAL",
+  "ETHEREAL_PARTY",
+  "EXPIRED_FILM",
+  "FLAT_ART",
+  "FLAT_VECTOR",
+  "FOREST_REVERIE",
+  "GEO_MINIMALIST",
+  "GLASS_PRISM",
+  "GOLDEN_HOUR",
+  "GRAFFITI_I",
+  "GRAFFITI_II",
+  "HALFTONE_PRINT",
+  "HIGH_CONTRAST",
+  "HIPPIE_ERA",
+  "ICONIC",
+  "JAPANDI_FUSION",
+  "JAZZY",
+  "LONG_EXPOSURE",
+  "MAGAZINE_EDITORIAL",
+  "MINIMAL_ILLUSTRATION",
+  "MIXED_MEDIA",
+  "MONOCHROME",
+  "NIGHTLIFE",
+  "OIL_PAINTING",
+  "OLD_CARTOONS",
+  "PAINT_GESTURE",
+  "POP_ART",
+  "RETRO_ETCHING",
+  "RIVIERA_POP",
+  "SPOTLIGHT_80S",
+  "STYLIZED_RED",
+  "SURREAL_COLLAGE",
+  "TRAVEL_POSTER",
+  "VINTAGE_GEO",
+  "VINTAGE_POSTER",
+] as const;
+
+export const ideogramResolutions = [
+  "512x1536",
+  "576x1408",
+  "576x1472",
+  "576x1536",
+  "640x1344",
+  "640x1408",
+  "640x1472",
+  "640x1536",
+  "704x1152",
+  "704x1216",
+  "704x1280",
+  "704x1344",
+  "704x1408",
+  "704x1472",
+  "736x1312",
+  "768x1088",
+  "768x1216",
+  "768x1280",
+  "768x1344",
+  "800x1280",
+  "832x960",
+  "832x1024",
+  "832x1088",
+  "832x1152",
+  "832x1216",
+  "832x1248",
+  "864x1152",
+  "896x960",
+  "896x1024",
+  "896x1088",
+  "896x1120",
+  "896x1152",
+  "960x832",
+  "960x896",
+  "960x1024",
+  "960x1088",
+  "960x1120",
+  "960x1152",
+  "1024x832",
+  "1024x896",
+  "1024x960",
+  "1024x1024",
+  "1088x768",
+  "1088x832",
+  "1088x896",
+  "1088x960",
+  "1120x896",
+  "1152x704",
+  "1152x832",
+  "1152x864",
+  "1152x896",
+  "1216x704",
+  "1216x768",
+  "1216x832",
+  "1248x832",
+  "1280x704",
+  "1280x768",
+  "1280x800",
+  "1312x736",
+  "1344x640",
+  "1344x704",
+  "1344x768",
+  "1408x576",
+  "1408x640",
+  "1408x704",
+  "1472x576",
+  "1472x640",
+  "1472x704",
+  "1536x512",
+  "1536x576",
+  "1536x640",
+] as const;
+
+export const ideogramAspectRatios = [
+  "1x3",
+  "3x1",
+  "1x2",
+  "2x1",
+  "9x16",
+  "16x9",
+  "10x16",
+  "16x10",
+  "2x3",
+  "3x2",
+  "3x4",
+  "4x3",
+  "4x5",
+  "5x4",
+  "1x1",
+] as const;
+
+export const ideogramRenderingSpeed = ["FLASH", "TURBO", "DEFAULT", "QUALITY"] as const;
+
+export const ideogramMagicPrompt = ["AUTO", "ON", "OFF"] as const;
+
+export const ideogramColorPalettePreset = [
+  "EMBER",
+  "FRESH",
+  "JUNGLE",
+  "MAGIC",
+  "MELON",
+  "MOSAIC",
+  "PASTEL",
+  "ULTRAMARINE",
+] as const;

package/src/modules/ai/ideogram/ideogram.dto.ts

@@ -0,0 +1,64 @@
+import { z } from "zod";
+import {
+  ideogramAspectRatios,
+  ideogramColorPalettePreset,
+  ideogramMagicPrompt,
+  ideogramRenderingSpeed,
+  ideogramResolutions,
+  ideogramStylePresets,
+  ideogramStyleTypes,
+} from "#modules/ai/ideogram/ideogram.constants";
+
+export const ideogramAISchema = z.object({
+  prompt: z.string().describe("Create a prompt for the image generation."),
+  style_type: z.enum(ideogramStyleTypes).describe("Choose a style type for the image generation."),
+  // style_preset: z
+  //   .enum(ideogramStylePresets)
+  //   .describe("Choose a style preset for the image generation."),
+});
+
+export const ideogramV3GenerateInputSchema = z.object({
+  prompt: z.string(),
+  seed: z.number().optional(),
+  resolution: z.enum(ideogramResolutions).optional(),
+  rendering_speed: z.enum(ideogramRenderingSpeed).optional(),
+  magic_prompt: z.enum(ideogramMagicPrompt).optional(),
+  negative_prompt: z.string().optional(),
+  num_images: z.number().optional(),
+  color_palette: z
+    .object({
+      ColorPaletteWithPresetName: z.object({ name: z.enum(ideogramColorPalettePreset) }).optional(),
+      ColorPaletteWithMembers: z
+        .object({
+          members: z.array(
+            z.object({ color_hex: z.string(), color_weight: z.number().min(0.05).max(1) })
+          ),
+        })
+        .optional(),
+    })
+    .optional(),
+  style_codes: z.array(z.string()).optional(),
+  aspect_ratio: z.enum(ideogramAspectRatios).optional(),
+  style_type: z.enum(ideogramStyleTypes).optional(),
+  style_preset: z.enum(ideogramStylePresets).optional(),
+  style_reference_images: z.array(z.file()).optional(),
+  character_reference_images: z.array(z.file()).optional(),
+  character_reference_images_mask: z.file().optional(),
+});
+
+export const ideogramV3GenerateOutputSchema = z.object({
+  created: z.string(),
+  data: z.array(
+    z.object({
+      prompt: z.string(),
+      resolution: z.string(),
+      is_image_safe: z.boolean(),
+      seed: z.number(),
+      url: z.url(),
+      style_type: z.enum(ideogramStyleTypes),
+    })
+  ),
+});
+
+export type IdeogramV3GenerateInput = z.infer<typeof ideogramV3GenerateInputSchema>;
+export type IdeogramV3GenerateOutput = z.infer<typeof ideogramV3GenerateOutputSchema>;