vsegments 0.1.2 → 0.1.4

package/package.json

@@ -1,11 +1,11 @@
 {
   "name": "vsegments",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "description": "Visual segmentation and bounding box detection using Google Gemini AI",
   "main": "src/index.js",
   "types": "src/index.d.ts",
   "bin": {
-    "vsegments": "./bin/cli.js"
+    "vsegments": "bin/cli.js"
   },
   "scripts": {
     "test": "jest",
@@ -30,7 +30,7 @@
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "git@github.com:nxtphaseai/node_vsegments.git"
+    "url": "git+ssh://git@github.com/nxtphaseai/node_vsegments.git"
   },
   "homepage": "https://github.com/nxtphaseai/node_vsegments#readme",
   "bugs": {
@@ -42,7 +42,8 @@
   "dependencies": {
     "@google/generative-ai": "^0.21.0",
     "canvas": "^2.11.2",
-    "commander": "^12.0.0"
+    "commander": "^12.0.0",
+    "sharp": "^0.33.0"
   },
   "devDependencies": {
     "@types/jest": "^30.0.0",

package/src/core.js

@@ -32,7 +32,7 @@ class VSegments {
       );
     }
     
-    this.model = options.model || 'gemini-flash-latest';
+    this.model = options.model || 'gemini-3-pro-preview';
     this.temperature = options.temperature !== undefined ? options.temperature : 0.5;
     this.maxObjects = options.maxObjects || 25;
     
@@ -60,9 +60,19 @@ If an object is present multiple times, name them according to their unique char
    * @returns {Promise<Object>} - Image data for API
    */
   async _loadImage(imagePath) {
-    const imageBuffer = await fs.readFile(imagePath);
+    let imageBuffer = await fs.readFile(imagePath);
+    let mimeType = this._getMimeType(imagePath);
+    
+    // Convert SVG to PNG for API compatibility
+    if (mimeType === 'image/svg+xml') {
+      const sharp = require('sharp');
+      imageBuffer = await sharp(imageBuffer)
+        .png()
+        .toBuffer();
+      mimeType = 'image/png';
+    }
+    
     const base64Data = imageBuffer.toString('base64');
-    const mimeType = this._getMimeType(imagePath);
     
     return {
       inlineData: {
@@ -84,7 +94,8 @@ If an object is present multiple times, name them according to their unique char
       'jpeg': 'image/jpeg',
       'png': 'image/png',
       'gif': 'image/gif',
-      'webp': 'image/webp'
+      'webp': 'image/webp',
+      'svg': 'image/svg+xml'
     };
     return mimeTypes[ext] || 'image/jpeg';
   }
@@ -131,15 +142,25 @@ If an object is present multiple times, name them according to their unique char
     });
     
     // Generate content
-    const result = await model.generateContent({
-      contents: [{ role: 'user', parts: [{ text: prompt }, image] }],
-      generationConfig: {
-        temperature: this.temperature,
+    let result, response, text;
+    try {
+      result = await model.generateContent({
+        contents: [{ role: 'user', parts: [{ text: prompt }, image] }],
+        generationConfig: {
+          temperature: this.temperature,
+        }
+      });
+      
+      response = result.response;
+      text = response.text();
+    } catch (error) {
+      if (error.status === 500) {
+        throw new Error(
+          `Google Gemini API error (500): This may be a temporary issue. Try again later or verify your API key and image. Original error: ${error.message}`
+        );
       }
-    });
-    
-    const response = result.response;
-    const text = response.text();
+      throw error;
+    }
     
     // Parse response
     const boxes = parseBoundingBoxes(text);
@@ -176,15 +197,25 @@ If an object is present multiple times, name them according to their unique char
     });
     
     // Generate content
-    const result = await model.generateContent({
-      contents: [{ role: 'user', parts: [{ text: prompt }, image] }],
-      generationConfig: {
-        temperature: this.temperature,
+    let result, response, text;
+    try {
+      result = await model.generateContent({
+        contents: [{ role: 'user', parts: [{ text: prompt }, image] }],
+        generationConfig: {
+          temperature: this.temperature,
+        }
+      });
+      
+      response = result.response;
+      text = response.text();
+    } catch (error) {
+      if (error.status === 500) {
+        throw new Error(
+          `Google Gemini API error (500): This may be a temporary issue. Try again later or verify your API key and image. Original error: ${error.message}`
+        );
       }
-    });
-    
-    const response = result.response;
-    const text = response.text();
+      throw error;
+    }
     
     // Parse response
     const boxes = parseBoundingBoxes(text);