Add ai logo generation.

Author: crunchwrap89

BACKLOG.md

@@ -0,0 +1,9 @@
+# Backlog
+
+---
+## Feature name
+Generate a logo with AI.
+
+### Description
+Ask the user to provide a prompt that will be used for logo generation with AI as a final step before the last step in the cli init process.
+---

CHANGELOG.md

@@ -4,8 +4,12 @@
 
 ### Features
 
+- Implement AI logo generation using Google Gemini.
+
 ### Fixes
 
+- Fix error handling in `lib/ai.ts` to avoid local exception catching and improve reliability.
+- 
 ## [0.1.18] - 2026-01-13
 
 ### Miscellaneous

init.ts

@@ -14,6 +14,7 @@ import { promptInput, promptSelect } from "./lib/prompt.ts";
 import { degitRepoToDir } from "./lib/degit.ts";
 import { replacePlaceholdersInRepo } from "./lib/replace.ts";
 import { isValidProjectName, slugifyProjectName } from "./lib/validate.ts";
+import { generateLogo } from "./lib/ai.ts";
 
 export async function initCommand() {
   console.log(magenta(bold(`
@@ -80,6 +81,20 @@ export async function initCommand() {
   );
   const phone = promptInput("📞 Phone number (optional)", "+46 7182387123");
 
+  // Logo generation prompt
+  console.log(cyan(bold("\n🎨 AI Logo Generation")));
+  const logoPrompt = promptInput(
+    "✨ Enter a prompt for your logo (optional, leave empty to skip)",
+    "",
+  );
+
+  let geminiApiKey = "";
+  if (logoPrompt) {
+    geminiApiKey = promptInput("🔑 Enter your Google Gemini API key", "", {
+      required: true,
+    });
+  }
+
   const meta = {
     projectname: projectname.trim(),
     shortname: shortname.trim(),
@@ -369,6 +384,14 @@ export async function initCommand() {
       }
     }
 
+    // ------------------------
+    // Step 4: AI Logo Generation (if prompt and key provided)
+    // ------------------------
+    if (logoPrompt && geminiApiKey) {
+      console.log("");
+      await generateLogo(logoPrompt, geminiApiKey, destDir);
+    }
+
     console.log(
       green(bold(`\n🎉 Project successfully generated in: ${meta.projectname}`)),
     );

lib/ai.ts

@@ -0,0 +1,68 @@
+import { startSpinner } from "./ui.ts";
+import { join } from "@std/path";
+import { ensureDir } from "@std/fs";
+
+/**
+ * Generates a logo using Google's Gemini AI (Imagen model)
+ */
+export async function generateLogo(
+  prompt: string,
+  apiKey: string,
+  destDir: string,
+): Promise<string | null> {
+  const spinner = startSpinner("🎨 Generating logo with AI...");
+
+  try {
+    // Note: Using imagen-3.0-generate-001 model via Gemini API
+    const url = `https://generativelanguage.googleapis.com/v1beta/models/imagen-3.0-generate-001:predict?key=${apiKey}`;
+
+    const response = await fetch(url, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+      },
+      body: JSON.stringify({
+        instances: [
+          {
+            prompt: prompt,
+          },
+        ],
+        parameters: {
+          sampleCount: 1,
+          aspectRatio: "1:1",
+          outputMimeType: "image/png",
+        },
+      }),
+    });
+
+    if (!response.ok) {
+      const errorData = await response.json().catch(() => ({}));
+      const errorMessage = errorData.error?.message || `API request failed with status ${response.status}`;
+      spinner.stop(false);
+      console.log(`  ⚠️  Failed to generate logo: ${errorMessage}`);
+      return null;
+    }
+
+    const data = await response.json();
+    const base64Image = data.predictions?.[0]?.bytesBase64Encoded;
+
+    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);
+
+    spinner.stop(true);
+    return logoPath;
+  } catch (err) {
+    spinner.stop(false);
+    console.log(`  ⚠️  An unexpected error occurred: ${err instanceof Error ? err.message : String(err)}`);
+    return null;
+  }
+}