vsegments 0.1.1 → 0.1.3

package/coverage/lcov.info

@@ -0,0 +1,377 @@
+TN:
+SF:src/core.js
+FN:26,(anonymous_0)
+FN:62,(anonymous_1)
+FN:80,(anonymous_2)
+FN:97,(anonymous_3)
+FN:116,(anonymous_4)
+FN:168,(anonymous_5)
+FN:227,(anonymous_6)
+FNF:7
+FNH:3
+FNDA:23,(anonymous_0)
+FNDA:0,(anonymous_1)
+FNDA:9,(anonymous_2)
+FNDA:3,(anonymous_3)
+FNDA:0,(anonymous_4)
+FNDA:0,(anonymous_5)
+FNDA:0,(anonymous_6)
+DA:5,2
+DA:6,2
+DA:7,2
+DA:8,2
+DA:9,2
+DA:15,2
+DA:27,23
+DA:29,23
+DA:30,1
+DA:35,22
+DA:36,22
+DA:37,22
+DA:40,22
+DA:43,22
+DA:49,22
+DA:63,0
+DA:64,0
+DA:65,0
+DA:67,0
+DA:81,9
+DA:82,9
+DA:89,9
+DA:98,3
+DA:100,3
+DA:101,1
+DA:104,3
+DA:121,0
+DA:124,0
+DA:127,0
+DA:135,0
+DA:136,0
+DA:143,0
+DA:144,0
+DA:146,0
+DA:147,0
+DA:151,0
+DA:155,0
+DA:157,0
+DA:172,0
+DA:175,0
+DA:178,0
+DA:179,0
+DA:180,0
+DA:183,0
+DA:190,0
+DA:191,0
+DA:198,0
+DA:199,0
+DA:201,0
+DA:202,0
+DA:206,0
+DA:210,0
+DA:211,0
+DA:213,0
+DA:233,0
+DA:236,0
+DA:239,0
+DA:240,0
+DA:246,0
+DA:253,0
+DA:254,0
+DA:257,0
+DA:261,2
+LF:63
+LH:23
+BRDA:26,0,0,2
+BRDA:27,1,0,23
+BRDA:27,1,1,2
+BRDA:29,2,0,1
+BRDA:29,2,1,22
+BRDA:35,3,0,22
+BRDA:35,3,1,21
+BRDA:36,4,0,2
+BRDA:36,4,1,20
+BRDA:37,5,0,22
+BRDA:37,5,1,17
+BRDA:89,6,0,9
+BRDA:89,6,1,2
+BRDA:100,7,0,1
+BRDA:100,7,1,2
+BRDA:116,8,0,0
+BRDA:118,9,0,0
+BRDA:119,10,0,0
+BRDA:120,11,0,0
+BRDA:146,12,0,0
+BRDA:146,12,1,0
+BRDA:168,13,0,0
+BRDA:170,14,0,0
+BRDA:171,15,0,0
+BRDA:201,16,0,0
+BRDA:201,16,1,0
+BRDA:227,17,0,0
+BRDA:229,18,0,0
+BRDA:230,19,0,0
+BRDA:231,20,0,0
+BRDA:232,21,0,0
+BRDA:239,22,0,0
+BRDA:239,22,1,0
+BRDA:253,23,0,0
+BRDA:253,23,1,0
+BRF:35
+BRH:15
+end_of_record
+TN:
+SF:src/index.js
+FNF:0
+FNH:0
+DA:12,1
+DA:13,1
+DA:15,1
+DA:16,1
+DA:17,1
+DA:18,1
+DA:19,1
+DA:20,1
+LF:8
+LH:8
+BRF:0
+BRH:0
+end_of_record
+TN:
+SF:src/models.js
+FN:14,(anonymous_0)
+FN:28,(anonymous_1)
+FN:41,(anonymous_2)
+FN:63,(anonymous_3)
+FN:80,(anonymous_4)
+FN:90,(anonymous_5)
+FNF:6
+FNH:6
+FNDA:20,(anonymous_0)
+FNDA:3,(anonymous_1)
+FNDA:8,(anonymous_2)
+FNDA:4,(anonymous_3)
+FNDA:6,(anonymous_4)
+FNDA:3,(anonymous_5)
+DA:15,20
+DA:16,20
+DA:17,20
+DA:18,20
+DA:19,20
+DA:29,3
+DA:30,3
+DA:31,3
+DA:32,3
+DA:33,3
+DA:42,8
+DA:43,8
+DA:64,4
+DA:65,4
+DA:66,4
+DA:67,4
+DA:68,4
+DA:69,4
+DA:81,6
+DA:82,6
+DA:83,6
+DA:91,3
+DA:95,4
+LF:23
+LH:23
+BRDA:80,0,0,4
+BRDA:80,1,0,5
+BRF:2
+BRH:2
+end_of_record
+TN:
+SF:src/utils.js
+FN:13,parseJson
+FN:30,parseBoundingBoxes
+FN:51,parseSegmentationMasks
+FNF:3
+FNH:3
+FNDA:12,parseJson
+FNDA:5,parseBoundingBoxes
+FNDA:2,parseSegmentationMasks
+DA:5,3
+DA:6,3
+DA:14,12
+DA:15,12
+DA:16,13
+DA:17,4
+DA:18,4
+DA:19,4
+DA:22,12
+DA:31,5
+DA:32,5
+DA:34,5
+DA:35,5
+DA:36,8
+DA:37,7
+DA:41,5
+DA:52,2
+DA:53,2
+DA:55,2
+DA:57,2
+DA:58,3
+DA:60,0
+DA:61,0
+DA:64,0
+DA:65,0
+DA:66,0
+DA:67,0
+DA:69,0
+DA:70,0
+DA:73,0
+DA:74,0
+DA:77,0
+DA:78,0
+DA:79,0
+DA:80,0
+DA:83,0
+DA:84,0
+DA:86,0
+DA:87,0
+DA:91,0
+DA:93,0
+DA:94,0
+DA:95,0
+DA:96,0
+DA:97,0
+DA:99,0
+DA:100,0
+DA:101,0
+DA:106,0
+DA:111,2
+DA:114,3
+LF:51
+LH:23
+BRDA:16,0,0,4
+BRDA:16,0,1,9
+BRDA:36,1,0,7
+BRDA:36,1,1,1
+BRDA:58,2,0,3
+BRDA:58,2,1,0
+BRDA:58,3,0,3
+BRDA:58,3,1,2
+BRDA:99,4,0,0
+BRDA:99,4,1,0
+BRDA:99,5,0,0
+BRDA:99,5,1,0
+BRF:12
+BRH:7
+end_of_record
+TN:
+SF:src/visualize.js
+FN:25,plotBoundingBoxes
+FN:39,(anonymous_1)
+FN:66,overlayMask
+FN:100,plotSegmentationMasks
+FN:113,(anonymous_4)
+FN:122,(anonymous_5)
+FN:145,loadImageToCanvas
+FN:171,saveCanvas
+FNF:8
+FNH:0
+FNDA:0,plotBoundingBoxes
+FNDA:0,(anonymous_1)
+FNDA:0,overlayMask
+FNDA:0,plotSegmentationMasks
+FNDA:0,(anonymous_4)
+FNDA:0,(anonymous_5)
+FNDA:0,loadImageToCanvas
+FNDA:0,saveCanvas
+DA:5,2
+DA:6,2
+DA:7,2
+DA:10,2
+DA:30,0
+DA:32,0
+DA:33,0
+DA:34,0
+DA:36,0
+DA:37,0
+DA:39,0
+DA:40,0
+DA:41,0
+DA:44,0
+DA:45,0
+DA:48,0
+DA:49,0
+DA:50,0
+DA:54,0
+DA:67,0
+DA:68,0
+DA:69,0
+DA:72,0
+DA:73,0
+DA:74,0
+DA:75,0
+DA:78,0
+DA:79,0
+DA:80,0
+DA:81,0
+DA:82,0
+DA:84,0
+DA:85,0
+DA:86,0
+DA:90,0
+DA:106,0
+DA:108,0
+DA:109,0
+DA:110,0
+DA:113,0
+DA:114,0
+DA:115,0
+DA:119,0
+DA:120,0
+DA:122,0
+DA:123,0
+DA:126,0
+DA:127,0
+DA:130,0
+DA:131,0
+DA:132,0
+DA:136,0
+DA:146,0
+DA:149,0
+DA:150,0
+DA:152,0
+DA:153,0
+DA:154,0
+DA:155,0
+DA:159,0
+DA:160,0
+DA:161,0
+DA:163,0
+DA:172,0
+DA:173,0
+DA:176,2
+LF:66
+LH:5
+BRDA:25,0,0,0
+BRDA:27,1,0,0
+BRDA:28,2,0,0
+BRDA:29,3,0,0
+BRDA:48,4,0,0
+BRDA:48,4,1,0
+BRDA:48,5,0,0
+BRDA:48,5,1,0
+BRDA:80,6,0,0
+BRDA:80,6,1,0
+BRDA:100,7,0,0
+BRDA:102,8,0,0
+BRDA:103,9,0,0
+BRDA:104,10,0,0
+BRDA:105,11,0,0
+BRDA:130,12,0,0
+BRDA:130,12,1,0
+BRDA:130,13,0,0
+BRDA:130,13,1,0
+BRDA:145,14,0,0
+BRDA:152,15,0,0
+BRDA:152,15,1,0
+BRDA:152,16,0,0
+BRDA:152,16,1,0
+BRF:24
+BRH:0
+end_of_record

package/jest.config.js

@@ -0,0 +1,15 @@
+module.exports = {
+  testEnvironment: 'node',
+  testMatch: ['**/test/**/*.test.js'],
+  coverageDirectory: 'coverage',
+  collectCoverageFrom: [
+    'src/**/*.js',
+    '!src/index.d.ts',
+    '!src/**/*.test.js'
+  ],
+  coveragePathIgnorePatterns: [
+    '/node_modules/'
+  ],
+  verbose: true,
+  testTimeout: 10000
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vsegments",
-  "version": "0.1.1",
+  "version": "0.1.3",
   "description": "Visual segmentation and bounding box detection using Google Gemini AI",
   "main": "src/index.js",
   "types": "src/index.d.ts",
@@ -8,7 +8,10 @@
     "vsegments": "./bin/cli.js"
   },
   "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1",
+    "test": "jest",
+    "test:watch": "jest --watch",
+    "test:coverage": "jest --coverage",
+    "pre-deploy": "node scripts/pre-deploy.js",
     "lint": "eslint src/**/*.js",
     "format": "prettier --write \"src/**/*.js\""
   },
@@ -40,5 +43,9 @@
     "@google/generative-ai": "^0.21.0",
     "canvas": "^2.11.2",
     "commander": "^12.0.0"
+  },
+  "devDependencies": {
+    "@types/jest": "^30.0.0",
+    "jest": "^30.2.0"
   }
 }

package/scripts/pre-deploy.js

@@ -0,0 +1,105 @@
+#!/usr/bin/env node
+
+/**
+ * Pre-deployment verification script
+ * Run this before publishing to npm to ensure everything is ready
+ */
+
+const { execSync } = require('child_process');
+const fs = require('fs');
+const path = require('path');
+
+console.log('🔍 Running pre-deployment checks...\n');
+
+let hasErrors = false;
+
+// Check 1: Run tests
+console.log('1️⃣  Running tests...');
+try {
+  execSync('npm test', { stdio: 'inherit' });
+  console.log('✅ All tests passed\n');
+} catch (error) {
+  console.error('❌ Tests failed\n');
+  hasErrors = true;
+}
+
+// Check 2: Check for required files
+console.log('2️⃣  Checking required files...');
+const requiredFiles = [
+  'package.json',
+  'README.md',
+  'LICENSE',
+  'src/index.js',
+  'src/core.js',
+  'src/models.js',
+  'src/utils.js',
+  'bin/cli.js'
+];
+
+let allFilesExist = true;
+for (const file of requiredFiles) {
+  if (!fs.existsSync(path.join(__dirname, '..', file))) {
+    console.error(`❌ Missing required file: ${file}`);
+    allFilesExist = false;
+    hasErrors = true;
+  }
+}
+if (allFilesExist) {
+  console.log('✅ All required files present\n');
+}
+
+// Check 3: Verify package.json
+console.log('3️⃣  Verifying package.json...');
+const pkg = require('../package.json');
+const requiredFields = ['name', 'version', 'description', 'main', 'author', 'license'];
+let allFieldsPresent = true;
+
+for (const field of requiredFields) {
+  if (!pkg[field]) {
+    console.error(`❌ Missing required field in package.json: ${field}`);
+    allFieldsPresent = false;
+    hasErrors = true;
+  }
+}
+if (allFieldsPresent) {
+  console.log('✅ package.json is valid\n');
+}
+
+// Check 4: Verify dependencies
+console.log('4️⃣  Checking dependencies...');
+const requiredDeps = ['@google/generative-ai', 'canvas', 'commander'];
+let allDepsPresent = true;
+
+for (const dep of requiredDeps) {
+  if (!pkg.dependencies || !pkg.dependencies[dep]) {
+    console.error(`❌ Missing required dependency: ${dep}`);
+    allDepsPresent = false;
+    hasErrors = true;
+  }
+}
+if (allDepsPresent) {
+  console.log('✅ All dependencies present\n');
+}
+
+// Check 5: Test coverage
+console.log('5️⃣  Running test coverage...');
+try {
+  execSync('npm run test:coverage -- --silent', { stdio: 'pipe' });
+  console.log('✅ Coverage report generated\n');
+} catch (error) {
+  console.error('❌ Coverage check failed\n');
+  hasErrors = true;
+}
+
+// Final result
+console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
+if (hasErrors) {
+  console.log('❌ Pre-deployment checks failed!');
+  console.log('Please fix the issues above before deploying.\n');
+  process.exit(1);
+} else {
+  console.log('✅ All pre-deployment checks passed!');
+  console.log('Your package is ready to deploy.\n');
+  console.log('To publish to npm, run: npm publish\n');
+  process.exit(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;
     
@@ -131,15 +131,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 +186,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);