@exreve/exk 1.0.63 → 1.0.65

package/dist/cli/moduleMcpServer.js

@@ -7,7 +7,7 @@
  */
 import { createSdkMcpServer, tool } from '@anthropic-ai/claude-agent-sdk';
 import { z } from 'zod';
-import { executeAnalyzeImage, executeSendFile, executeBrowserQuery } from './sharedTools.js';
+import { executeAnalyzeImage, executeSendFile, executeBrowserQuery, executeGenImage } from './sharedTools.js';
 /**
  * Build the shared config from MCP server config.
  */
@@ -56,6 +56,15 @@ export function createModuleMcpServer(config) {
             country: args.country,
             mobile: args.mobile,
         }, sharedConfig)),
+        tool('gen_image', 'Generate an image from a text prompt using Flux.2-flex. The generated PNG is saved to a temp directory and the file path is returned. ' +
+            'Use this to create illustrations, icons, diagrams, or any visual content for the user. ' +
+            'The image is saved locally — use send_file to display it in chat after generation.', {
+            prompt: z.string().describe('Detailed text description of the image to generate. Be specific about style, colors, composition, etc.'),
+            filename: z.string().optional().describe('Output filename (without extension, or with .png). Defaults to gen_<timestamp>.png'),
+        }, async (args) => executeGenImage({
+            prompt: args.prompt,
+            filename: args.filename,
+        }, sharedConfig)),
     ];
     return createSdkMcpServer({
         name: 'claude-voice-modules',

package/dist/cli/sharedTools.js

@@ -257,3 +257,94 @@ export async function executeBrowserQuery(args, config) {
         };
     }
 }
+// ============ Image Generation ============
+const FLUX_MODEL = 'black-forest-labs/flux.2-flex';
+/**
+ * gen_image — generate an image using Flux.2-flex via OpenRouter.
+ * Saves the result as a PNG file in a temp directory and returns the file path.
+ */
+export async function executeGenImage(args, _config) {
+    const apiKey = getOpenrouterApiKey();
+    if (!apiKey) {
+        return { content: [{ type: 'text', text: 'Error: OPENROUTER_API_KEY not configured. Set it in ai-config.json or OPENROUTER_API_KEY env var.' }], isError: true };
+    }
+    try {
+        const res = await fetch('https://openrouter.ai/api/v1/chat/completions', {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${apiKey}`,
+                'Content-Type': 'application/json',
+                'HTTP-Referer': 'https://talk-to-code.com',
+                'X-Title': 'TalkToCode',
+            },
+            body: JSON.stringify({
+                model: FLUX_MODEL,
+                messages: [{ role: 'user', content: args.prompt }],
+                modalities: ['image'],
+            }),
+            signal: AbortSignal.timeout(120_000), // 2 min timeout for image gen
+        });
+        if (!res.ok) {
+            const errText = await res.text();
+            return { content: [{ type: 'text', text: `Error from Flux API (${res.status}): ${errText.slice(0, 500)}` }], isError: true };
+        }
+        const data = await res.json();
+        // Extract image from response
+        const images = data.choices?.[0]?.message?.images;
+        if (!images || images.length === 0) {
+            return { content: [{ type: 'text', text: 'Error: No image in response from Flux API.' }], isError: true };
+        }
+        const dataUrl = images[0]?.image_url?.url || images[0];
+        if (!dataUrl || !dataUrl.startsWith('data:')) {
+            return { content: [{ type: 'text', text: `Error: Unexpected image format from Flux API.` }], isError: true };
+        }
+        // Decode base64
+        const base64 = dataUrl.split(',')[1];
+        if (!base64) {
+            return { content: [{ type: 'text', text: 'Error: Empty image data from Flux API.' }], isError: true };
+        }
+        const rawBuf = Buffer.from(base64, 'base64');
+        const rawKb = (rawBuf.length / 1024).toFixed(0);
+        // Compress to WebP with sharp (high quality, same dimensions)
+        let finalBuf;
+        let ext = 'webp';
+        try {
+            const sharp = (await import('sharp')).default;
+            finalBuf = await sharp(rawBuf)
+                .webp({ quality: 90, effort: 4 })
+                .toBuffer();
+            // If compression didn't help (rare), keep original PNG
+            if (finalBuf.length >= rawBuf.length) {
+                finalBuf = rawBuf;
+                ext = 'png';
+            }
+        }
+        catch {
+            // sharp not available — save raw PNG
+            finalBuf = rawBuf;
+            ext = 'png';
+        }
+        const tmpDir = path.join(os.tmpdir(), 'talk-to-code', 'gen-images');
+        fs.mkdirSync(tmpDir, { recursive: true });
+        const safeName = (args.filename || `gen_${Date.now()}`).replace(/[^a-zA-Z0-9._-]/g, '_');
+        const baseName = safeName.replace(/\.(png|webp|jpg|jpeg)$/i, '');
+        const fileName = `${baseName}.${ext}`;
+        const filePath = path.join(tmpDir, fileName);
+        fs.writeFileSync(filePath, finalBuf);
+        const sizeKb = (finalBuf.length / 1024).toFixed(0);
+        const saved = ext === 'webp' ? ` (${rawKb} KB PNG → ${sizeKb} KB WebP)` : '';
+        return {
+            content: [{
+                    type: 'text',
+                    text: `Image generated successfully.\n\nFile: ${filePath}\nSize: ${sizeKb} KB${saved}\nPrompt: ${args.prompt}`,
+                }],
+            details: { filePath, sizeKb, prompt: args.prompt },
+        };
+    }
+    catch (error) {
+        if (error.name === 'TimeoutError') {
+            return { content: [{ type: 'text', text: 'Image generation timed out after 2 minutes. Try a simpler prompt.' }], isError: true };
+        }
+        return { content: [{ type: 'text', text: `Error generating image: ${error.message}` }], isError: true };
+    }
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@exreve/exk",
-  "version": "1.0.63",
+  "version": "1.0.65",
   "description": "exk - Control Claude CLI with voice and programmable interfaces",
   "type": "module",
   "bin": {