Try refactor

Author: crunchwrap89

BACKLOG.md

@@ -1,9 +1,7 @@
 # Backlog
 
 ---
-## Feature name
-Verify if there are any issues.
+## Completed
+- Refactor Logo Generation api in cli (2026-01-15)
 
-### 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.
 ---

CHANGELOG.md

@@ -14,6 +14,11 @@
 
 ### Features
 
+- Refactored AI logo generation to use the latest `gemini-3-pro-image-preview` model via the standard `@google/genai` SDK.
+- Added support for `GEMINI_API_KEY` environment variable to skip manual API key entry.
+- Improved image processing and response handling for AI-generated logos.
+- Enhanced error reporting for Gemini API issues.
+
 ### Fixes
 
 ## [0.1.25] - 2026-01-15

init.ts

@@ -88,8 +88,8 @@ export async function initCommand() {
     "",
   );
 
-  let geminiApiKey = "";
-  if (logoPrompt) {
+  let geminiApiKey = Deno.env.get("GEMINI_API_KEY") || "";
+  if (logoPrompt && !geminiApiKey) {
     geminiApiKey = promptInput("🔑 Enter your Google Gemini API key", "", {
       required: true,
     });

lib/ai.ts

@@ -18,11 +18,12 @@ export async function generateLogo(
   try {
     const ai = new GoogleGenAI({ apiKey });
 
-    // Initial message to start the chat
     const chat = ai.chats.create({
       model: "gemini-3-pro-image-preview",
       config: {
-        responseModalities: ["TEXT", "IMAGE"],
+        imageConfig: {
+          aspectRatio: "1:1",
+        },
       },
     });
 
@@ -32,9 +33,13 @@ export async function generateLogo(
       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;
+      const candidates = response.candidates;
+      if (candidates && candidates.length > 0) {
+        for (const part of candidates[0]?.content?.parts ?? []) {
+          if (part.inlineData && (part.inlineData.mimeType === "image/png" || part.inlineData.mimeType === "image/jpeg")) {
+            base64Image = part.inlineData.data;
+            break;
+          }
         }
       }
 
@@ -69,7 +74,11 @@ export async function generateLogo(
     }
   } catch (err) {
     spinner.stop(false);
-    console.log(`  ⚠️  An unexpected error occurred: ${err instanceof Error ? err.message : String(err)}`);
+    if (err instanceof Error && err.name === "ApiError") {
+      console.log(`  ⚠️  Gemini API Error: ${err.message} (Status: ${(err as any).status})`);
+    } else {
+      console.log(`  ⚠️  An unexpected error occurred: ${err instanceof Error ? err.message : String(err)}`);
+    }
     return null;
   }
 }

new-logo.ts

@@ -41,7 +41,8 @@ export async function newLogoCommand() {
     { required: true }
   );
 
-  const geminiApiKey = promptInput("🔑 Enter your Google Gemini API key", "", {
+  const envApiKey = Deno.env.get("GEMINI_API_KEY");
+  const geminiApiKey = envApiKey || promptInput("🔑 Enter your Google Gemini API key", "", {
     required: true,
   });