ADd ability to chat to improve the logo image

Author: crunchwrap89

BACKLOG.md

@@ -1,6 +1,9 @@
 # Backlog
 
 ---
-[x] Convert generated logo to multiple sizes and formats
+## Feature name
+Verify if there are any issues.
 
+### Description
+Verify if there are any issues and if you are using the latest way to call the api. I want to generate logos via chat in my CLI.
 ---

deno.json

@@ -11,7 +11,7 @@
     "@std/fs": "jsr:@std/fs@^1.0.21",
     "@std/path": "jsr:@std/path@^1.1.4",
     "@std/fmt/colors": "jsr:@std/fmt@^1.0.8/colors",
-    "@google/genai": "npm:@google/genai@^1.35.0"
+    "@google/genai": "npm:@google/genai@^1.36.0"
   },
   "publish": {
     "exclude": [

deno.lock

@@ -3,6 +3,7 @@ import { join } from "@std/path";
 import { ensureDir } from "@std/fs";
 import { GoogleGenAI } from "@google/genai";
 import { processLogo } from "./image.ts";
+import { promptInput } from "./prompt.ts";
 
 /**
  * Generates a logo using Google's Gemini AI (Imagen model)
@@ -12,41 +13,60 @@ export async function generateLogo(
   apiKey: string,
   destDir: string,
 ): Promise<string | null> {
-  const spinner = startSpinner("🎨 Generating logo with AI...");
+  let spinner = startSpinner("🎨 Generating logo with AI...");
 
   try {
     const ai = new GoogleGenAI({ apiKey });
 
-    const response = await ai.models.generateImages({
+    // Initial message to start the chat
+    const chat = ai.chats.create({
       model: "gemini-3-pro-image-preview",
-      prompt: prompt,
       config: {
-        numberOfImages: 1,
-        aspectRatio: "1:1",
-        outputMimeType: "image/png",
+        responseModalities: ["TEXT", "IMAGE"],
       },
     });
 
-    const base64Image = response.generatedImages?.[0]?.image?.imageBytes;
+    let currentPrompt = prompt;
 
-    if (!base64Image) {
-      spinner.stop(false);
-      console.log("  ⚠️  Failed to generate logo: No image data received from AI.");
-      return null;
-    }
+    while (true) {
+      const response = await chat.sendMessage({ message: currentPrompt });
+
+      let base64Image: string | undefined;
+      for (const part of response.candidates?.[0]?.content?.parts || []) {
+        if (part.inlineData && part.inlineData.mimeType === "image/png") {
+          base64Image = part.inlineData.data;
+        }
+      }
+
+      if (!base64Image) {
+        spinner.stop(false);
+        console.log("  ⚠️  Failed to generate logo: No image data received from AI.");
+        return null;
+      }
 
-    const imageBuffer = Uint8Array.from(atob(base64Image), (c) => c.charCodeAt(0));
-    const publicDir = join(destDir, "public");
-    await ensureDir(publicDir);
-    const logoPath = join(publicDir, "logo.png");
-    await Deno.writeFile(logoPath, imageBuffer);
+      const imageBuffer = Uint8Array.from(atob(base64Image), (c) => c.charCodeAt(0));
+      const publicDir = join(destDir, "public");
+      await ensureDir(publicDir);
+      const logoPath = join(publicDir, "logo.png");
+      await Deno.writeFile(logoPath, imageBuffer);
 
-    spinner.stop(true);
+      spinner.stop(true);
 
-    // Process the logo into multiple sizes and formats
-    await processLogo(logoPath, destDir);
+      // Process the logo into multiple sizes and formats
+      await processLogo(logoPath, destDir);
 
-    return logoPath;
+      const improvement = promptInput(
+        "✨ Would you like to improve this logo? (enter your instructions, or leave empty to finish)",
+        "",
+      );
+
+      if (!improvement) {
+        return logoPath;
+      }
+
+      currentPrompt = improvement;
+      spinner = startSpinner("🎨 Improving logo with AI...");
+    }
   } catch (err) {
     spinner.stop(false);
     console.log(`  ⚠️  An unexpected error occurred: ${err instanceof Error ? err.message : String(err)}`);

lib/ai.ts

@@ -11,6 +11,9 @@ export function startSpinner(text: string) {
     update(newText: string) {
       currentText = newText;
     },
+    start() {
+      // No-op for now, as update already changes the text and the timer is running
+    },
     stop(success: boolean) {
       clearInterval(timer);
       const icon = success ? "✅" : "❌";

lib/ui.ts

@@ -48,8 +48,8 @@ export async function newLogoCommand() {
   const logoPath = await generateLogo(logoPrompt, geminiApiKey, destDir);
 
   if (logoPath) {
-    console.log(blue(`\n  ✅ Logo generated and processed successfully!`));
-    console.log(`  📂 Files created in: ${bold(destDir + "/public")}`);
+    console.log(blue(`\n  ✅ All done!`));
+    console.log(`  📂 Files updated in: ${bold(destDir + "/public")}`);
   } else {
     console.log(red(`\n  ❌ Failed to generate logo.`));
   }