@visionengine/video-style-transfer 1.0.0

package/README-zh.md

@@ -0,0 +1,24 @@
+# VE 视频风格重绘 MCP
+
+通过 `ve-backend` 代理实现阿里云 DashScope 视频风格重绘的异步 MCP 服务。
+
+## 环境变量
+
+- `API_URL`：后端代理地址，默认 `https://api.visionengine-tech.com/api/v1/video-style-transfer`
+- `API_KEY`：VisionEngine 平台用户 API Key
+- `WORKDIR`：本地工作目录根路径
+- `DEFAULT_OUTPUT_DIR`：默认输出目录，默认 `public/videos`
+
+## 工具
+
+- `video_style_transfer`
+
+当前版本 `videoPath` 支持两种输入：
+
+- 公网可访问视频 URL
+- 后端与 MCP 运行时共同挂载的 NAS 路径
+
+如果 DashScope 最终返回的是 NAS 文件路径而不是直接 URL，可由后端结合以下环境变量转换出下载地址：
+
+- `VIDEO_STYLE_TRANSFER_SHARED_BASE_URL`
+- `VIDEO_STYLE_TRANSFER_SHARED_MOUNT_ROOT`

package/README.md

@@ -0,0 +1,24 @@
+# VE Video Style Transfer MCP
+
+Async MCP server for video style redraw via `ve-backend` proxy.
+
+## Environment
+
+- `API_URL`: backend base url, default `https://api.visionengine-tech.com/api/v1/video-style-transfer`
+- `API_KEY`: user API key from VisionEngine backend
+- `WORKDIR`: local workspace root
+- `DEFAULT_OUTPUT_DIR`: default relative output directory, default `public/videos`
+
+## Tool
+
+- `video_style_transfer`
+
+`videoPath` can be either:
+
+- a public video URL, or
+- a shared NAS path that both backend and MCP runtime can access.
+
+If DashScope returns a shared file path instead of a direct URL, backend can expose it by configuring:
+
+- `VIDEO_STYLE_TRANSFER_SHARED_BASE_URL`
+- `VIDEO_STYLE_TRANSFER_SHARED_MOUNT_ROOT`

package/dist/index.js

@@ -0,0 +1,5 @@
+#!/usr/bin/env node
+import { server } from "./server.js";
+server.start({
+    transportType: "stdio",
+});

package/dist/server.js

@@ -0,0 +1,185 @@
+import { FastMCP } from "fastmcp";
+import { z } from "zod";
+import * as fs from "fs";
+import * as path from "path";
+import * as http from "http";
+import * as https from "https";
+import * as crypto from "crypto";
+const server = new FastMCP({
+    name: "VE Video Style Transfer",
+    version: "1.0.0",
+});
+const API_URL = process.env.API_URL || "https://api.visionengine-tech.com/api/v1/video-style-transfer";
+const API_KEY = process.env.API_KEY || "";
+const WORKDIR = process.env.WORKDIR || "./";
+const DEFAULT_OUTPUT_DIR = process.env.DEFAULT_OUTPUT_DIR || "public/videos";
+function getWorkDir() {
+    return path.isAbsolute(WORKDIR) ? WORKDIR : path.resolve(process.cwd(), WORKDIR);
+}
+function ensureDir(dirPath) {
+    if (!fs.existsSync(dirPath)) {
+        fs.mkdirSync(dirPath, { recursive: true });
+    }
+    return dirPath;
+}
+function resolveInputPath(inputPath) {
+    return path.isAbsolute(inputPath) ? inputPath : path.resolve(getWorkDir(), inputPath);
+}
+function getDefaultOutputDir() {
+    const workDir = getWorkDir();
+    const outputDir = path.isAbsolute(DEFAULT_OUTPUT_DIR)
+        ? DEFAULT_OUTPUT_DIR
+        : path.join(workDir, DEFAULT_OUTPUT_DIR);
+    return ensureDir(outputDir);
+}
+function generateFilename(extension = ".mp4") {
+    return `styled_${Date.now()}_${crypto.randomBytes(4).toString("hex")}${extension}`;
+}
+function isHttpUrl(value) {
+    return /^https?:\/\//i.test(value);
+}
+function detectMediaUrl(input) {
+    if (isHttpUrl(input)) {
+        return input;
+    }
+    const fullPath = resolveInputPath(input);
+    if (!fs.existsSync(fullPath)) {
+        throw new Error(`Video file not found: ${fullPath}`);
+    }
+    return fullPath;
+}
+async function requestJson(url, init) {
+    if (!API_KEY) {
+        throw new Error("API_KEY environment variable is required");
+    }
+    const response = await fetch(url, {
+        method: init.method,
+        headers: {
+            Authorization: `Bearer ${API_KEY}`,
+            "Content-Type": "application/json",
+        },
+        body: init.body !== undefined ? JSON.stringify(init.body) : undefined,
+    });
+    if (!response.ok) {
+        throw new Error(`Backend API error (${response.status}): ${await response.text()}`);
+    }
+    return (await response.json());
+}
+async function downloadFile(url, outputPath) {
+    const parsed = new URL(url);
+    const mod = parsed.protocol === "https:" ? https : http;
+    await new Promise((resolve, reject) => {
+        const request = mod.get(url, (response) => {
+            if (response.statusCode && response.statusCode >= 400) {
+                reject(new Error(`Failed to download file: HTTP ${response.statusCode}`));
+                return;
+            }
+            const stream = fs.createWriteStream(outputPath);
+            response.pipe(stream);
+            stream.on("finish", () => {
+                stream.close();
+                resolve();
+            });
+            stream.on("error", reject);
+        });
+        request.on("error", reject);
+    });
+}
+async function pollTask(taskId, pollIntervalMs, timeoutMs) {
+    const started = Date.now();
+    while (true) {
+        const result = await requestJson(`${API_URL}/query?task_id=${encodeURIComponent(taskId)}`, { method: "GET" });
+        if (["SUCCEEDED", "FAILED", "CANCELED", "CANCELLED", "UNKNOWN"].includes(result.task_status)) {
+            return result;
+        }
+        if (Date.now() - started > timeoutMs) {
+            throw new Error(`Video style transfer timed out after ${timeoutMs}ms`);
+        }
+        await new Promise((resolve) => setTimeout(resolve, pollIntervalMs));
+    }
+}
+async function transferVideoStyle(options) {
+    const videoUrl = detectMediaUrl(options.videoPath);
+    const submitPayload = {
+        video_url: videoUrl,
+        parameters: {
+            style: options.style ?? 0,
+            video_fps: options.videoFps ?? 15,
+            animate_emotion: options.animateEmotion ?? true,
+            min_len: options.minLen ?? 720,
+            use_SR: options.useSR ?? false,
+        },
+    };
+    const submitResult = await requestJson(`${API_URL}/submit`, {
+        method: "POST",
+        body: submitPayload,
+    });
+    const queryResult = await pollTask(submitResult.task_id, options.pollIntervalMs ?? 5000, options.timeoutMs ?? 10 * 60 * 1000);
+    if (queryResult.task_status !== "SUCCEEDED") {
+        throw new Error(`Video style transfer failed with status ${queryResult.task_status}`);
+    }
+    if (!queryResult.output_video_url) {
+        throw new Error("Task succeeded but output_video_url is missing");
+    }
+    let resolvedOutputPath;
+    if (options.outputPath) {
+        resolvedOutputPath = path.isAbsolute(options.outputPath)
+            ? options.outputPath
+            : path.resolve(getWorkDir(), options.outputPath);
+        ensureDir(path.dirname(resolvedOutputPath));
+    }
+    else {
+        const outputDir = getDefaultOutputDir();
+        resolvedOutputPath = path.join(outputDir, generateFilename(".mp4"));
+    }
+    await downloadFile(queryResult.output_video_url, resolvedOutputPath);
+    return {
+        success: true,
+        taskId: submitResult.task_id,
+        taskStatus: queryResult.task_status,
+        style: options.style ?? 0,
+        outputVideoPath: resolvedOutputPath,
+        outputVideoUrl: queryResult.output_video_url,
+        usage: {
+            duration: queryResult.usage?.duration ?? null,
+            sr: queryResult.usage?.sr ?? null,
+        },
+        billingStatus: queryResult.billing_status,
+        message: "Video style transfer completed successfully",
+    };
+}
+server.addTool({
+    annotations: {
+        openWorldHint: true,
+        readOnlyHint: false,
+        title: "Video Style Transfer",
+    },
+    description: "Submit a video style redraw task to the backend proxy, poll until completion, then download the generated video locally.",
+    execute: async (args) => {
+        const result = await transferVideoStyle({
+            videoPath: args.videoPath,
+            style: args.style,
+            videoFps: args.videoFps,
+            animateEmotion: args.animateEmotion,
+            minLen: args.minLen,
+            useSR: args.useSR,
+            outputPath: args.outputPath,
+            pollIntervalMs: args.pollIntervalMs,
+            timeoutMs: args.timeoutMs,
+        });
+        return JSON.stringify(result, null, 2);
+    },
+    name: "video_style_transfer",
+    parameters: z.object({
+        videoPath: z.string().describe("Public video URL, or shared NAS path visible to both backend and MCP runtime."),
+        style: z.number().int().min(0).max(7).optional().describe("Style index: 0 日式漫画, 1 美式漫画, 2 清新漫画, 3 3D卡通, 4 国风卡通, 5 纸艺风格, 6 简易插画, 7 国风水墨."),
+        videoFps: z.number().int().min(15).max(25).optional().describe("Output FPS, 15~25, default 15."),
+        animateEmotion: z.boolean().optional().describe("Whether to enable animate emotion optimization. Default true."),
+        minLen: z.union([z.literal(540), z.literal(720)]).optional().describe("Output short side resolution, 540 or 720. Default 720."),
+        useSR: z.boolean().optional().describe("Whether to enable super-resolution. Default false."),
+        outputPath: z.string().optional().describe("Output path for the generated video. Relative to WORKDIR or absolute path."),
+        pollIntervalMs: z.number().int().positive().optional().describe("Polling interval in milliseconds. Default 5000."),
+        timeoutMs: z.number().int().positive().optional().describe("Overall timeout in milliseconds. Default 600000 (10 minutes)."),
+    }),
+});
+export { server };

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "@visionengine/video-style-transfer",
+  "version": "1.0.0",
+  "description": "VisionEngine Video Style Transfer MCP Server - Async video style redraw via backend proxy",
+  "main": "dist/index.js",
+  "type": "module",
+  "bin": {
+    "ve-video-style-transfer": "./dist/index.js"
+  },
+  "files": [
+    "dist",
+    "README.md",
+    "README-zh.md"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "clear": "node -e \"require('fs').rmSync('dist', { recursive: true, force: true })\"",
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "mcp",
+    "video-style-transfer",
+    "video-redraw",
+    "dashscope",
+    "visionengine"
+  ],
+  "author": "team@visionengine-tech.com",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/crazyyanchao/ve-mcp.git",
+    "directory": "packages/video-style-transfer"
+  },
+  "homepage": "https://visionengine-tech.com/mcp",
+  "dependencies": {
+    "fastmcp": "^3.26.8",
+    "zod": "^4.1.12"
+  },
+  "devDependencies": {
+    "@types/node": "^24.10.1",
+    "typescript": "^5.8.3",
+    "vitest": "^3.1.3"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  }
+}