studio-lumiere-cli 0.1.4 → 0.1.5

package/src/pipelines/refineImage.ts

@@ -1,36 +0,0 @@
-import path from "node:path";
-import { GeminiClient } from "../clients/geminiClient.js";
-import { LumiereConfig, RefineRequest } from "../types.js";
-import { readImageAsPart, resolveOutputDir, saveBase64Image, writeJson } from "../storage/files.js";
-
-export const refineImage = async (config: LumiereConfig, request: RefineRequest): Promise<string> => {
-  const client = new GeminiClient(config);
-  const outputDir = resolveOutputDir(request.outputDir ?? config.outputDir, "refinements");
-
-  const sizeAdjustment = request.sizeAdjustment;
-  let prompt = `You are an expert high-end jewelry photo retoucher. Refine this image according to: "${request.instruction}". Preserve jewelry design and materials. Maintain photorealism.`;
-
-  if (sizeAdjustment !== undefined && sizeAdjustment !== 100) {
-    const direction = sizeAdjustment < 100 ? "smaller" : "larger";
-    prompt = `Adjust the size of all jewelry pieces to ${sizeAdjustment}% of current size (${direction}). Preserve all details, lighting, and composition. ${request.instruction}`;
-  }
-
-  const imagePart = await readImageAsPart(request.inputImage);
-  const images = await client.generateImage({
-    parts: [{ text: prompt }, imagePart],
-    imageSize: "2K"
-  });
-
-  const outputPath = path.join(outputDir, "refined.png");
-  await saveBase64Image(images[0], outputPath);
-
-  await writeJson(path.join(outputDir, "refine.json"), {
-    inputImage: request.inputImage,
-    instruction: request.instruction,
-    sizeAdjustment,
-    outputPath
-  });
-
-  return outputPath;
-};
-

package/src/pipelines/resolve.ts

@@ -1,88 +0,0 @@
-import { Option, OccasionTheme, Resolution, Template } from "../types.js";
-import { TEMPLATE_MAP } from "../config/templates.js";
-import {
-  BACKGROUNDS,
-  BACKGROUND_TYPES,
-  ETHNICITIES,
-  HAIR_COLORS,
-  OCCASIONS,
-  RESOLUTIONS,
-  SKIN_TONES,
-  VIBES
-} from "../config/options.js";
-
-const optionMap = (options: Option[]): Map<string, Option> => {
-  return new Map(options.map((opt) => [opt.id, opt]));
-};
-
-const ETHNICITY_MAP = optionMap(ETHNICITIES);
-const SKIN_TONE_MAP = optionMap(SKIN_TONES);
-const HAIR_COLOR_MAP = optionMap(HAIR_COLORS);
-const BACKGROUND_MAP = optionMap(BACKGROUNDS);
-const BACKGROUND_TYPE_MAP = optionMap(BACKGROUND_TYPES);
-const VIBE_MAP = optionMap(VIBES);
-const OCCASION_MAP = new Map(OCCASIONS.map((o) => [o.id, o]));
-const RESOLUTION_MAP = new Map(RESOLUTIONS.map((r) => [r.id, r]));
-
-export const getTemplate = (id: string): Template => {
-  const template = TEMPLATE_MAP.get(id);
-  if (!template) {
-    throw new Error(`Unknown template: ${id}`);
-  }
-  return template;
-};
-
-export const resolveTemplateOption = (template: Template, id?: string, which?: "primary" | "secondary" | "tertiary"): Option | undefined => {
-  if (!id) return undefined;
-  const options =
-    which === "secondary"
-      ? template.secondaryCustomizationOptions
-      : which === "tertiary"
-        ? template.tertiaryCustomizationOptions
-        : template.customizationOptions;
-
-  return options?.find((opt: Option) => opt.id === id) ?? { id, name: id, value: id };
-};
-
-export const resolveEthnicity = (id?: string): Option | undefined => {
-  if (!id) return undefined;
-  return ETHNICITY_MAP.get(id) ?? { id, name: id, value: id };
-};
-
-export const resolveSkinTone = (id?: string): Option | undefined => {
-  if (!id) return undefined;
-  return SKIN_TONE_MAP.get(id) ?? { id, name: id, value: id };
-};
-
-export const resolveHairColor = (id?: string): Option | undefined => {
-  if (!id) return undefined;
-  return HAIR_COLOR_MAP.get(id) ?? { id, name: id, value: id };
-};
-
-export const resolveBackground = (id?: string): Option | undefined => {
-  if (!id) return undefined;
-  return BACKGROUND_MAP.get(id) ?? { id, name: id, value: id };
-};
-
-export const resolveBackgroundType = (id?: string): Option | undefined => {
-  if (!id) return undefined;
-  return BACKGROUND_TYPE_MAP.get(id) ?? { id, name: id, value: id };
-};
-
-export const resolveVibe = (id?: string): Option | undefined => {
-  if (!id) return undefined;
-  return VIBE_MAP.get(id) ?? { id, name: id, value: id };
-};
-
-export const resolveOccasion = (id?: string): OccasionTheme | undefined => {
-  if (!id) return undefined;
-  return OCCASION_MAP.get(id);
-};
-
-export const resolveResolution = (id?: string): Resolution => {
-  const fallback = RESOLUTION_MAP.get("portrait") ?? Array.from(RESOLUTION_MAP.values())[0];
-  if (!fallback) throw new Error("No resolutions configured.");
-  if (!id) return fallback;
-  return RESOLUTION_MAP.get(id) ?? fallback;
-};
-

package/src/pipelines/upscaleImage.ts

@@ -1,30 +0,0 @@
-import path from "node:path";
-import { GeminiClient } from "../clients/geminiClient.js";
-import { LumiereConfig, UpscaleRequest } from "../types.js";
-import { readImageAsPart, resolveOutputDir, saveBase64Image, writeJson } from "../storage/files.js";
-
-export const upscaleImage = async (config: LumiereConfig, request: UpscaleRequest): Promise<string> => {
-  const client = new GeminiClient(config);
-  const outputDir = resolveOutputDir(request.outputDir ?? config.outputDir, "upscales");
-
-  const scale = request.scale ?? 2;
-  const prompt = `Upscale this image to ${scale}x resolution. Preserve all details, composition, and color. Do not change jewelry design, lighting, or background.`;
-
-  const imagePart = await readImageAsPart(request.inputImage);
-  const images = await client.generateImage({
-    parts: [{ text: prompt }, imagePart],
-    imageSize: scale >= 4 ? "4K" : "2K"
-  });
-
-  const outputPath = path.join(outputDir, "upscaled.png");
-  await saveBase64Image(images[0], outputPath);
-
-  await writeJson(path.join(outputDir, "upscale.json"), {
-    inputImage: request.inputImage,
-    scale,
-    outputPath
-  });
-
-  return outputPath;
-};
-

package/src/prompt/tiredGirlPrompt.ts

@@ -1,35 +0,0 @@
-import { Option } from "../types.js";
-import { TIRED_GIRL_STYLE_MAP, TIRED_GIRL_STYLES } from "../config/tiredGirl.js";
-
-export const resolveTiredGirlStyles = (styleIds?: string[], quantity: number = 1): Option[] => {
-  if (styleIds && styleIds.length > 0) {
-    const resolved = styleIds
-      .map((id) => TIRED_GIRL_STYLE_MAP.get(id))
-      .filter((style): style is Option => !!style);
-    if (resolved.length > 0) return resolved;
-  }
-
-  // Default cycling through predefined styles
-  const styles: Option[] = [];
-  for (let i = 0; i < quantity; i += 1) {
-    styles.push(TIRED_GIRL_STYLES[i % TIRED_GIRL_STYLES.length]);
-  }
-  return styles;
-};
-
-export const buildTiredGirlPrompt = (style: Option, hasMuse: boolean): string => {
-  return `You are a world-renowned fashion photographer.
-Your task is to write a prompt for an AI image generator to create a photorealistic portrait of a model in a "before" look.
-
-CRITICAL GUIDELINES:
-1. JEWELRY REMOVAL: The final image must contain NO jewelry whatsoever (no rings, earrings, necklaces, bracelets, watches).
-   - If jewelry appears in reference images, REMOVE it.
-2. IDENTITY CONSISTENCY: The model must match the reference images exactly (same face, features, skin tone).
-${hasMuse ? "3. The reference images are of the same model (Muse). Use them to preserve identity." : "3. Use the reference image to preserve identity."}
-4. STYLE: ${style.value}
-5. WARDROBE: casual, at-home, or relaxed styling (if not explicitly specified in style).
-6. LIGHTING: naturalistic, soft, real-world light.
-7. REALISM: no stylized effects, no glamour retouching.
-
-OUTPUT: A single, dense prompt describing the scene. Do not include any jewelry.`;
-};

package/src/storage/files.ts

@@ -1,44 +0,0 @@
-import fs from "node:fs/promises";
-import fsSync from "node:fs";
-import path from "node:path";
-
-export const ensureDir = async (dir: string): Promise<void> => {
-  await fs.mkdir(dir, { recursive: true });
-};
-
-export const readFileAsBase64 = async (filePath: string): Promise<{ data: string; mimeType: string }> => {
-  const buffer = await fs.readFile(filePath);
-  const ext = path.extname(filePath).toLowerCase();
-  const mimeType = ext === ".png" ? "image/png" : ext === ".webp" ? "image/webp" : "image/jpeg";
-  return { data: buffer.toString("base64"), mimeType };
-};
-
-export const readImageAsPart = async (filePath: string): Promise<{ inlineData: { mimeType: string; data: string } }> => {
-  const { data, mimeType } = await readFileAsBase64(filePath);
-  return { inlineData: { data, mimeType } };
-};
-
-export const saveBase64Image = async (base64DataUrl: string, outputPath: string): Promise<void> => {
-  const base64 = base64DataUrl.includes(",") ? base64DataUrl.split(",")[1] : base64DataUrl;
-  const buffer = Buffer.from(base64, "base64");
-  await ensureDir(path.dirname(outputPath));
-  await fs.writeFile(outputPath, buffer);
-};
-
-export const saveBinary = async (bytes: Uint8Array, outputPath: string): Promise<void> => {
-  await ensureDir(path.dirname(outputPath));
-  await fs.writeFile(outputPath, Buffer.from(bytes));
-};
-
-export const writeJson = async (outputPath: string, payload: unknown): Promise<void> => {
-  await ensureDir(path.dirname(outputPath));
-  await fs.writeFile(outputPath, JSON.stringify(payload, null, 2));
-};
-
-export const resolveOutputDir = (baseDir: string, subdir: string): string => {
-  const stamp = new Date().toISOString().replace(/[:.]/g, "-");
-  const outputDir = path.join(baseDir, subdir, stamp);
-  fsSync.mkdirSync(outputDir, { recursive: true });
-  return outputDir;
-};
-

package/src/storage/museStore.ts

@@ -1,31 +0,0 @@
-import fs from "node:fs/promises";
-import path from "node:path";
-import { MuseRecord } from "../types.js";
-
-const museIndexPath = (baseDir: string) => path.join(baseDir, "muses.json");
-
-export const loadMuses = async (baseDir: string): Promise<MuseRecord[]> => {
-  try {
-    const raw = await fs.readFile(museIndexPath(baseDir), "utf-8");
-    return JSON.parse(raw) as MuseRecord[];
-  } catch {
-    return [];
-  }
-};
-
-export const saveMuses = async (baseDir: string, muses: MuseRecord[]): Promise<void> => {
-  await fs.mkdir(baseDir, { recursive: true });
-  await fs.writeFile(museIndexPath(baseDir), JSON.stringify(muses, null, 2));
-};
-
-export const addMuse = async (baseDir: string, muse: MuseRecord): Promise<void> => {
-  const muses = await loadMuses(baseDir);
-  muses.push(muse);
-  await saveMuses(baseDir, muses);
-};
-
-export const getMuseById = async (baseDir: string, id: string): Promise<MuseRecord | undefined> => {
-  const muses = await loadMuses(baseDir);
-  return muses.find((m) => m.id === id);
-};
-

package/src/types.ts

@@ -1,198 +0,0 @@
-export type AspectRatio = "1:1" | "3:4" | "4:3" | "9:16" | "16:9";
-
-export interface Option {
-  id: string;
-  name: string;
-  value: string;
-  visualColor?: string;
-  visualImage?: string;
-}
-
-export interface Template {
-  id: string;
-  name: string;
-  description: string;
-  visualUrl?: string;
-  basePrompt: string;
-  customizationLabel?: string;
-  customizationOptions?: Option[];
-  secondaryCustomizationLabel?: string;
-  secondaryCustomizationOptions?: Option[];
-  tertiaryCustomizationLabel?: string;
-  tertiaryCustomizationOptions?: Option[];
-  isAdminOnly?: boolean;
-  museEnabled?: boolean;
-  videoEnabled?: boolean;
-}
-
-export interface Resolution {
-  id: string;
-  label: string;
-  aspectRatio: AspectRatio;
-  icon?: string;
-  description: string;
-  instagramUse: string;
-}
-
-export interface OccasionTheme {
-  id: string;
-  name: string;
-  icon: string;
-  description: string;
-  autoConfig: {
-    vibe: string;
-    background: string;
-    backgroundType: string;
-  };
-  promptDetails?: {
-    setting: string;
-    atmosphere: string;
-    colorPalette: string;
-    culturalElements: string;
-    styling: string;
-  };
-}
-
-export interface GenerationSelections {
-  templateId: string;
-  detailId?: string;
-  secondaryDetailId?: string;
-  tertiaryDetailId?: string;
-  ethnicityId?: string;
-  skinToneId?: string;
-  hairColorId?: string;
-  backgroundId?: string;
-  backgroundTypeId?: string;
-  vibeId?: string;
-  resolutionId?: string;
-  occasionId?: string;
-}
-
-export interface GenerationRequest {
-  inputImages: string[];
-  quantity?: number;
-  selections: GenerationSelections;
-  styleHint?: string;
-  museId?: string;
-  museImagePaths?: string[];
-  outputDir?: string;
-  enhancePrompt?: boolean;
-}
-
-export interface GenerationResult {
-  prompt: string;
-  enhancedPrompt?: string;
-  outputImages: string[];
-  logPath: string;
-}
-
-export interface RefineRequest {
-  inputImage: string;
-  instruction: string;
-  sizeAdjustment?: number;
-  outputDir?: string;
-}
-
-export interface UpscaleRequest {
-  inputImage: string;
-  scale?: number;
-  outputDir?: string;
-}
-
-export interface MuseRecord {
-  id: string;
-  name: string;
-  imagePaths: string[];
-  createdAt: string;
-}
-
-export interface CreateMuseRequest {
-  name: string;
-  sourceImage: string;
-  variations?: number;
-  outputDir?: string;
-}
-
-export interface CreateMuseResult {
-  muse: MuseRecord;
-  variationPaths: string[];
-  logPath: string;
-}
-
-export interface VideoRequest {
-  prompt: string;
-  imageInput?: string;
-  durationSeconds?: number;
-  aspectRatio?: AspectRatio;
-  outputDir?: string;
-}
-
-export interface VideoResult {
-  operationName?: string;
-  videoPath?: string;
-  logPath: string;
-}
-
-export interface TiredGirlRequest {
-  inputImage?: string;
-  museId?: string;
-  styleIds?: string[];
-  quantity?: number;
-  outputDir?: string;
-}
-
-export interface TiredGirlResult {
-  outputImages: string[];
-  logPath: string;
-}
-
-export type TextPosition =
-  | "top-center"
-  | "top-left"
-  | "top-right"
-  | "bottom-center"
-  | "bottom-center-high"
-  | "bottom-left"
-  | "bottom-right"
-  | "center";
-
-export interface TextOverlayOptions {
-  fontFamily?: string;
-  fontSize?: number;
-  fontWeight?: number;
-  color?: string;
-  strokeColor?: string;
-  strokeWidth?: number;
-  banner?: boolean;
-  bannerColor?: string;
-  bannerPadding?: number;
-  bannerRadius?: number;
-  position?: TextPosition;
-  padding?: number;
-  maxWidthRatio?: number;
-}
-
-export interface GridOptions {
-  columns: number;
-  rows: number;
-  padding?: number;
-  background?: string;
-  tileWidth?: number;
-  tileHeight?: number;
-}
-
-export interface LumiereConfig {
-  apiKey: string;
-  outputDir: string;
-  models?: {
-    prompt?: string;
-    image?: string;
-    video?: string;
-  };
-  retry?: {
-    maxRetries?: number;
-    baseDelayMs?: number;
-    maxDelayMs?: number;
-  };
-}
-

package/tsconfig.json

@@ -1,15 +0,0 @@
-{
-  "compilerOptions": {
-    "target": "ES2022",
-    "module": "NodeNext",
-    "moduleResolution": "NodeNext",
-    "outDir": "dist",
-    "rootDir": "src",
-    "declaration": true,
-    "sourceMap": true,
-    "strict": true,
-    "esModuleInterop": true,
-    "skipLibCheck": true
-  },
-  "include": ["src"]
-}