genmix 1.0.0 → 1.0.2

package/README.md

@@ -45,7 +45,7 @@ const result = await generator.generate(
 );
 
 // Save images
-const savedPaths = generator.save('./output');
+const savedPaths = await generator.save({ directory: './output' });
 console.log('Images saved:', savedPaths);
 ```
 
@@ -62,7 +62,7 @@ const result = await generator.generate(
   }
 );
 
-const savedPaths = generator.save('./output');
+const savedPaths = await generator.save({ directory: './output' });
 ```
 
 ### Using URLs as Reference
@@ -76,7 +76,7 @@ const result = await generator.generate(
   }
 );
 
-generator.save('./output');
+generator.save({ directory: './output' });
 ```
 
 ## Configuration Options
@@ -106,6 +106,46 @@ await generator.generate(prompt, options)
 | `options.quality`        | string        | Quality: '1K', '2K', '4K'                   | -       |
 | `options.aspectRatio`    | string        | Aspect ratio: '1:1', '16:9', '4:3', etc.    | -       |
 
+### save() Method
+
+```javascript
+await generator.save(options)
+```
+
+**Parameters:**
+
+| Option              | Type   | Description                                      | Default |
+| ------------------- | ------ | ------------------------------------------------ | ------- |
+| `options.directory` | string | Target directory path to save images             | `'.'`   |
+| `options.filename`  | string | Custom filename (without extension). If not provided, uses hash-based filename | -       |
+| `options.extension` | string | File format: 'jpg', 'png', 'webp', 'avif', 'tiff' | `'jpg'` |
+
+**Examples:**
+
+```javascript
+// Save to specific directory with auto-generated filename (jpg by default)
+await generator.save({ directory: './output' });
+
+// Save as PNG
+await generator.save({ directory: './output', extension: 'png' });
+
+// Save to specific directory with custom filename
+await generator.save({ directory: './output', filename: 'my-image' });
+
+// Save as WebP with custom filename
+await generator.save({ directory: './output', filename: 'my-image', extension: 'webp' });
+
+// Save to current directory with custom filename
+await generator.save({ filename: 'my-image' });
+
+// Save to current directory with auto-generated filename (jpg)
+await generator.save();
+```
+
+**Note:** 
+- When multiple images are generated and a custom filename is provided, they will be saved as `filename_0.jpg`, `filename_1.jpg`, etc.
+- The method uses Sharp for image conversion, supporting high-quality format conversion
+
 ## Advanced Examples
 
 ### Style Transfer
@@ -179,7 +219,7 @@ try {
   const result = await generator.generate(prompt, options);
   
   if (result.images && result.images.length > 0) {
-    const paths = generator.save('./output');
+    const paths = generator.save({ directory: './output' });
     console.log('Success!', paths);
   } else {
     console.log('No images generated');

package/demo/example-translate.js

@@ -17,13 +17,25 @@ async function main() {
                 'Translate this image to Portuguese',
                 {
                     referenceImage: localImagePath,
-                    quality: '2K',
+                    quality: '1K',
                     numberOfImages: 1
                 }
             );
 
             if (result1.images && result1.images.length > 0) {
-                const saved = generator.save(__dirname);
+                // Create pt directory if it doesn't exist
+                const ptDir = path.join(__dirname, 'pt');
+                if (!fs.existsSync(ptDir)) {
+                    fs.mkdirSync(ptDir, { recursive: true });
+                }
+                
+                // Use the same filename as the original
+                const originalName = path.basename(localImagePath, path.extname(localImagePath));
+                
+                const saved = await generator.save({ 
+                    directory: ptDir,
+                    filename: originalName
+                });
                 console.log('✅ Modified image saved at:', saved[0], '\n');
             } else if (result1.text) {
                 console.log('📝 Result:', result1.text, '\n');
@@ -31,26 +43,26 @@ async function main() {
         }
 
 
-        // === Option 3: Multiple variations of the same image ===
-        if (fs.existsSync(localImagePath)) {
-            console.log('🎨 Generating multiple variations...\n');
+        // // === Option 3: Multiple variations of the same image ===
+        // if (fs.existsSync(localImagePath)) {
+        //     console.log('🎨 Generating multiple variations...\n');
 
-            const result3 = await generator.generate(
-                'Add a vintage film look with grain and vignette effect',
-                {
-                    referenceImage: localImagePath,
-                    quality: '1K',
-                    numberOfImages: 3
-                }
-            );
+        //     const result3 = await generator.generate(
+        //         'Add a vintage film look with grain and vignette effect',
+        //         {
+        //             referenceImage: localImagePath,
+        //             quality: '1K',
+        //             numberOfImages: 3
+        //         }
+        //     );
 
-            if (result3.images && result3.images.length > 0) {
-                const saved = generator.save(__dirname);
-                console.log(`✅ ${saved.length} variations saved:`);
-                saved.forEach(p => console.log(`   - ${p}`));
-                console.log();
-            }
-        }
+        //     if (result3.images && result3.images.length > 0) {
+        //         const saved = await generator.save({ directory: __dirname });
+        //         console.log(`✅ ${saved.length} variations saved:`);
+        //         saved.forEach(p => console.log(`   - ${p}`));
+        //         console.log();
+        //     }
+        // }
 
         console.log('🎉 All examples completed successfully!\n');
 

package/demo/example.js

@@ -25,7 +25,7 @@ async function exampleBasicGeneration() {
     if (result.images && result.images.length > 0) {
         console.log(`Found ${result.images.length} images.`);
 
-        const savedPaths = generator.save(__dirname);
+        const savedPaths = await generator.save({ directory: __dirname });
         savedPaths.forEach(p => console.log(`Saved image to ${p}`));
     } else {
         console.log('No images generated.');

package/generators/BaseGenerator.js

@@ -1,6 +1,7 @@
 const fs = require('fs');
 const path = require('path');
 const hashFactory = require('hash-factory');
+const sharp = require('sharp');
 
 class BaseGenerator {
     constructor(config = {}) {
@@ -20,12 +21,14 @@ class BaseGenerator {
     }
 
     /**
-     * Saves generated images to the specified directory using hashed filenames.
-     * @param {string} directory - Target directory path.
-     * @returns {string[]} Array of saved file paths.
+     * Saves generated images to the specified directory using hashed filenames or custom filename.
+     * @param {Object} [options={}] - Save options
+     * @param {string} [options.directory='.'] - Target directory path. Defaults to current directory.
+     * @param {string} [options.filename] - Optional custom filename (without extension). If not provided, uses hashed filenames.
+     * @param {string} [options.extension='jpg'] - File extension: 'jpg', 'png', 'webp', 'avif', 'tiff'. Defaults to 'jpg'.
+     * @returns {Promise<string[]>} Promise that resolves to array of saved file paths.
      */
-    save(directory) {
-        const extension = 'png';
+    async save({directory = '.', filename = null, extension = 'jpg'} = {}) {
         const targetImages = this.lastGeneration?.images;
         const targetPrompt = this.lastGeneration?.prompt;
 
@@ -34,7 +37,7 @@ class BaseGenerator {
             return [];
         }
 
-        if (!targetPrompt) {
+        if (!targetPrompt && !filename) {
             console.warn('No prompt available for hash generation.');
         }
 
@@ -43,18 +46,54 @@ class BaseGenerator {
             fs.mkdirSync(directory, { recursive: true });
         }
 
+        // Normalize extension
+        extension = extension.toLowerCase().replace(/^\./, '');
+        
+        // Validate extension
+        const supportedFormats = ['jpg', 'jpeg', 'png', 'webp', 'avif', 'tiff', 'tif'];
+        if (!supportedFormats.includes(extension)) {
+            throw new Error(`Unsupported extension: ${extension}. Supported formats: ${supportedFormats.join(', ')}`);
+        }
+
+        // Normalize jpeg/jpg
+        const sharpFormat = extension === 'jpeg' ? 'jpg' : extension;
+        const fileExtension = extension === 'jpeg' ? 'jpg' : extension;
+
         const savedPaths = [];
-        targetImages.forEach((imgData, index) => {
+        
+        for (let index = 0; index < targetImages.length; index++) {
+            const imgData = targetImages[index];
             const base64Data = imgData.replace(/^data:image\/\w+;base64,/, "");
             const buffer = Buffer.from(base64Data, 'base64');
 
-            const hash = this.generateHash(targetPrompt + '_' + index);
-            const fileName = `${hash}.${extension}`;
+            let fileName;
+            if (filename) {
+                // Use custom filename, add index if multiple images
+                if (targetImages.length > 1) {
+                    fileName = `${filename}_${index}.${fileExtension}`;
+                } else {
+                    fileName = `${filename}.${fileExtension}`;
+                }
+            } else {
+                // Use hash-based filename
+                const hash = this.generateHash(targetPrompt + '_' + index);
+                fileName = `${hash}.${fileExtension}`;
+            }
+            
             const outputPath = path.join(directory, fileName);
 
-            fs.writeFileSync(outputPath, buffer);
-            savedPaths.push(outputPath);
-        });
+            // Convert image format using sharp
+            try {
+                await sharp(buffer)
+                    .toFormat(sharpFormat)
+                    .toFile(outputPath);
+                
+                savedPaths.push(outputPath);
+            } catch (error) {
+                console.error(`Error saving image ${fileName}:`, error.message);
+                throw error;
+            }
+        }
 
         return savedPaths;
     }

package/package.json

@@ -1,9 +1,23 @@
 {
   "name": "genmix",
-  "version": "1.0.0",
-  "description": "",
-  "license": "ISC",
-  "author": "",
+  "version": "1.0.2",
+  "description": "AI-powered image generator using Google Gemini API. Supports image generation from text prompts and image modification with reference images.",
+  "license": "MIT",
+  "author": "Martin Clasen",
+  "keywords": [
+    "nano",
+    "banana",
+    "pro",
+    "image-generator",
+    "google-gemini",
+    "gemini-api",
+    "text-to-image",
+    "image-modification",
+    "style-transfer",
+    "generative-ai",
+    "genmix",
+    "clasen"
+  ],
   "type": "commonjs",
   "main": "index.js",
   "scripts": {
@@ -11,6 +25,7 @@
   },
   "dependencies": {
     "axios": "^1.13.2",
-    "hash-factory": "^1.1.2"
+    "hash-factory": "^1.1.2",
+    "sharp": "^0.33.5"
   }
 }

package/demo/.env

@@ -1 +0,0 @@
-GEMINI_API_KEY="AIzaSyDMy_iCPVzMHF3sSrbCYr2K9PO9lHkCPsg"

package/demo/genmix/generators/BaseGenerator.js

@@ -1,58 +0,0 @@
-const fs = require('fs');
-const path = require('path');
-const hashFactory = require('hash-factory');
-
-class BaseGenerator {
-  constructor(config = {}) {
-    this.config = config;
-  }
-
-  /**
-   * Generates a hash from the text prompt.
-   * @param {string} prompt 
-   * @param {Object} options 
-   * @returns {string}
-   */
-  generateHash(prompt, options = { length: 10 }) {
-    return hashFactory(prompt, options);
-  }
-
-  /**
-   * Saves generated images to the specified directory using hashed filenames.
-   * @param {string[]} images - Array of base64 image strings.
-   * @param {string} directory - Target directory path.
-   * @param {string} prompt - The prompt used for generation (to create hash).
-   * @param {string} [extension='png'] - File extension.
-   * @returns {string[]} Array of saved file paths.
-   */
-  saveImages(images, directory, prompt, extension = 'png') {
-    if (!images || !Array.isArray(images) || images.length === 0) {
-      return [];
-    }
-
-    // Ensure directory exists
-    if (!fs.existsSync(directory)) {
-      fs.mkdirSync(directory, { recursive: true });
-    }
-
-    const savedPaths = [];
-    const hash = this.generateHash(prompt);
-
-    images.forEach((imgData, index) => {
-      const base64Data = imgData.replace(/^data:image\/\w+;base64,/, "");
-      const buffer = Buffer.from(base64Data, 'base64');
-      
-      // Using format: hash_index.ext
-      const fileName = `${hash}_${index + 1}.${extension}`;
-      const outputPath = path.join(directory, fileName);
-      
-      fs.writeFileSync(outputPath, buffer);
-      savedPaths.push(outputPath);
-    });
-
-    return savedPaths;
-  }
-}
-
-module.exports = BaseGenerator;
-