clawbr 0.0.41 → 0.0.42

package/dist/commands/generate.command.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/commands/generate.command.ts"],"sourcesContent":["import { Command, CommandRunner, Option } from \"nest-commander\";\nimport { writeFileSync } from \"fs\";\nimport ora from \"ora\";\nimport fetch from \"node-fetch\";\nimport { resolve } from \"path\";\nimport { generateImage } from \"ai\";\nimport { createOpenAI } from \"@ai-sdk/openai\";\nimport { createGoogleGenerativeAI } from \"@ai-sdk/google\";\nimport { loadCredentials } from \"../utils/credentials.js\";\nimport { resolveImageToDataUri, validateImageInput } from \"../utils/image.js\";\nimport { requireOnboarding } from \"../utils/config.js\";\nimport {\n  getProviderModels,\n  getModelById,\n  isValidModel,\n  getPrimaryModel,\n  getFallbackModels,\n  supportsReferenceImage,\n  formatModelList,\n} from \"../config/image-models.js\";\n\ninterface GenerateCommandOptions {\n  prompt?: string;\n  output?: string;\n  size?: string;\n  sourceImage?: string;\n  model?: string;\n  json?: boolean;\n}\n\n@Command({\n  name: \"generate\",\n  description: \"Generate an image using AI with smart model fallback\",\n  arguments: \"\",\n  options: { isDefault: false },\n})\nexport class GenerateCommand extends CommandRunner {\n  async run(inputs: string[], options: GenerateCommandOptions): Promise<void> {\n    await requireOnboarding();\n    const { prompt, output, size = \"1024x1024\", sourceImage, model, json = false } = options;\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Validation\n    // ─────────────────────────────────────────────────────────────────────\n    if (!prompt) {\n      throw new Error(\n        '--prompt is required. Example: clawbr generate --prompt \"a robot building software\" --output \"./robot.png\"'\n      );\n    }\n\n    if (!output) {\n      throw new Error(\n        '--output is required. Example: clawbr generate --prompt \"...\" --output \"./image.png\"'\n      );\n    }\n\n    // Validate source image if provided\n    if (sourceImage) {\n      const validation = validateImageInput(sourceImage);\n      if (!validation.valid) {\n        throw new Error(validation.error);\n      }\n    }\n\n    // Validate size\n    const validSizes = [\"256x256\", \"512x512\", \"1024x1024\", \"1792x1024\", \"1024x1792\"];\n    if (!validSizes.includes(size)) {\n      throw new Error(`Invalid size. Must be one of: ${validSizes.join(\", \")}`);\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Load Credentials\n    // ─────────────────────────────────────────────────────────────────────\n    const credentials = loadCredentials();\n\n    if (!credentials) {\n      throw new Error(\"Credentials not found. Run 'clawbr onboard' first to set up your account.\");\n    }\n\n    const { aiProvider, apiKeys } = credentials;\n    const apiKey = apiKeys[aiProvider as keyof typeof apiKeys];\n\n    if (!apiKey) {\n      throw new Error(\n        `No API key found for provider '${aiProvider}'. Run 'clawbr onboard' to configure.`\n      );\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Validate model if provided\n    // ─────────────────────────────────────────────────────────────────────\n    if (model && !isValidModel(aiProvider, model)) {\n      const availableModels = formatModelList(aiProvider);\n      throw new Error(\n        `Invalid model '${model}' for provider '${aiProvider}'.\\n\\nAvailable models:\\n${availableModels}`\n      );\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Check reference image support\n    // ─────────────────────────────────────────────────────────────────────\n    if (sourceImage && model && !supportsReferenceImage(aiProvider, model)) {\n      const modelInfo = getModelById(aiProvider, model);\n      throw new Error(\n        `Model '${modelInfo?.name || model}' does not support reference images.\\n\\n` +\n          `For reference image support with ${aiProvider}, use one of:\\n` +\n          getProviderModels(aiProvider)\n            .filter((m) => m.supportsReferenceImage)\n            .map((m) => `  • ${m.id}`)\n            .join(\"\\n\")\n      );\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Prepare source image if provided\n    // ─────────────────────────────────────────────────────────────────────\n    const sourceImageData = sourceImage ? await resolveImageToDataUri(sourceImage) : undefined;\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Generate Image with Smart Fallback\n    // ─────────────────────────────────────────────────────────────────────\n    const spinner = json\n      ? null\n      : ora(sourceImageData ? \"Generating image from source...\" : \"Generating image...\").start();\n\n    try {\n      let imageBuffer: Buffer;\n\n      // Determine models to use\n      const primaryModel = model || getPrimaryModel(aiProvider);\n      const fallbackModels = model ? [] : getFallbackModels(aiProvider);\n\n      if (aiProvider === \"openrouter\") {\n        imageBuffer = await this.generateWithFallback(\n          prompt,\n          size,\n          apiKey,\n          \"openrouter\",\n          { primary: primaryModel, fallbacks: fallbackModels },\n          spinner,\n          sourceImageData\n        );\n      } else if (aiProvider === \"openai\") {\n        if (sourceImageData) {\n          throw new Error(\n            \"OpenAI does not support image-to-image generation. Use OpenRouter with a model that supports reference images.\"\n          );\n        }\n        imageBuffer = await this.generateWithFallback(\n          prompt,\n          size,\n          apiKey,\n          \"openai\",\n          { primary: primaryModel, fallbacks: fallbackModels },\n          spinner\n        );\n      } else if (aiProvider === \"google\") {\n        if (sourceImageData) {\n          throw new Error(\n            \"Google Imagen does not support image-to-image generation. Use OpenRouter with a model that supports reference images.\"\n          );\n        }\n        imageBuffer = await this.generateWithFallback(\n          prompt,\n          size,\n          apiKey,\n          \"google\",\n          { primary: primaryModel, fallbacks: fallbackModels },\n          spinner\n        );\n      } else {\n        if (spinner) spinner.fail();\n        throw new Error(`Unsupported AI provider: ${aiProvider}`);\n      }\n\n      // ─────────────────────────────────────────────────────────────────────\n      // Save Image\n      // ─────────────────────────────────────────────────────────────────────\n      const outputPath = resolve(output);\n      writeFileSync(outputPath, imageBuffer);\n\n      if (spinner) {\n        spinner.succeed(`Image generated and saved to: ${outputPath}`);\n      }\n\n      // ─────────────────────────────────────────────────────────────────────\n      // Output\n      // ─────────────────────────────────────────────────────────────────────\n      if (json) {\n        console.log(\n          JSON.stringify(\n            {\n              success: true,\n              prompt,\n              output: outputPath,\n              size,\n              provider: aiProvider,\n            },\n            null,\n            2\n          )\n        );\n      } else {\n        console.log(\"\\n🎨 Image Generation Complete!\");\n        console.log(\"─────────────────────────────────────\");\n        console.log(`Prompt: ${prompt}`);\n        console.log(`Size: ${size}`);\n        if (sourceImageData) {\n          console.log(`Source Image: ${sourceImage}`);\n        }\n        console.log(`Output: ${outputPath}`);\n        console.log(`Provider: ${aiProvider}`);\n        if (model) {\n          console.log(`Model: ${model}`);\n        }\n        console.log(\"─────────────────────────────────────\\n\");\n      }\n    } catch (error) {\n      if (spinner && spinner.isSpinning) {\n        spinner.fail(\"Image generation failed\");\n      }\n      throw error;\n    }\n  }\n\n  /**\n   * Generate image with smart fallback chain\n   * Tries primary model first, then falls back to alternatives if it fails\n   */\n  private async generateWithFallback(\n    prompt: string,\n    size: string,\n    apiKey: string,\n    provider: \"openrouter\" | \"openai\" | \"google\",\n    config: { primary: string | null; fallbacks: string[] },\n    spinner: {\n      text: string;\n      info: (msg: string) => void;\n      warn: (msg: string) => void;\n      isSpinning?: boolean;\n    } | null,\n    sourceImage?: string\n  ): Promise<Buffer> {\n    const modelsToTry = [config.primary, ...config.fallbacks].filter(\n      (model): model is string => model !== null\n    );\n\n    let lastError: Error | null = null;\n\n    for (let i = 0; i < modelsToTry.length; i++) {\n      const model = modelsToTry[i];\n\n      try {\n        if (spinner) {\n          const modelName = model.split(\"/\").pop() || model;\n          spinner.text = `Generating image with ${modelName}... (attempt ${i + 1}/${modelsToTry.length})`;\n        }\n\n        const imageBuffer = await this.generateWithModel(\n          prompt,\n          size,\n          apiKey,\n          provider,\n          model,\n          sourceImage\n        );\n\n        if (spinner && i > 0) {\n          // Only show fallback message if we had to fall back\n          spinner.info(`Successfully generated with fallback model: ${model}`);\n        }\n\n        return imageBuffer;\n      } catch (error) {\n        lastError = error as Error;\n\n        // If this wasn't the last model, log the failure and try the next one\n        if (i < modelsToTry.length - 1) {\n          if (spinner) {\n            spinner.warn(`Model ${model} failed, trying fallback...`);\n          } else {\n            console.warn(`Model ${model} failed: ${lastError.message}`);\n          }\n          continue;\n        }\n      }\n    }\n\n    // If we get here, all models failed\n    throw new Error(\n      `All models failed to generate image. Last error: ${lastError?.message || \"Unknown error\"}`\n    );\n  }\n\n  /**\n   * get the model configuration for the AI SDK\n   */\n  private getImageModel(provider: string, apiKey: string, model: string) {\n    if (provider === \"openai\") {\n      const openai = createOpenAI({ apiKey });\n      return openai.image(model);\n    } else if (provider === \"google\") {\n      const google = createGoogleGenerativeAI({ apiKey });\n      return google.image(model);\n    }\n    throw new Error(`Provider ${provider} not supported via AI SDK`);\n  }\n\n  /**\n   * Generate image using a specific model\n   */\n  private async generateWithModel(\n    prompt: string,\n    size: string,\n    apiKey: string,\n    provider: \"openrouter\" | \"openai\" | \"google\",\n    model: string,\n    sourceImage?: string\n  ): Promise<Buffer> {\n    // ─────────────────────────────────────────────────────────────────────\n    // OPENROUTER (Via Fetch / Chat Completions)\n    // ─────────────────────────────────────────────────────────────────────\n    if (provider === \"openrouter\") {\n      // Parse aspect ratio from size\n      const [width, height] = size.split(\"x\").map(Number);\n      let aspectRatio = \"1:1\";\n      if (width && height) {\n        const gcd = (a: number, b: number): number => (b === 0 ? a : gcd(b, a % b));\n        const divisor = gcd(width, height);\n        aspectRatio = `${width / divisor}:${height / divisor}`;\n      }\n\n      // Build content array based on whether we have a source image\n      let content: any;\n      if (sourceImage) {\n        // Image-to-image generation: include source image in content\n        content = [\n          {\n            type: \"text\",\n            text: prompt,\n          },\n          {\n            type: \"image_url\",\n            image_url: {\n              url: sourceImage,\n            },\n          },\n        ];\n      } else {\n        // Text-to-image generation: just the prompt\n        content = prompt;\n      }\n\n      const response = await fetch(\"https://openrouter.ai/api/v1/chat/completions\", {\n        method: \"POST\",\n        headers: {\n          Authorization: `Bearer ${apiKey}`,\n          \"Content-Type\": \"application/json\",\n          \"HTTP-Referer\": \"https://clawbr.bricks-studio.ai\",\n          \"X-Title\": \"clawbr CLI\",\n        },\n        body: JSON.stringify({\n          model: model,\n          messages: [\n            {\n              role: \"user\",\n              content: content,\n            },\n          ],\n          // Specific to Gemini/OpenRouter multimodal\n          modalities: [\"image\", \"text\"],\n          image_config: {\n            aspect_ratio: aspectRatio,\n            // image_size: \"4K\", // Optional based on snippet, but maybe risky for all models\n          },\n        }),\n      });\n\n      if (!response.ok) {\n        const text = await response.text();\n        throw new Error(`OpenRouter API error: ${text}`);\n      }\n\n      const result = (await response.json()) as any;\n\n      if (result.choices?.[0]?.message?.images?.[0]?.image_url?.url) {\n        const imageUrl = result.choices[0].message.images[0].image_url.url;\n\n        // If it's a URL, fetch it\n        if (imageUrl.startsWith(\"http\")) {\n          const imgRes = await fetch(imageUrl);\n          const arrayBuffer = await imgRes.arrayBuffer();\n          return Buffer.from(arrayBuffer);\n        }\n\n        // If it's base64 data URI\n        if (imageUrl.startsWith(\"data:image\")) {\n          const base64Data = imageUrl.split(\",\")[1];\n          return Buffer.from(base64Data, \"base64\");\n        }\n\n        throw new Error(\"Unknown image URL format\");\n      }\n\n      throw new Error(\"No image generated from OpenRouter response\");\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // OPENAI / GOOGLE (Via AI SDK)\n    // ─────────────────────────────────────────────────────────────────────\n    const imageModel = this.getImageModel(provider, apiKey, model);\n\n    // Pass size as string directly as per SDK requirements.\n    // We cast to 'any' to avoid strict template literal validation errors\n    // since we know validSizes allows specifically \"1024x1024\" etc.\n    const { image } = await generateImage({\n      model: imageModel,\n      prompt,\n      n: 1,\n      size: size as any,\n    });\n\n    // The image object from 'ai' SDK contains the base64 string\n    return Buffer.from(image.base64, \"base64\");\n  }\n\n  @Option({\n    flags: \"-p, --prompt <text>\",\n    description: \"Text description of the image to generate\",\n  })\n  parsePrompt(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"-o, --output <path>\",\n    description: \"Path where the generated image will be saved\",\n  })\n  parseOutput(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"-s, --size <size>\",\n    description: \"Image size (256x256, 512x512, 1024x1024, 1792x1024, 1024x1792)\",\n  })\n  parseSize(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"--source-image <path>\",\n    description: \"Path to source image or URL (for image-to-image generation, OpenRouter only)\",\n  })\n  parseSourceImage(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"-m, --model <modelId>\",\n    description:\n      \"Specific model to use (provider-dependent). Use model ID from your provider's list. Note: Not all models support reference images (--source-image).\",\n  })\n  parseModel(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"--json\",\n    description: \"Output result in JSON format\",\n  })\n  parseJson(): boolean {\n    return true;\n  }\n}\n"],"names":["Command","CommandRunner","Option","writeFileSync","ora","fetch","resolve","generateImage","createOpenAI","createGoogleGenerativeAI","loadCredentials","resolveImageToDataUri","validateImageInput","requireOnboarding","getProviderModels","getModelById","isValidModel","getPrimaryModel","getFallbackModels","supportsReferenceImage","formatModelList","GenerateCommand","run","inputs","options","prompt","output","size","sourceImage","model","json","Error","validation","valid","error","validSizes","includes","join","credentials","aiProvider","apiKeys","apiKey","availableModels","modelInfo","name","filter","m","map","id","sourceImageData","undefined","spinner","start","imageBuffer","primaryModel","fallbackModels","generateWithFallback","primary","fallbacks","fail","outputPath","succeed","console","log","JSON","stringify","success","provider","isSpinning","config","modelsToTry","lastError","i","length","modelName","split","pop","text","generateWithModel","info","warn","message","getImageModel","openai","image","google","width","height","Number","aspectRatio","gcd","a","b","divisor","content","type","image_url","url","response","method","headers","Authorization","body","messages","role","modalities","image_config","aspect_ratio","ok","result","choices","images","imageUrl","startsWith","imgRes","arrayBuffer","Buffer","from","base64Data","imageModel","n","base64","parsePrompt","val","parseOutput","parseSize","parseSourceImage","parseModel","parseJson","flags","description","arguments","isDefault"],"mappings":";;;;;;;;;AAAA,SAASA,OAAO,EAAEC,aAAa,EAAEC,MAAM,QAAQ,iBAAiB;AAChE,SAASC,aAAa,QAAQ,KAAK;AACnC,OAAOC,SAAS,MAAM;AACtB,OAAOC,WAAW,aAAa;AAC/B,SAASC,OAAO,QAAQ,OAAO;AAC/B,SAASC,aAAa,QAAQ,KAAK;AACnC,SAASC,YAAY,QAAQ,iBAAiB;AAC9C,SAASC,wBAAwB,QAAQ,iBAAiB;AAC1D,SAASC,eAAe,QAAQ,0BAA0B;AAC1D,SAASC,qBAAqB,EAAEC,kBAAkB,QAAQ,oBAAoB;AAC9E,SAASC,iBAAiB,QAAQ,qBAAqB;AACvD,SACEC,iBAAiB,EACjBC,YAAY,EACZC,YAAY,EACZC,eAAe,EACfC,iBAAiB,EACjBC,sBAAsB,EACtBC,eAAe,QACV,4BAA4B;AAiBnC,OAAO,MAAMC,wBAAwBpB;IACnC,MAAMqB,IAAIC,MAAgB,EAAEC,OAA+B,EAAiB;QAC1E,MAAMX;QACN,MAAM,EAAEY,MAAM,EAAEC,MAAM,EAAEC,OAAO,WAAW,EAAEC,WAAW,EAAEC,KAAK,EAAEC,OAAO,KAAK,EAAE,GAAGN;QAEjF,wEAAwE;QACxE,aAAa;QACb,wEAAwE;QACxE,IAAI,CAACC,QAAQ;YACX,MAAM,IAAIM,MACR;QAEJ;QAEA,IAAI,CAACL,QAAQ;YACX,MAAM,IAAIK,MACR;QAEJ;QAEA,oCAAoC;QACpC,IAAIH,aAAa;YACf,MAAMI,aAAapB,mBAAmBgB;YACtC,IAAI,CAACI,WAAWC,KAAK,EAAE;gBACrB,MAAM,IAAIF,MAAMC,WAAWE,KAAK;YAClC;QACF;QAEA,gBAAgB;QAChB,MAAMC,aAAa;YAAC;YAAW;YAAW;YAAa;YAAa;SAAY;QAChF,IAAI,CAACA,WAAWC,QAAQ,CAACT,OAAO;YAC9B,MAAM,IAAII,MAAM,CAAC,8BAA8B,EAAEI,WAAWE,IAAI,CAAC,OAAO;QAC1E;QAEA,wEAAwE;QACxE,mBAAmB;QACnB,wEAAwE;QACxE,MAAMC,cAAc5B;QAEpB,IAAI,CAAC4B,aAAa;YAChB,MAAM,IAAIP,MAAM;QAClB;QAEA,MAAM,EAAEQ,UAAU,EAAEC,OAAO,EAAE,GAAGF;QAChC,MAAMG,SAASD,OAAO,CAACD,WAAmC;QAE1D,IAAI,CAACE,QAAQ;YACX,MAAM,IAAIV,MACR,CAAC,+BAA+B,EAAEQ,WAAW,qCAAqC,CAAC;QAEvF;QAEA,wEAAwE;QACxE,6BAA6B;QAC7B,wEAAwE;QACxE,IAAIV,SAAS,CAACb,aAAauB,YAAYV,QAAQ;YAC7C,MAAMa,kBAAkBtB,gBAAgBmB;YACxC,MAAM,IAAIR,MACR,CAAC,eAAe,EAAEF,MAAM,gBAAgB,EAAEU,WAAW,yBAAyB,EAAEG,iBAAiB;QAErG;QAEA,wEAAwE;QACxE,gCAAgC;QAChC,wEAAwE;QACxE,IAAId,eAAeC,SAAS,CAACV,uBAAuBoB,YAAYV,QAAQ;YACtE,MAAMc,YAAY5B,aAAawB,YAAYV;YAC3C,MAAM,IAAIE,MACR,CAAC,OAAO,EAAEY,WAAWC,QAAQf,MAAM,wCAAwC,CAAC,GAC1E,CAAC,iCAAiC,EAAEU,WAAW,eAAe,CAAC,GAC/DzB,kBAAkByB,YACfM,MAAM,CAAC,CAACC,IAAMA,EAAE3B,sBAAsB,EACtC4B,GAAG,CAAC,CAACD,IAAM,CAAC,IAAI,EAAEA,EAAEE,EAAE,EAAE,EACxBX,IAAI,CAAC;QAEd;QAEA,wEAAwE;QACxE,mCAAmC;QACnC,wEAAwE;QACxE,MAAMY,kBAAkBrB,cAAc,MAAMjB,sBAAsBiB,eAAesB;QAEjF,wEAAwE;QACxE,qCAAqC;QACrC,wEAAwE;QACxE,MAAMC,UAAUrB,OACZ,OACA1B,IAAI6C,kBAAkB,oCAAoC,uBAAuBG,KAAK;QAE1F,IAAI;YACF,IAAIC;YAEJ,0BAA0B;YAC1B,MAAMC,eAAezB,SAASZ,gBAAgBsB;YAC9C,MAAMgB,iBAAiB1B,QAAQ,EAAE,GAAGX,kBAAkBqB;YAEtD,IAAIA,eAAe,cAAc;gBAC/Bc,cAAc,MAAM,IAAI,CAACG,oBAAoB,CAC3C/B,QACAE,MACAc,QACA,cACA;oBAAEgB,SAASH;oBAAcI,WAAWH;gBAAe,GACnDJ,SACAF;YAEJ,OAAO,IAAIV,eAAe,UAAU;gBAClC,IAAIU,iBAAiB;oBACnB,MAAM,IAAIlB,MACR;gBAEJ;gBACAsB,cAAc,MAAM,IAAI,CAACG,oBAAoB,CAC3C/B,QACAE,MACAc,QACA,UACA;oBAAEgB,SAASH;oBAAcI,WAAWH;gBAAe,GACnDJ;YAEJ,OAAO,IAAIZ,eAAe,UAAU;gBAClC,IAAIU,iBAAiB;oBACnB,MAAM,IAAIlB,MACR;gBAEJ;gBACAsB,cAAc,MAAM,IAAI,CAACG,oBAAoB,CAC3C/B,QACAE,MACAc,QACA,UACA;oBAAEgB,SAASH;oBAAcI,WAAWH;gBAAe,GACnDJ;YAEJ,OAAO;gBACL,IAAIA,SAASA,QAAQQ,IAAI;gBACzB,MAAM,IAAI5B,MAAM,CAAC,yBAAyB,EAAEQ,YAAY;YAC1D;YAEA,wEAAwE;YACxE,aAAa;YACb,wEAAwE;YACxE,MAAMqB,aAAatD,QAAQoB;YAC3BvB,cAAcyD,YAAYP;YAE1B,IAAIF,SAAS;gBACXA,QAAQU,OAAO,CAAC,CAAC,8BAA8B,EAAED,YAAY;YAC/D;YAEA,wEAAwE;YACxE,SAAS;YACT,wEAAwE;YACxE,IAAI9B,MAAM;gBACRgC,QAAQC,GAAG,CACTC,KAAKC,SAAS,CACZ;oBACEC,SAAS;oBACTzC;oBACAC,QAAQkC;oBACRjC;oBACAwC,UAAU5B;gBACZ,GACA,MACA;YAGN,OAAO;gBACLuB,QAAQC,GAAG,CAAC;gBACZD,QAAQC,GAAG,CAAC;gBACZD,QAAQC,GAAG,CAAC,CAAC,QAAQ,EAAEtC,QAAQ;gBAC/BqC,QAAQC,GAAG,CAAC,CAAC,MAAM,EAAEpC,MAAM;gBAC3B,IAAIsB,iBAAiB;oBACnBa,QAAQC,GAAG,CAAC,CAAC,cAAc,EAAEnC,aAAa;gBAC5C;gBACAkC,QAAQC,GAAG,CAAC,CAAC,QAAQ,EAAEH,YAAY;gBACnCE,QAAQC,GAAG,CAAC,CAAC,UAAU,EAAExB,YAAY;gBACrC,IAAIV,OAAO;oBACTiC,QAAQC,GAAG,CAAC,CAAC,OAAO,EAAElC,OAAO;gBAC/B;gBACAiC,QAAQC,GAAG,CAAC;YACd;QACF,EAAE,OAAO7B,OAAO;YACd,IAAIiB,WAAWA,QAAQiB,UAAU,EAAE;gBACjCjB,QAAQQ,IAAI,CAAC;YACf;YACA,MAAMzB;QACR;IACF;IAEA;;;GAGC,GACD,MAAcsB,qBACZ/B,MAAc,EACdE,IAAY,EACZc,MAAc,EACd0B,QAA4C,EAC5CE,MAAuD,EACvDlB,OAKQ,EACRvB,WAAoB,EACH;QACjB,MAAM0C,cAAc;YAACD,OAAOZ,OAAO;eAAKY,OAAOX,SAAS;SAAC,CAACb,MAAM,CAC9D,CAAChB,QAA2BA,UAAU;QAGxC,IAAI0C,YAA0B;QAE9B,IAAK,IAAIC,IAAI,GAAGA,IAAIF,YAAYG,MAAM,EAAED,IAAK;YAC3C,MAAM3C,QAAQyC,WAAW,CAACE,EAAE;YAE5B,IAAI;gBACF,IAAIrB,SAAS;oBACX,MAAMuB,YAAY7C,MAAM8C,KAAK,CAAC,KAAKC,GAAG,MAAM/C;oBAC5CsB,QAAQ0B,IAAI,GAAG,CAAC,sBAAsB,EAAEH,UAAU,aAAa,EAAEF,IAAI,EAAE,CAAC,EAAEF,YAAYG,MAAM,CAAC,CAAC,CAAC;gBACjG;gBAEA,MAAMpB,cAAc,MAAM,IAAI,CAACyB,iBAAiB,CAC9CrD,QACAE,MACAc,QACA0B,UACAtC,OACAD;gBAGF,IAAIuB,WAAWqB,IAAI,GAAG;oBACpB,oDAAoD;oBACpDrB,QAAQ4B,IAAI,CAAC,CAAC,4CAA4C,EAAElD,OAAO;gBACrE;gBAEA,OAAOwB;YACT,EAAE,OAAOnB,OAAO;gBACdqC,YAAYrC;gBAEZ,sEAAsE;gBACtE,IAAIsC,IAAIF,YAAYG,MAAM,GAAG,GAAG;oBAC9B,IAAItB,SAAS;wBACXA,QAAQ6B,IAAI,CAAC,CAAC,MAAM,EAAEnD,MAAM,2BAA2B,CAAC;oBAC1D,OAAO;wBACLiC,QAAQkB,IAAI,CAAC,CAAC,MAAM,EAAEnD,MAAM,SAAS,EAAE0C,UAAUU,OAAO,EAAE;oBAC5D;oBACA;gBACF;YACF;QACF;QAEA,oCAAoC;QACpC,MAAM,IAAIlD,MACR,CAAC,iDAAiD,EAAEwC,WAAWU,WAAW,iBAAiB;IAE/F;IAEA;;GAEC,GACD,AAAQC,cAAcf,QAAgB,EAAE1B,MAAc,EAAEZ,KAAa,EAAE;QACrE,IAAIsC,aAAa,UAAU;YACzB,MAAMgB,SAAS3E,aAAa;gBAAEiC;YAAO;YACrC,OAAO0C,OAAOC,KAAK,CAACvD;QACtB,OAAO,IAAIsC,aAAa,UAAU;YAChC,MAAMkB,SAAS5E,yBAAyB;gBAAEgC;YAAO;YACjD,OAAO4C,OAAOD,KAAK,CAACvD;QACtB;QACA,MAAM,IAAIE,MAAM,CAAC,SAAS,EAAEoC,SAAS,yBAAyB,CAAC;IACjE;IAEA;;GAEC,GACD,MAAcW,kBACZrD,MAAc,EACdE,IAAY,EACZc,MAAc,EACd0B,QAA4C,EAC5CtC,KAAa,EACbD,WAAoB,EACH;QACjB,wEAAwE;QACxE,4CAA4C;QAC5C,wEAAwE;QACxE,IAAIuC,aAAa,cAAc;YAC7B,+BAA+B;YAC/B,MAAM,CAACmB,OAAOC,OAAO,GAAG5D,KAAKgD,KAAK,CAAC,KAAK5B,GAAG,CAACyC;YAC5C,IAAIC,cAAc;YAClB,IAAIH,SAASC,QAAQ;gBACnB,MAAMG,MAAM,CAACC,GAAWC,IAAuBA,MAAM,IAAID,IAAID,IAAIE,GAAGD,IAAIC;gBACxE,MAAMC,UAAUH,IAAIJ,OAAOC;gBAC3BE,cAAc,GAAGH,QAAQO,QAAQ,CAAC,EAAEN,SAASM,SAAS;YACxD;YAEA,8DAA8D;YAC9D,IAAIC;YACJ,IAAIlE,aAAa;gBACf,6DAA6D;gBAC7DkE,UAAU;oBACR;wBACEC,MAAM;wBACNlB,MAAMpD;oBACR;oBACA;wBACEsE,MAAM;wBACNC,WAAW;4BACTC,KAAKrE;wBACP;oBACF;iBACD;YACH,OAAO;gBACL,4CAA4C;gBAC5CkE,UAAUrE;YACZ;YAEA,MAAMyE,WAAW,MAAM7F,MAAM,iDAAiD;gBAC5E8F,QAAQ;gBACRC,SAAS;oBACPC,eAAe,CAAC,OAAO,EAAE5D,QAAQ;oBACjC,gBAAgB;oBAChB,gBAAgB;oBAChB,WAAW;gBACb;gBACA6D,MAAMtC,KAAKC,SAAS,CAAC;oBACnBpC,OAAOA;oBACP0E,UAAU;wBACR;4BACEC,MAAM;4BACNV,SAASA;wBACX;qBACD;oBACD,2CAA2C;oBAC3CW,YAAY;wBAAC;wBAAS;qBAAO;oBAC7BC,cAAc;wBACZC,cAAclB;oBAEhB;gBACF;YACF;YAEA,IAAI,CAACS,SAASU,EAAE,EAAE;gBAChB,MAAM/B,OAAO,MAAMqB,SAASrB,IAAI;gBAChC,MAAM,IAAI9C,MAAM,CAAC,sBAAsB,EAAE8C,MAAM;YACjD;YAEA,MAAMgC,SAAU,MAAMX,SAASpE,IAAI;YAEnC,IAAI+E,OAAOC,OAAO,EAAE,CAAC,EAAE,EAAE7B,SAAS8B,QAAQ,CAAC,EAAE,EAAEf,WAAWC,KAAK;gBAC7D,MAAMe,WAAWH,OAAOC,OAAO,CAAC,EAAE,CAAC7B,OAAO,CAAC8B,MAAM,CAAC,EAAE,CAACf,SAAS,CAACC,GAAG;gBAElE,0BAA0B;gBAC1B,IAAIe,SAASC,UAAU,CAAC,SAAS;oBAC/B,MAAMC,SAAS,MAAM7G,MAAM2G;oBAC3B,MAAMG,cAAc,MAAMD,OAAOC,WAAW;oBAC5C,OAAOC,OAAOC,IAAI,CAACF;gBACrB;gBAEA,0BAA0B;gBAC1B,IAAIH,SAASC,UAAU,CAAC,eAAe;oBACrC,MAAMK,aAAaN,SAASrC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACzC,OAAOyC,OAAOC,IAAI,CAACC,YAAY;gBACjC;gBAEA,MAAM,IAAIvF,MAAM;YAClB;YAEA,MAAM,IAAIA,MAAM;QAClB;QAEA,wEAAwE;QACxE,+BAA+B;QAC/B,wEAAwE;QACxE,MAAMwF,aAAa,IAAI,CAACrC,aAAa,CAACf,UAAU1B,QAAQZ;QAExD,wDAAwD;QACxD,sEAAsE;QACtE,gEAAgE;QAChE,MAAM,EAAEuD,KAAK,EAAE,GAAG,MAAM7E,cAAc;YACpCsB,OAAO0F;YACP9F;YACA+F,GAAG;YACH7F,MAAMA;QACR;QAEA,4DAA4D;QAC5D,OAAOyF,OAAOC,IAAI,CAACjC,MAAMqC,MAAM,EAAE;IACnC;IAMAC,YAAYC,GAAW,EAAU;QAC/B,OAAOA;IACT;IAMAC,YAAYD,GAAW,EAAU;QAC/B,OAAOA;IACT;IAMAE,UAAUF,GAAW,EAAU;QAC7B,OAAOA;IACT;IAMAG,iBAAiBH,GAAW,EAAU;QACpC,OAAOA;IACT;IAOAI,WAAWJ,GAAW,EAAU;QAC9B,OAAOA;IACT;IAMAK,YAAqB;QACnB,OAAO;IACT;AACF;;;QA/CIC,OAAO;QACPC,aAAa;;;;;;;;;;QAObD,OAAO;QACPC,aAAa;;;;;;;;;;QAObD,OAAO;QACPC,aAAa;;;;;;;;;;QAObD,OAAO;QACPC,aAAa;;;;;;;;;;QAObD,OAAO;QACPC,aACE;;;;;;;;;;QAOFD,OAAO;QACPC,aAAa;;;;;;;;QAtbftF,MAAM;QACNsF,aAAa;QACbC,WAAW;QACX3G,SAAS;YAAE4G,WAAW;QAAM"}
+{"version":3,"sources":["../../src/commands/generate.command.ts"],"sourcesContent":["import { Command, CommandRunner, Option } from \"nest-commander\";\nimport { writeFileSync } from \"fs\";\nimport ora from \"ora\";\nimport fetch from \"node-fetch\";\nimport { resolve } from \"path\";\nimport { generateImage } from \"ai\";\nimport { createOpenAI } from \"@ai-sdk/openai\";\nimport { createGoogleGenerativeAI } from \"@ai-sdk/google\";\nimport { loadCredentials } from \"../utils/credentials.js\";\nimport { resolveImageToDataUri, validateImageInput } from \"../utils/image.js\";\nimport { requireOnboarding } from \"../utils/config.js\";\nimport {\n  getProviderModels,\n  getModelById,\n  isValidModel,\n  getPrimaryModel,\n  getFallbackModels,\n  supportsReferenceImage,\n  formatModelList,\n} from \"../config/image-models.js\";\n\ninterface GenerateCommandOptions {\n  prompt?: string;\n  output?: string;\n  size?: string;\n  sourceImage?: string;\n  model?: string;\n  aspectRatio?: string;\n  imageSize?: string;\n  json?: boolean;\n}\n\n@Command({\n  name: \"generate\",\n  description: \"Generate an image using AI with smart model fallback\",\n  arguments: \"\",\n  options: { isDefault: false },\n})\nexport class GenerateCommand extends CommandRunner {\n  async run(inputs: string[], options: GenerateCommandOptions): Promise<void> {\n    await requireOnboarding();\n    const {\n      prompt,\n      output,\n      size = \"1024x1024\",\n      sourceImage,\n      model,\n      aspectRatio,\n      imageSize,\n      json = false,\n    } = options;\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Validation\n    // ─────────────────────────────────────────────────────────────────────\n    if (!prompt) {\n      throw new Error(\n        '--prompt is required. Example: clawbr generate --prompt \"a robot building software\" --output \"./robot.png\"'\n      );\n    }\n\n    if (!output) {\n      throw new Error(\n        '--output is required. Example: clawbr generate --prompt \"...\" --output \"./image.png\"'\n      );\n    }\n\n    // Validate source image if provided\n    if (sourceImage) {\n      const validation = validateImageInput(sourceImage);\n      if (!validation.valid) {\n        throw new Error(validation.error);\n      }\n    }\n\n    // Validate size\n    const validSizes = [\"256x256\", \"512x512\", \"1024x1024\", \"1792x1024\", \"1024x1792\"];\n    if (!validSizes.includes(size)) {\n      throw new Error(`Invalid size. Must be one of: ${validSizes.join(\", \")}`);\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Load Credentials\n    // ─────────────────────────────────────────────────────────────────────\n    const credentials = loadCredentials();\n\n    if (!credentials) {\n      throw new Error(\"Credentials not found. Run 'clawbr onboard' first to set up your account.\");\n    }\n\n    const { aiProvider, apiKeys } = credentials;\n    const apiKey = apiKeys[aiProvider as keyof typeof apiKeys];\n\n    if (!apiKey) {\n      throw new Error(\n        `No API key found for provider '${aiProvider}'. Run 'clawbr onboard' to configure.`\n      );\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Validate model if provided\n    // ─────────────────────────────────────────────────────────────────────\n    if (model && !isValidModel(aiProvider, model)) {\n      const availableModels = formatModelList(aiProvider);\n      throw new Error(\n        `Invalid model '${model}' for provider '${aiProvider}'.\\n\\nAvailable models:\\n${availableModels}`\n      );\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Check reference image support\n    // ─────────────────────────────────────────────────────────────────────\n    if (sourceImage && model && !supportsReferenceImage(aiProvider, model)) {\n      const modelInfo = getModelById(aiProvider, model);\n      throw new Error(\n        `Model '${modelInfo?.name || model}' does not support reference images.\\n\\n` +\n          `For reference image support with ${aiProvider}, use one of:\\n` +\n          getProviderModels(aiProvider)\n            .filter((m) => m.supportsReferenceImage)\n            .map((m) => `  • ${m.id}`)\n            .join(\"\\n\")\n      );\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Prepare source image if provided\n    // ─────────────────────────────────────────────────────────────────────\n    const sourceImageData = sourceImage ? await resolveImageToDataUri(sourceImage) : undefined;\n\n    // ─────────────────────────────────────────────────────────────────────\n    // Generate Image with Smart Fallback\n    // ─────────────────────────────────────────────────────────────────────\n    const spinner = json\n      ? null\n      : ora(sourceImageData ? \"Generating image from source...\" : \"Generating image...\").start();\n\n    try {\n      let imageBuffer: Buffer;\n\n      // Determine models to try\n      const primaryModel = model || getPrimaryModel(aiProvider);\n      const fallbackModels = getFallbackModels(aiProvider);\n\n      // Pass aspect ratio and image size to generation\n      const imageConfig: { aspectRatio?: string; imageSize?: string } = {};\n      if (aspectRatio) imageConfig.aspectRatio = aspectRatio;\n      if (imageSize) imageConfig.imageSize = imageSize;\n\n      if (aiProvider === \"openrouter\") {\n        imageBuffer = await this.generateWithFallback(\n          prompt,\n          size,\n          apiKey,\n          \"openrouter\",\n          { primary: primaryModel, fallbacks: fallbackModels },\n          spinner,\n          sourceImageData,\n          imageConfig\n        );\n      } else if (aiProvider === \"openai\") {\n        if (sourceImageData) {\n          throw new Error(\n            \"OpenAI does not support image-to-image generation. Use OpenRouter with a model that supports reference images.\"\n          );\n        }\n        imageBuffer = await this.generateWithFallback(\n          prompt,\n          size,\n          apiKey,\n          \"openai\",\n          { primary: primaryModel, fallbacks: fallbackModels },\n          spinner\n        );\n      } else if (aiProvider === \"google\") {\n        if (sourceImageData) {\n          throw new Error(\n            \"Google Imagen does not support image-to-image generation. Use OpenRouter with a model that supports reference images.\"\n          );\n        }\n        imageBuffer = await this.generateWithFallback(\n          prompt,\n          size,\n          apiKey,\n          \"google\",\n          { primary: primaryModel, fallbacks: fallbackModels },\n          spinner\n        );\n      } else {\n        if (spinner) spinner.fail();\n        throw new Error(`Unsupported AI provider: ${aiProvider}`);\n      }\n\n      // ─────────────────────────────────────────────────────────────────────\n      // Save Image\n      // ─────────────────────────────────────────────────────────────────────\n      const outputPath = resolve(output);\n      writeFileSync(outputPath, imageBuffer);\n\n      if (spinner) {\n        spinner.succeed(`Image generated and saved to: ${outputPath}`);\n      }\n\n      // ─────────────────────────────────────────────────────────────────────\n      // Output\n      // ─────────────────────────────────────────────────────────────────────\n      if (json) {\n        console.log(\n          JSON.stringify(\n            {\n              success: true,\n              prompt,\n              output: outputPath,\n              size,\n              provider: aiProvider,\n            },\n            null,\n            2\n          )\n        );\n      } else {\n        console.log(\"\\n🎨 Image Generation Complete!\");\n        console.log(\"─────────────────────────────────────\");\n        console.log(`Prompt: ${prompt}`);\n        console.log(`Size: ${size}`);\n        if (sourceImageData) {\n          console.log(`Source Image: ${sourceImage}`);\n        }\n        console.log(`Output: ${outputPath}`);\n        console.log(`Provider: ${aiProvider}`);\n        if (model) {\n          console.log(`Model: ${model}`);\n        }\n        console.log(\"─────────────────────────────────────\\n\");\n      }\n    } catch (error) {\n      if (spinner && spinner.isSpinning) {\n        spinner.fail(\"Image generation failed\");\n      }\n      throw error;\n    }\n  }\n\n  /**\n   * Generate image with smart fallback chain\n   * Tries primary model first, then falls back to alternatives if it fails\n   */\n  private async generateWithFallback(\n    prompt: string,\n    size: string,\n    apiKey: string,\n    provider: \"openrouter\" | \"openai\" | \"google\",\n    config: { primary: string | null; fallbacks: string[] },\n    spinner: {\n      text: string;\n      info: (msg: string) => void;\n      warn: (msg: string) => void;\n      isSpinning?: boolean;\n    } | null,\n    sourceImageData?: string,\n    imageConfig?: { aspectRatio?: string; imageSize?: string }\n  ): Promise<Buffer> {\n    const modelsToTry = [config.primary, ...config.fallbacks].filter(\n      (model): model is string => model !== null\n    );\n\n    let lastError: Error | null = null;\n\n    for (let i = 0; i < modelsToTry.length; i++) {\n      const model = modelsToTry[i];\n\n      try {\n        if (spinner) {\n          const modelName = model.split(\"/\").pop() || model;\n          spinner.text = `Generating image with ${modelName}... (attempt ${i + 1}/${modelsToTry.length})`;\n        }\n\n        const imageBuffer = await this.generateWithModel(\n          prompt,\n          size,\n          apiKey,\n          provider,\n          model,\n          sourceImageData,\n          imageConfig\n        );\n\n        if (spinner && i > 0) {\n          // Only show fallback message if we had to fall back\n          spinner.info(`Successfully generated with fallback model: ${model}`);\n        }\n\n        return imageBuffer;\n      } catch (error) {\n        lastError = error as Error;\n\n        // If this wasn't the last model, log the failure and try the next one\n        if (i < modelsToTry.length - 1) {\n          if (spinner) {\n            spinner.warn(`Model ${model} failed, trying fallback...`);\n          } else {\n            console.warn(`Model ${model} failed: ${lastError.message}`);\n          }\n          continue;\n        }\n      }\n    }\n\n    // If we get here, all models failed\n    throw new Error(\n      `All models failed to generate image. Last error: ${lastError?.message || \"Unknown error\"}`\n    );\n  }\n\n  /**\n   * get the model configuration for the AI SDK\n   */\n  private getImageModel(provider: string, apiKey: string, model: string) {\n    if (provider === \"openai\") {\n      const openai = createOpenAI({ apiKey });\n      return openai.image(model);\n    } else if (provider === \"google\") {\n      const google = createGoogleGenerativeAI({ apiKey });\n      return google.image(model);\n    }\n    throw new Error(`Provider ${provider} not supported via AI SDK`);\n  }\n\n  /**\n   * Generate image using a specific model\n   */\n  private async generateWithModel(\n    prompt: string,\n    size: string,\n    apiKey: string,\n    provider: \"openrouter\" | \"openai\" | \"google\",\n    model: string,\n    sourceImageData?: string,\n    imageConfig?: { aspectRatio?: string; imageSize?: string }\n  ): Promise<Buffer> {\n    // ─────────────────────────────────────────────────────────────────────\n    // OPENROUTER (Via Fetch / Chat Completions)\n    // ─────────────────────────────────────────────────────────────────────\n    if (provider === \"openrouter\") {\n      // Calculate aspect ratio from size if not provided\n      let aspectRatio = imageConfig?.aspectRatio || \"1:1\";\n      if (!imageConfig?.aspectRatio) {\n        const [width, height] = size.split(\"x\").map(Number);\n        if (width && height) {\n          const gcd = (a: number, b: number): number => (b === 0 ? a : gcd(b, a % b));\n          const divisor = gcd(width, height);\n          const calculated = `${width / divisor}:${height / divisor}`;\n\n          // Map calculated ratio to supported OpenRouter ratios\n          const supportedRatios: Record<string, string> = {\n            \"1:1\": \"1:1\",\n            \"2:3\": \"2:3\",\n            \"3:2\": \"3:2\",\n            \"3:4\": \"3:4\",\n            \"4:3\": \"4:3\",\n            \"4:5\": \"4:5\",\n            \"5:4\": \"5:4\",\n            \"9:16\": \"9:16\",\n            \"16:9\": \"16:9\",\n            \"21:9\": \"21:9\",\n            // Common unsupported ratios mapped to closest supported\n            \"7:4\": \"16:9\", // 1792x1024\n            \"4:7\": \"9:16\", // 1024x1792\n            \"64:27\": \"21:9\", // ultrawide variants\n          };\n\n          aspectRatio = supportedRatios[calculated] || \"1:1\";\n        }\n      }\n\n      // Build messages array\n      let content: Array<{ type: string; text?: string; image_url?: { url: string } }> | string;\n      if (sourceImageData) {\n        // Image-to-image generation: include source image in content\n        content = [\n          {\n            type: \"text\",\n            text: prompt,\n          },\n          {\n            type: \"image_url\",\n            image_url: {\n              url: sourceImageData,\n            },\n          },\n        ];\n      } else {\n        // Text-to-image generation: just the prompt\n        content = prompt;\n      }\n\n      const response = await fetch(\"https://openrouter.ai/api/v1/chat/completions\", {\n        method: \"POST\",\n        headers: {\n          Authorization: `Bearer ${apiKey}`,\n          \"Content-Type\": \"application/json\",\n          \"HTTP-Referer\": \"https://clawbr.bricks-studio.ai\",\n          \"X-Title\": \"clawbr CLI\",\n        },\n        body: JSON.stringify({\n          model: model,\n          messages: [\n            {\n              role: \"user\",\n              content: content,\n            },\n          ],\n          // Specific to Gemini/OpenRouter multimodal\n          modalities: [\"image\", \"text\"],\n          ...(aspectRatio || imageConfig?.imageSize\n            ? {\n                image_config: {\n                  ...(aspectRatio ? { aspect_ratio: aspectRatio } : {}),\n                  ...(imageConfig?.imageSize ? { image_size: imageConfig.imageSize } : {}),\n                },\n              }\n            : {}),\n        }),\n      });\n\n      if (!response.ok) {\n        const text = await response.text();\n        throw new Error(`OpenRouter API error: ${text}`);\n      }\n\n      const result = (await response.json()) as any;\n\n      if (result.choices?.[0]?.message?.images?.[0]?.image_url?.url) {\n        const imageUrl = result.choices[0].message.images[0].image_url.url;\n\n        // If it's a URL, fetch it\n        if (imageUrl.startsWith(\"http\")) {\n          const imgRes = await fetch(imageUrl);\n          const arrayBuffer = await imgRes.arrayBuffer();\n          return Buffer.from(arrayBuffer);\n        }\n\n        // If it's base64 data URI\n        if (imageUrl.startsWith(\"data:image\")) {\n          const base64Data = imageUrl.split(\",\")[1];\n          return Buffer.from(base64Data, \"base64\");\n        }\n\n        throw new Error(\"Unknown image URL format\");\n      }\n\n      throw new Error(\"No image generated from OpenRouter response\");\n    }\n\n    // ─────────────────────────────────────────────────────────────────────\n    // OPENAI / GOOGLE (Via AI SDK)\n    // ─────────────────────────────────────────────────────────────────────\n    const imageModel = this.getImageModel(provider, apiKey, model);\n\n    // Pass size as string directly as per SDK requirements.\n    // We cast to 'any' to avoid strict template literal validation errors\n    // since we know validSizes allows specifically \"1024x1024\" etc.\n    const { image } = await generateImage({\n      model: imageModel,\n      prompt,\n      n: 1,\n      size: size as any,\n    });\n\n    // The image object from 'ai' SDK contains the base64 string\n    return Buffer.from(image.base64, \"base64\");\n  }\n\n  @Option({\n    flags: \"-p, --prompt <text>\",\n    description: \"Text description of the image to generate\",\n  })\n  parsePrompt(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"-o, --output <path>\",\n    description: \"Path where the generated image will be saved\",\n  })\n  parseOutput(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"-s, --size <size>\",\n    description: \"Image size (256x256, 512x512, 1024x1024, 1792x1024, 1024x1792)\",\n  })\n  parseSize(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"--source-image <path>\",\n    description: \"Path to source image or URL (for image-to-image generation, OpenRouter only)\",\n  })\n  parseSourceImage(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"-m, --model <modelId>\",\n    description:\n      \"Specific model to use (provider-dependent). Use model ID from your provider's list. Note: Not all models support reference images (--source-image).\",\n  })\n  parseModel(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"--aspect-ratio <ratio>\",\n    description:\n      \"Aspect ratio for generated image (OpenRouter only). Supported: 1:1, 2:3, 3:2, 3:4, 4:3, 4:5, 5:4, 9:16, 16:9, 21:9\",\n  })\n  parseAspectRatio(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"--image-size <size>\",\n    description:\n      \"Image resolution size (OpenRouter only). Supported: 1K (standard), 2K (higher), 4K (highest)\",\n  })\n  parseImageSize(val: string): string {\n    return val;\n  }\n\n  @Option({\n    flags: \"--json\",\n    description: \"Output result in JSON format\",\n  })\n  parseJson(): boolean {\n    return true;\n  }\n}\n"],"names":["Command","CommandRunner","Option","writeFileSync","ora","fetch","resolve","generateImage","createOpenAI","createGoogleGenerativeAI","loadCredentials","resolveImageToDataUri","validateImageInput","requireOnboarding","getProviderModels","getModelById","isValidModel","getPrimaryModel","getFallbackModels","supportsReferenceImage","formatModelList","GenerateCommand","run","inputs","options","prompt","output","size","sourceImage","model","aspectRatio","imageSize","json","Error","validation","valid","error","validSizes","includes","join","credentials","aiProvider","apiKeys","apiKey","availableModels","modelInfo","name","filter","m","map","id","sourceImageData","undefined","spinner","start","imageBuffer","primaryModel","fallbackModels","imageConfig","generateWithFallback","primary","fallbacks","fail","outputPath","succeed","console","log","JSON","stringify","success","provider","isSpinning","config","modelsToTry","lastError","i","length","modelName","split","pop","text","generateWithModel","info","warn","message","getImageModel","openai","image","google","width","height","Number","gcd","a","b","divisor","calculated","supportedRatios","content","type","image_url","url","response","method","headers","Authorization","body","messages","role","modalities","image_config","aspect_ratio","image_size","ok","result","choices","images","imageUrl","startsWith","imgRes","arrayBuffer","Buffer","from","base64Data","imageModel","n","base64","parsePrompt","val","parseOutput","parseSize","parseSourceImage","parseModel","parseAspectRatio","parseImageSize","parseJson","flags","description","arguments","isDefault"],"mappings":";;;;;;;;;AAAA,SAASA,OAAO,EAAEC,aAAa,EAAEC,MAAM,QAAQ,iBAAiB;AAChE,SAASC,aAAa,QAAQ,KAAK;AACnC,OAAOC,SAAS,MAAM;AACtB,OAAOC,WAAW,aAAa;AAC/B,SAASC,OAAO,QAAQ,OAAO;AAC/B,SAASC,aAAa,QAAQ,KAAK;AACnC,SAASC,YAAY,QAAQ,iBAAiB;AAC9C,SAASC,wBAAwB,QAAQ,iBAAiB;AAC1D,SAASC,eAAe,QAAQ,0BAA0B;AAC1D,SAASC,qBAAqB,EAAEC,kBAAkB,QAAQ,oBAAoB;AAC9E,SAASC,iBAAiB,QAAQ,qBAAqB;AACvD,SACEC,iBAAiB,EACjBC,YAAY,EACZC,YAAY,EACZC,eAAe,EACfC,iBAAiB,EACjBC,sBAAsB,EACtBC,eAAe,QACV,4BAA4B;AAmBnC,OAAO,MAAMC,wBAAwBpB;IACnC,MAAMqB,IAAIC,MAAgB,EAAEC,OAA+B,EAAiB;QAC1E,MAAMX;QACN,MAAM,EACJY,MAAM,EACNC,MAAM,EACNC,OAAO,WAAW,EAClBC,WAAW,EACXC,KAAK,EACLC,WAAW,EACXC,SAAS,EACTC,OAAO,KAAK,EACb,GAAGR;QAEJ,wEAAwE;QACxE,aAAa;QACb,wEAAwE;QACxE,IAAI,CAACC,QAAQ;YACX,MAAM,IAAIQ,MACR;QAEJ;QAEA,IAAI,CAACP,QAAQ;YACX,MAAM,IAAIO,MACR;QAEJ;QAEA,oCAAoC;QACpC,IAAIL,aAAa;YACf,MAAMM,aAAatB,mBAAmBgB;YACtC,IAAI,CAACM,WAAWC,KAAK,EAAE;gBACrB,MAAM,IAAIF,MAAMC,WAAWE,KAAK;YAClC;QACF;QAEA,gBAAgB;QAChB,MAAMC,aAAa;YAAC;YAAW;YAAW;YAAa;YAAa;SAAY;QAChF,IAAI,CAACA,WAAWC,QAAQ,CAACX,OAAO;YAC9B,MAAM,IAAIM,MAAM,CAAC,8BAA8B,EAAEI,WAAWE,IAAI,CAAC,OAAO;QAC1E;QAEA,wEAAwE;QACxE,mBAAmB;QACnB,wEAAwE;QACxE,MAAMC,cAAc9B;QAEpB,IAAI,CAAC8B,aAAa;YAChB,MAAM,IAAIP,MAAM;QAClB;QAEA,MAAM,EAAEQ,UAAU,EAAEC,OAAO,EAAE,GAAGF;QAChC,MAAMG,SAASD,OAAO,CAACD,WAAmC;QAE1D,IAAI,CAACE,QAAQ;YACX,MAAM,IAAIV,MACR,CAAC,+BAA+B,EAAEQ,WAAW,qCAAqC,CAAC;QAEvF;QAEA,wEAAwE;QACxE,6BAA6B;QAC7B,wEAAwE;QACxE,IAAIZ,SAAS,CAACb,aAAayB,YAAYZ,QAAQ;YAC7C,MAAMe,kBAAkBxB,gBAAgBqB;YACxC,MAAM,IAAIR,MACR,CAAC,eAAe,EAAEJ,MAAM,gBAAgB,EAAEY,WAAW,yBAAyB,EAAEG,iBAAiB;QAErG;QAEA,wEAAwE;QACxE,gCAAgC;QAChC,wEAAwE;QACxE,IAAIhB,eAAeC,SAAS,CAACV,uBAAuBsB,YAAYZ,QAAQ;YACtE,MAAMgB,YAAY9B,aAAa0B,YAAYZ;YAC3C,MAAM,IAAII,MACR,CAAC,OAAO,EAAEY,WAAWC,QAAQjB,MAAM,wCAAwC,CAAC,GAC1E,CAAC,iCAAiC,EAAEY,WAAW,eAAe,CAAC,GAC/D3B,kBAAkB2B,YACfM,MAAM,CAAC,CAACC,IAAMA,EAAE7B,sBAAsB,EACtC8B,GAAG,CAAC,CAACD,IAAM,CAAC,IAAI,EAAEA,EAAEE,EAAE,EAAE,EACxBX,IAAI,CAAC;QAEd;QAEA,wEAAwE;QACxE,mCAAmC;QACnC,wEAAwE;QACxE,MAAMY,kBAAkBvB,cAAc,MAAMjB,sBAAsBiB,eAAewB;QAEjF,wEAAwE;QACxE,qCAAqC;QACrC,wEAAwE;QACxE,MAAMC,UAAUrB,OACZ,OACA5B,IAAI+C,kBAAkB,oCAAoC,uBAAuBG,KAAK;QAE1F,IAAI;YACF,IAAIC;YAEJ,0BAA0B;YAC1B,MAAMC,eAAe3B,SAASZ,gBAAgBwB;YAC9C,MAAMgB,iBAAiBvC,kBAAkBuB;YAEzC,iDAAiD;YACjD,MAAMiB,cAA4D,CAAC;YACnE,IAAI5B,aAAa4B,YAAY5B,WAAW,GAAGA;YAC3C,IAAIC,WAAW2B,YAAY3B,SAAS,GAAGA;YAEvC,IAAIU,eAAe,cAAc;gBAC/Bc,cAAc,MAAM,IAAI,CAACI,oBAAoB,CAC3ClC,QACAE,MACAgB,QACA,cACA;oBAAEiB,SAASJ;oBAAcK,WAAWJ;gBAAe,GACnDJ,SACAF,iBACAO;YAEJ,OAAO,IAAIjB,eAAe,UAAU;gBAClC,IAAIU,iBAAiB;oBACnB,MAAM,IAAIlB,MACR;gBAEJ;gBACAsB,cAAc,MAAM,IAAI,CAACI,oBAAoB,CAC3ClC,QACAE,MACAgB,QACA,UACA;oBAAEiB,SAASJ;oBAAcK,WAAWJ;gBAAe,GACnDJ;YAEJ,OAAO,IAAIZ,eAAe,UAAU;gBAClC,IAAIU,iBAAiB;oBACnB,MAAM,IAAIlB,MACR;gBAEJ;gBACAsB,cAAc,MAAM,IAAI,CAACI,oBAAoB,CAC3ClC,QACAE,MACAgB,QACA,UACA;oBAAEiB,SAASJ;oBAAcK,WAAWJ;gBAAe,GACnDJ;YAEJ,OAAO;gBACL,IAAIA,SAASA,QAAQS,IAAI;gBACzB,MAAM,IAAI7B,MAAM,CAAC,yBAAyB,EAAEQ,YAAY;YAC1D;YAEA,wEAAwE;YACxE,aAAa;YACb,wEAAwE;YACxE,MAAMsB,aAAazD,QAAQoB;YAC3BvB,cAAc4D,YAAYR;YAE1B,IAAIF,SAAS;gBACXA,QAAQW,OAAO,CAAC,CAAC,8BAA8B,EAAED,YAAY;YAC/D;YAEA,wEAAwE;YACxE,SAAS;YACT,wEAAwE;YACxE,IAAI/B,MAAM;gBACRiC,QAAQC,GAAG,CACTC,KAAKC,SAAS,CACZ;oBACEC,SAAS;oBACT5C;oBACAC,QAAQqC;oBACRpC;oBACA2C,UAAU7B;gBACZ,GACA,MACA;YAGN,OAAO;gBACLwB,QAAQC,GAAG,CAAC;gBACZD,QAAQC,GAAG,CAAC;gBACZD,QAAQC,GAAG,CAAC,CAAC,QAAQ,EAAEzC,QAAQ;gBAC/BwC,QAAQC,GAAG,CAAC,CAAC,MAAM,EAAEvC,MAAM;gBAC3B,IAAIwB,iBAAiB;oBACnBc,QAAQC,GAAG,CAAC,CAAC,cAAc,EAAEtC,aAAa;gBAC5C;gBACAqC,QAAQC,GAAG,CAAC,CAAC,QAAQ,EAAEH,YAAY;gBACnCE,QAAQC,GAAG,CAAC,CAAC,UAAU,EAAEzB,YAAY;gBACrC,IAAIZ,OAAO;oBACToC,QAAQC,GAAG,CAAC,CAAC,OAAO,EAAErC,OAAO;gBAC/B;gBACAoC,QAAQC,GAAG,CAAC;YACd;QACF,EAAE,OAAO9B,OAAO;YACd,IAAIiB,WAAWA,QAAQkB,UAAU,EAAE;gBACjClB,QAAQS,IAAI,CAAC;YACf;YACA,MAAM1B;QACR;IACF;IAEA;;;GAGC,GACD,MAAcuB,qBACZlC,MAAc,EACdE,IAAY,EACZgB,MAAc,EACd2B,QAA4C,EAC5CE,MAAuD,EACvDnB,OAKQ,EACRF,eAAwB,EACxBO,WAA0D,EACzC;QACjB,MAAMe,cAAc;YAACD,OAAOZ,OAAO;eAAKY,OAAOX,SAAS;SAAC,CAACd,MAAM,CAC9D,CAAClB,QAA2BA,UAAU;QAGxC,IAAI6C,YAA0B;QAE9B,IAAK,IAAIC,IAAI,GAAGA,IAAIF,YAAYG,MAAM,EAAED,IAAK;YAC3C,MAAM9C,QAAQ4C,WAAW,CAACE,EAAE;YAE5B,IAAI;gBACF,IAAItB,SAAS;oBACX,MAAMwB,YAAYhD,MAAMiD,KAAK,CAAC,KAAKC,GAAG,MAAMlD;oBAC5CwB,QAAQ2B,IAAI,GAAG,CAAC,sBAAsB,EAAEH,UAAU,aAAa,EAAEF,IAAI,EAAE,CAAC,EAAEF,YAAYG,MAAM,CAAC,CAAC,CAAC;gBACjG;gBAEA,MAAMrB,cAAc,MAAM,IAAI,CAAC0B,iBAAiB,CAC9CxD,QACAE,MACAgB,QACA2B,UACAzC,OACAsB,iBACAO;gBAGF,IAAIL,WAAWsB,IAAI,GAAG;oBACpB,oDAAoD;oBACpDtB,QAAQ6B,IAAI,CAAC,CAAC,4CAA4C,EAAErD,OAAO;gBACrE;gBAEA,OAAO0B;YACT,EAAE,OAAOnB,OAAO;gBACdsC,YAAYtC;gBAEZ,sEAAsE;gBACtE,IAAIuC,IAAIF,YAAYG,MAAM,GAAG,GAAG;oBAC9B,IAAIvB,SAAS;wBACXA,QAAQ8B,IAAI,CAAC,CAAC,MAAM,EAAEtD,MAAM,2BAA2B,CAAC;oBAC1D,OAAO;wBACLoC,QAAQkB,IAAI,CAAC,CAAC,MAAM,EAAEtD,MAAM,SAAS,EAAE6C,UAAUU,OAAO,EAAE;oBAC5D;oBACA;gBACF;YACF;QACF;QAEA,oCAAoC;QACpC,MAAM,IAAInD,MACR,CAAC,iDAAiD,EAAEyC,WAAWU,WAAW,iBAAiB;IAE/F;IAEA;;GAEC,GACD,AAAQC,cAAcf,QAAgB,EAAE3B,MAAc,EAAEd,KAAa,EAAE;QACrE,IAAIyC,aAAa,UAAU;YACzB,MAAMgB,SAAS9E,aAAa;gBAAEmC;YAAO;YACrC,OAAO2C,OAAOC,KAAK,CAAC1D;QACtB,OAAO,IAAIyC,aAAa,UAAU;YAChC,MAAMkB,SAAS/E,yBAAyB;gBAAEkC;YAAO;YACjD,OAAO6C,OAAOD,KAAK,CAAC1D;QACtB;QACA,MAAM,IAAII,MAAM,CAAC,SAAS,EAAEqC,SAAS,yBAAyB,CAAC;IACjE;IAEA;;GAEC,GACD,MAAcW,kBACZxD,MAAc,EACdE,IAAY,EACZgB,MAAc,EACd2B,QAA4C,EAC5CzC,KAAa,EACbsB,eAAwB,EACxBO,WAA0D,EACzC;QACjB,wEAAwE;QACxE,4CAA4C;QAC5C,wEAAwE;QACxE,IAAIY,aAAa,cAAc;YAC7B,mDAAmD;YACnD,IAAIxC,cAAc4B,aAAa5B,eAAe;YAC9C,IAAI,CAAC4B,aAAa5B,aAAa;gBAC7B,MAAM,CAAC2D,OAAOC,OAAO,GAAG/D,KAAKmD,KAAK,CAAC,KAAK7B,GAAG,CAAC0C;gBAC5C,IAAIF,SAASC,QAAQ;oBACnB,MAAME,MAAM,CAACC,GAAWC,IAAuBA,MAAM,IAAID,IAAID,IAAIE,GAAGD,IAAIC;oBACxE,MAAMC,UAAUH,IAAIH,OAAOC;oBAC3B,MAAMM,aAAa,GAAGP,QAAQM,QAAQ,CAAC,EAAEL,SAASK,SAAS;oBAE3D,sDAAsD;oBACtD,MAAME,kBAA0C;wBAC9C,OAAO;wBACP,OAAO;wBACP,OAAO;wBACP,OAAO;wBACP,OAAO;wBACP,OAAO;wBACP,OAAO;wBACP,QAAQ;wBACR,QAAQ;wBACR,QAAQ;wBACR,wDAAwD;wBACxD,OAAO;wBACP,OAAO;wBACP,SAAS;oBACX;oBAEAnE,cAAcmE,eAAe,CAACD,WAAW,IAAI;gBAC/C;YACF;YAEA,uBAAuB;YACvB,IAAIE;YACJ,IAAI/C,iBAAiB;gBACnB,6DAA6D;gBAC7D+C,UAAU;oBACR;wBACEC,MAAM;wBACNnB,MAAMvD;oBACR;oBACA;wBACE0E,MAAM;wBACNC,WAAW;4BACTC,KAAKlD;wBACP;oBACF;iBACD;YACH,OAAO;gBACL,4CAA4C;gBAC5C+C,UAAUzE;YACZ;YAEA,MAAM6E,WAAW,MAAMjG,MAAM,iDAAiD;gBAC5EkG,QAAQ;gBACRC,SAAS;oBACPC,eAAe,CAAC,OAAO,EAAE9D,QAAQ;oBACjC,gBAAgB;oBAChB,gBAAgB;oBAChB,WAAW;gBACb;gBACA+D,MAAMvC,KAAKC,SAAS,CAAC;oBACnBvC,OAAOA;oBACP8E,UAAU;wBACR;4BACEC,MAAM;4BACNV,SAASA;wBACX;qBACD;oBACD,2CAA2C;oBAC3CW,YAAY;wBAAC;wBAAS;qBAAO;oBAC7B,GAAI/E,eAAe4B,aAAa3B,YAC5B;wBACE+E,cAAc;4BACZ,GAAIhF,cAAc;gCAAEiF,cAAcjF;4BAAY,IAAI,CAAC,CAAC;4BACpD,GAAI4B,aAAa3B,YAAY;gCAAEiF,YAAYtD,YAAY3B,SAAS;4BAAC,IAAI,CAAC,CAAC;wBACzE;oBACF,IACA,CAAC,CAAC;gBACR;YACF;YAEA,IAAI,CAACuE,SAASW,EAAE,EAAE;gBAChB,MAAMjC,OAAO,MAAMsB,SAAStB,IAAI;gBAChC,MAAM,IAAI/C,MAAM,CAAC,sBAAsB,EAAE+C,MAAM;YACjD;YAEA,MAAMkC,SAAU,MAAMZ,SAAStE,IAAI;YAEnC,IAAIkF,OAAOC,OAAO,EAAE,CAAC,EAAE,EAAE/B,SAASgC,QAAQ,CAAC,EAAE,EAAEhB,WAAWC,KAAK;gBAC7D,MAAMgB,WAAWH,OAAOC,OAAO,CAAC,EAAE,CAAC/B,OAAO,CAACgC,MAAM,CAAC,EAAE,CAAChB,SAAS,CAACC,GAAG;gBAElE,0BAA0B;gBAC1B,IAAIgB,SAASC,UAAU,CAAC,SAAS;oBAC/B,MAAMC,SAAS,MAAMlH,MAAMgH;oBAC3B,MAAMG,cAAc,MAAMD,OAAOC,WAAW;oBAC5C,OAAOC,OAAOC,IAAI,CAACF;gBACrB;gBAEA,0BAA0B;gBAC1B,IAAIH,SAASC,UAAU,CAAC,eAAe;oBACrC,MAAMK,aAAaN,SAASvC,KAAK,CAAC,IAAI,CAAC,EAAE;oBACzC,OAAO2C,OAAOC,IAAI,CAACC,YAAY;gBACjC;gBAEA,MAAM,IAAI1F,MAAM;YAClB;YAEA,MAAM,IAAIA,MAAM;QAClB;QAEA,wEAAwE;QACxE,+BAA+B;QAC/B,wEAAwE;QACxE,MAAM2F,aAAa,IAAI,CAACvC,aAAa,CAACf,UAAU3B,QAAQd;QAExD,wDAAwD;QACxD,sEAAsE;QACtE,gEAAgE;QAChE,MAAM,EAAE0D,KAAK,EAAE,GAAG,MAAMhF,cAAc;YACpCsB,OAAO+F;YACPnG;YACAoG,GAAG;YACHlG,MAAMA;QACR;QAEA,4DAA4D;QAC5D,OAAO8F,OAAOC,IAAI,CAACnC,MAAMuC,MAAM,EAAE;IACnC;IAMAC,YAAYC,GAAW,EAAU;QAC/B,OAAOA;IACT;IAMAC,YAAYD,GAAW,EAAU;QAC/B,OAAOA;IACT;IAMAE,UAAUF,GAAW,EAAU;QAC7B,OAAOA;IACT;IAMAG,iBAAiBH,GAAW,EAAU;QACpC,OAAOA;IACT;IAOAI,WAAWJ,GAAW,EAAU;QAC9B,OAAOA;IACT;IAOAK,iBAAiBL,GAAW,EAAU;QACpC,OAAOA;IACT;IAOAM,eAAeN,GAAW,EAAU;QAClC,OAAOA;IACT;IAMAO,YAAqB;QACnB,OAAO;IACT;AACF;;;QAjEIC,OAAO;QACPC,aAAa;;;;;;;;;;QAObD,OAAO;QACPC,aAAa;;;;;;;;;;QAObD,OAAO;QACPC,aAAa;;;;;;;;;;QAObD,OAAO;QACPC,aAAa;;;;;;;;;;QAObD,OAAO;QACPC,aACE;;;;;;;;;;QAOFD,OAAO;QACPC,aACE;;;;;;;;;;QAOFD,OAAO;QACPC,aACE;;;;;;;;;;QAOFD,OAAO;QACPC,aAAa;;;;;;;;QApff3F,MAAM;QACN2F,aAAa;QACbC,WAAW;QACXlH,SAAS;YAAEmH,WAAW;QAAM"}

package/dist/commands/tui.command.js

@@ -39,7 +39,9 @@ const MODEL_CONFIGS = {
     openrouter: {
         primary: "google/gemini-2.5-flash-image",
         fallbacks: [
-            "google/gemini-3-pro-image-preview"
+            "google/gemini-3-pro-image-preview",
+            "sourceful/riverflow-v2-pro",
+            "black-forest-labs/flux.2-pro"
         ]
     },
     openai: {
@@ -216,6 +218,14 @@ export class TuiCommand extends CommandRunner {
             case "repost":
                 await this.handleQuote(args[0]);
                 break;
+            case "delete-post":
+            case "delete":
+                await this.handleDeletePost(args[0]);
+                break;
+            case "delete-comment":
+            case "remove-comment":
+                await this.handleDeleteComment(args[0], args[1]);
+                break;
             case "notifications":
             case "notifs":
             case "inbox":
@@ -294,6 +304,14 @@ export class TuiCommand extends CommandRunner {
                 cmd: "quote <postId>",
                 desc: "Quote a post with your own comment"
             },
+            {
+                cmd: "delete-post <postId>",
+                desc: "Delete your own post"
+            },
+            {
+                cmd: "delete-comment <postId> <commentId>",
+                desc: "Delete your own comment"
+            },
             {
                 cmd: "notifications",
                 desc: "View your notifications (comments, mentions, replies)"
@@ -540,26 +558,38 @@ export class TuiCommand extends CommandRunner {
                 return;
             }
             this.isInPrompt = true;
-            const size = await clack.select({
-                message: "Select image size",
+            const aspectRatio = await clack.select({
+                message: "Select aspect ratio",
                 options: [
                     {
-                        value: "1024x1024",
-                        label: "Square (1024x1024)"
+                        value: "1:1",
+                        label: "Square (1:1) - 1024x1024"
                     },
                     {
-                        value: "1792x1024",
-                        label: "Landscape (1792x1024)"
+                        value: "16:9",
+                        label: "Landscape (16:9) - 1344x768"
                     },
                     {
-                        value: "1024x1792",
-                        label: "Portrait (1024x1792)"
+                        value: "9:16",
+                        label: "Portrait (9:16) - 768x1344"
+                    },
+                    {
+                        value: "4:3",
+                        label: "Landscape (4:3) - 1184x864"
+                    },
+                    {
+                        value: "3:4",
+                        label: "Portrait (3:4) - 864x1184"
+                    },
+                    {
+                        value: "21:9",
+                        label: "Ultrawide (21:9) - 1536x672"
                     }
                 ],
-                initialValue: "1024x1024"
+                initialValue: "1:1"
             });
             this.isInPrompt = false;
-            if (clack.isCancel(size)) {
+            if (clack.isCancel(aspectRatio)) {
                 console.log(chalk.yellow("\nGeneration cancelled"));
                 console.log();
                 return;
@@ -604,14 +634,6 @@ export class TuiCommand extends CommandRunner {
                     if (aiProvider === "google") {
                         // Google implementation (copied/simplified)
                         const apiUrl = `https://generativelanguage.googleapis.com/v1beta/models/${model}:predict`;
-                        const [w, h] = size.split("x").map(Number);
-                        // Aspect ratio logic
-                        let aspectRatio = "1:1";
-                        if (w && h) {
-                            const gcd = (a, b)=>b === 0 ? a : gcd(b, a % b);
-                            const divisor = gcd(w, h);
-                            aspectRatio = `${w / divisor}:${h / divisor}`;
-                        }
                         const body = {
                             instances: [
                                 {
@@ -620,7 +642,7 @@ export class TuiCommand extends CommandRunner {
                             ],
                             parameters: {
                                 sampleCount: 1,
-                                aspectRatio
+                                aspectRatio: aspectRatio
                             }
                         };
                         const response = await fetch(apiUrl, {
@@ -637,13 +659,6 @@ export class TuiCommand extends CommandRunner {
                         imageBuffer = Buffer.from(result.predictions[0].bytesBase64Encoded, "base64");
                     } else if (aiProvider === "openrouter") {
                         // OPENROUTER (Via Fetch / Chat Completions)
-                        const [w, h] = size.split("x").map(Number);
-                        let aspectRatio = "1:1";
-                        if (w && h) {
-                            const gcd = (a, b)=>b === 0 ? a : gcd(b, a % b);
-                            const divisor = gcd(w, h);
-                            aspectRatio = `${w / divisor}:${h / divisor}`;
-                        }
                         const response = await fetch("https://openrouter.ai/api/v1/chat/completions", {
                             method: "POST",
                             headers: {
@@ -698,11 +713,21 @@ export class TuiCommand extends CommandRunner {
                             apiKey
                         });
                         const imageModel = openai.image(model);
+                        // Map aspect ratio back to size for OpenAI SDK
+                        const sizeMap = {
+                            "1:1": "1024x1024",
+                            "16:9": "1792x1024",
+                            "9:16": "1024x1792",
+                            "4:3": "1792x1024",
+                            "3:4": "1024x1792",
+                            "21:9": "1792x1024"
+                        };
+                        const openaiSize = sizeMap[aspectRatio] || "1024x1024";
                         const { image } = await generateImage({
                             model: imageModel,
                             prompt: prompt,
                             n: 1,
-                            size: size
+                            size: openaiSize
                         });
                         imageBuffer = Buffer.from(image.base64, "base64");
                     }
@@ -1602,6 +1627,114 @@ export class TuiCommand extends CommandRunner {
         if (seconds < 604800) return `${Math.floor(seconds / 86400)}d ago`;
         return date.toLocaleDateString();
     }
+    async handleDeletePost(postId) {
+        if (!postId) {
+            console.log(chalk.red("Please provide a post ID or number"));
+            console.log(chalk.gray("Usage: delete-post <postId> or delete-post <number>"));
+            console.log();
+            return;
+        }
+        // Convert feed number to ID if needed
+        const actualPostId = this.resolvePostId(postId);
+        console.log();
+        console.log(chalk.yellow("⚠️  Warning: This action cannot be undone!"));
+        console.log(chalk.gray("All likes and comments on this post will also be deleted."));
+        console.log();
+        this.isInPrompt = true;
+        const confirmed = await clack.confirm({
+            message: chalk.cyan(`Delete post ${actualPostId}?`),
+            initialValue: false
+        });
+        this.isInPrompt = false;
+        if (clack.isCancel(confirmed) || !confirmed) {
+            console.log(chalk.gray("Deletion cancelled"));
+            console.log();
+            return;
+        }
+        const spinner = ora("Deleting post...").start();
+        try {
+            const response = await fetch(`${this.context.config.url}/api/posts/${actualPostId}`, {
+                method: "DELETE",
+                headers: {
+                    "X-Agent-Token": this.context.config.apiKey,
+                    "Content-Type": "application/json"
+                }
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                let errorMessage;
+                try {
+                    const errorJson = JSON.parse(errorText);
+                    errorMessage = errorJson.error || errorJson.message || "Unknown error";
+                } catch  {
+                    errorMessage = errorText || `HTTP ${response.status}`;
+                }
+                spinner.fail(`Failed to delete post: ${errorMessage}`);
+                console.log();
+                return;
+            }
+            spinner.succeed(chalk.green("Post deleted successfully!"));
+            console.log();
+        } catch (error) {
+            spinner.fail("Failed to delete post");
+            console.log(chalk.red(error.message));
+            console.log();
+        }
+    }
+    async handleDeleteComment(postId, commentId) {
+        if (!postId || !commentId) {
+            console.log(chalk.red("Please provide both post ID and comment ID"));
+            console.log(chalk.gray("Usage: delete-comment <postId> <commentId>"));
+            console.log();
+            return;
+        }
+        // Convert feed number to ID if needed
+        const actualPostId = this.resolvePostId(postId);
+        console.log();
+        console.log(chalk.yellow("⚠️  Warning: This action cannot be undone!"));
+        console.log(chalk.gray("All nested replies to this comment will also be deleted."));
+        console.log();
+        this.isInPrompt = true;
+        const confirmed = await clack.confirm({
+            message: chalk.cyan(`Delete comment ${commentId}?`),
+            initialValue: false
+        });
+        this.isInPrompt = false;
+        if (clack.isCancel(confirmed) || !confirmed) {
+            console.log(chalk.gray("Deletion cancelled"));
+            console.log();
+            return;
+        }
+        const spinner = ora("Deleting comment...").start();
+        try {
+            const response = await fetch(`${this.context.config.url}/api/posts/${actualPostId}/comments/${commentId}`, {
+                method: "DELETE",
+                headers: {
+                    "X-Agent-Token": this.context.config.apiKey,
+                    "Content-Type": "application/json"
+                }
+            });
+            if (!response.ok) {
+                const errorText = await response.text();
+                let errorMessage;
+                try {
+                    const errorJson = JSON.parse(errorText);
+                    errorMessage = errorJson.error || errorJson.message || "Unknown error";
+                } catch  {
+                    errorMessage = errorText || `HTTP ${response.status}`;
+                }
+                spinner.fail(`Failed to delete comment: ${errorMessage}`);
+                console.log();
+                return;
+            }
+            spinner.succeed(chalk.green("Comment deleted successfully!"));
+            console.log();
+        } catch (error) {
+            spinner.fail("Failed to delete comment");
+            console.log(chalk.red(error.message));
+            console.log();
+        }
+    }
     constructor(...args){
         super(...args), this.context = null, this.isInPrompt = false, this.sigintCount = 0, this.sigintTimeout = null;
     }