Function handler

plugins/imagine-diffusion.ts:44–76

(sock: any, message: any, args: string[], context: BotContext)

    usage: '.diffusion <prompt>',

    async handler(sock: any, message: any, args: string[], context: BotContext) {
        const chatId = context.chatId || message.key.remoteJid;
        const imagePrompt = args.join(' ').trim();

        if (!imagePrompt) {
            return sock.sendMessage(
                chatId,
                { text: '🎨 *AI Image Generator*\n\nUsage: `.diffusion <prompt>`\nExample: `.diffusion a beautiful sunset over mountains`' },
                { quoted: message }
            );
        }

        await sock.sendMessage(chatId, { react: { text: '🎨', key: message.key } });
        await sock.sendMessage(chatId, { text: '🎨 Generating your image... Please wait.' }, { quoted: message });

        try {
            const enhanced = enhancePrompt(imagePrompt);
            const imageBuffer = await generateImage(enhanced);

            await sock.sendMessage(chatId, {
                image: imageBuffer,
                caption: `🎨 *Generated Image*\n📝 Prompt: _${imagePrompt}_`
            }, { quoted: message });

        } catch (error: any) {
            console.error('Imagine error:', error.message);
            await sock.sendMessage(
                chatId,
                { text: '❌ Failed to generate image. Please try again later.' },
                { quoted: message }
            );
        }
    }
};