@mixio-pro/kalaasetu-mcp 1.0.11 → 1.0.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mixio-pro/kalaasetu-mcp",
-  "version": "1.0.11",
+  "version": "1.0.13",
   "description": "A powerful Model Context Protocol server providing AI tools for content generation and analysis",
   "type": "module",
   "module": "src/index.ts",

package/src/tools/gemini.ts

@@ -245,16 +245,13 @@ export const geminiTextToImage = {
             const imageData = part.inlineData.data;
             if (args.output_path) {
               const storage = getStorage();
-              const timestampedPath = generateTimestampedFilename(
-                args.output_path
-              );
               const url = await storage.writeFile(
-                timestampedPath,
+                args.output_path,
                 Buffer.from(imageData, "base64")
               );
               images.push({
                 url,
-                filename: timestampedPath,
+                filename: args.output_path,
                 mimeType: "image/png",
               });
             }
@@ -326,16 +323,13 @@ export const geminiEditImage = {
             const imageData = part.inlineData.data;
             if (args.output_path) {
               const storage = getStorage();
-              const timestampedPath = generateTimestampedFilename(
-                args.output_path
-              );
               const url = await storage.writeFile(
-                timestampedPath,
+                args.output_path,
                 Buffer.from(imageData, "base64")
               );
               images.push({
                 url,
-                filename: timestampedPath,
+                filename: args.output_path,
                 mimeType: "image/png",
               });
             }
@@ -470,12 +464,11 @@ export const geminiSingleSpeakerTts = {
 
       const audioBuffer = Buffer.from(data, "base64");
 
-      // Generate output filename if not provided
-      const outputPath = args.output_path || "voice_output.wav";
-      const timestampedPath = generateTimestampedFilename(outputPath);
+      // Use provided output path or generate default with timestamp
+      const outputPath = args.output_path || generateTimestampedFilename("voice_output.wav");
 
       const storage = getStorage();
-      const url = await storage.writeFile(timestampedPath, audioBuffer);
+      const url = await storage.writeFile(outputPath, audioBuffer);
 
       return JSON.stringify({
         audio: {

package/src/tools/image-to-video.ts

@@ -1,4 +1,3 @@
-import * as fs from "fs";
 import { GoogleAuth } from "google-auth-library";
 import { exec } from "child_process";
 import * as path from "path";
@@ -99,7 +98,9 @@ export const imageToVideo = {
     duration_seconds: z
       .string()
       .optional()
-      .describe("Video duration in seconds: '4', '6', or '8' (default: '6')"),
+      .describe(
+        "Video duration in seconds. MUST be one of: '4', '6', or '8' (default: '6'). Other values will be rejected by Vertex AI."
+      ),
     resolution: z
       .string()
       .optional()
@@ -164,6 +165,32 @@ export const imageToVideo = {
     const location = args.location_id || "us-central1";
     const modelId = args.model_id || "veo-3.1-fast-generate-preview";
 
+    // Validate and parse duration_seconds - snap to nearest 4, 6, or 8
+    let durationSeconds = parseInt(args.duration_seconds || "6");
+    if (isNaN(durationSeconds)) durationSeconds = 6;
+
+    const validDurations = [4, 6, 8];
+    // Find nearest valid duration
+    durationSeconds = validDurations.reduce((prev, curr) => {
+      return Math.abs(curr - durationSeconds) < Math.abs(prev - durationSeconds)
+        ? curr
+        : prev;
+    });
+
+    // Tie-breaking: if equidistant (e.g. 5), the reduce above keeps the first one (4) because < is strict.
+    // However, user requested "nearest duration with the ceil", effectively meaning round up if equidistant.
+    // Let's explicitly handle the equidistant cases or just use a custom finder.
+    // 5 -> equidistant to 4 and 6. "With ceil" implies 6.
+    // 7 -> equidistant to 6 and 8. "With ceil" implies 8.
+
+    // Simpler logic for these specific values:
+    if (durationSeconds === 4 && parseInt(args.duration_seconds || "6") === 5) {
+      durationSeconds = 6;
+    }
+    if (durationSeconds === 6 && parseInt(args.duration_seconds || "6") === 7) {
+      durationSeconds = 8;
+    }
+
     const token = await fetchAccessToken();
 
     const url = `https://${location}-aiplatform.googleapis.com/v1/projects/${projectId}/locations/${location}/publishers/google/models/${modelId}:predictLongRunning`;
@@ -242,7 +269,7 @@ export const imageToVideo = {
 
     const parameters: any = {
       aspectRatio: args.aspect_ratio || "9:16",
-      durationSeconds: parseInt(args.duration_seconds || "6") || 6,
+      durationSeconds: durationSeconds,
       resolution: args.resolution || "720p",
       negativePrompt: args.negative_prompt,
       generateAudio: args.generate_audio || false,
@@ -302,13 +329,20 @@ export const imageToVideo = {
       [];
     const saveVideo = async (base64: string, index: number) => {
       if (!base64) return;
-      const baseFilename = args.output_path
-        ? index === 0
-          ? args.output_path
-          : args.output_path.replace(/\.mp4$/i, `_${index}.mp4`)
-        : `video_output${index > 0 ? `_${index}` : ""}.mp4`;
 
-      const filePath = generateTimestampedFilename(baseFilename);
+      // Use provided output path or generate default with timestamp
+      let filePath: string;
+      if (args.output_path) {
+        // User provided path - use as-is for first video, add index for subsequent
+        filePath =
+          index === 0
+            ? args.output_path
+            : args.output_path.replace(/\.mp4$/i, `_${index}.mp4`);
+      } else {
+        // No path provided - generate timestamped default
+        const defaultName = `video_output${index > 0 ? `_${index}` : ""}.mp4`;
+        filePath = generateTimestampedFilename(defaultName);
+      }
 
       const buf = Buffer.from(base64, "base64");
       const storage = getStorage();

package/src/utils/filename.ts

@@ -1,8 +1,11 @@
+import * as path from "path";
+
 /**
  * Generate a timestamped filename to avoid conflicts
  * Format: YYYYMMDD_HHmmss_filename.ext
+ * Preserves directory structure if present in the input path
  */
-export function generateTimestampedFilename(basename: string): string {
+export function generateTimestampedFilename(filePath: string): string {
   const now = new Date();
   const timestamp = now
     .toISOString()
@@ -10,13 +13,26 @@ export function generateTimestampedFilename(basename: string): string {
     .replace(/\.\d{3}Z$/, "")
     .replace("T", "_");
 
+  // Split into directory and filename
+  const dir = path.dirname(filePath);
+  const basename = path.basename(filePath);
+
   // Extract extension if present
   const lastDot = basename.lastIndexOf(".");
+  let timestampedFilename: string;
+  
   if (lastDot > 0) {
     const name = basename.substring(0, lastDot);
     const ext = basename.substring(lastDot);
-    return `${timestamp}_${name}${ext}`;
+    timestampedFilename = `${timestamp}_${name}${ext}`;
+  } else {
+    timestampedFilename = `${timestamp}_${basename}`;
   }
 
-  return `${timestamp}_${basename}`;
+  // Reconstruct the full path
+  if (dir === ".") {
+    return timestampedFilename;
+  }
+  
+  return path.join(dir, timestampedFilename);
 }