@topazlabs/mcp 0.1.0

package/dist/tools/enhance.js

@@ -0,0 +1,175 @@
+import * as fs from "node:fs";
+import * as os from "node:os";
+import * as path from "node:path";
+import sharp from "sharp";
+import { loadImage } from "../lib/fetch-url.js";
+import { readImageFile } from "../api/client.js";
+import { autoSelectModel, ENHANCE_MODELS, FOUR_K_WIDTH } from "../api/models.js";
+/** Parse the size parameter into target pixel dimensions. */
+export function resolveSize(size, origWidth, origHeight) {
+    if (!size || Object.keys(size).length === 0) {
+        // Default: if under 4K, upscale to 4K width proportionally. Otherwise keep at 1x.
+        if (origWidth < FOUR_K_WIDTH) {
+            const scale = FOUR_K_WIDTH / origWidth;
+            return {
+                width: FOUR_K_WIDTH,
+                height: Math.round(origHeight * scale),
+            };
+        }
+        return { width: origWidth, height: origHeight };
+    }
+    if (size.scale !== undefined) {
+        return {
+            width: Math.round(origWidth * size.scale),
+            height: Math.round(origHeight * size.scale),
+        };
+    }
+    if (size.width !== undefined && size.height !== undefined) {
+        return { width: size.width, height: size.height };
+    }
+    if (size.width !== undefined) {
+        const scale = size.width / origWidth;
+        return { width: size.width, height: Math.round(origHeight * scale) };
+    }
+    if (size.height !== undefined) {
+        const scale = size.height / origHeight;
+        return { width: Math.round(origWidth * scale), height: size.height };
+    }
+    return { width: origWidth, height: origHeight };
+}
+export async function handleEnhance(input, client) {
+    if (!input.image) {
+        return {
+            content: [{ type: "text", text: "Error: image is required." }],
+            isError: true,
+        };
+    }
+    const isUrl = input.image.startsWith("http://") || input.image.startsWith("https://");
+    // Load image data to get dimensions
+    let imageBuffer;
+    let imageFilename;
+    if (isUrl) {
+        imageBuffer = await loadImage(input.image);
+        imageFilename = "image.jpg";
+    }
+    else {
+        const result = readImageFile(input.image);
+        imageBuffer = result.buffer;
+        imageFilename = result.filename;
+    }
+    // Get dimensions using sharp
+    const metadata = await sharp(imageBuffer).metadata();
+    const origWidth = metadata.width ?? 0;
+    const origHeight = metadata.height ?? 0;
+    if (origWidth === 0 || origHeight === 0) {
+        return {
+            content: [
+                { type: "text", text: "Error: Could not determine image dimensions." },
+            ],
+            isError: true,
+        };
+    }
+    // Resolve target size
+    const targetSize = resolveSize(input.size, origWidth, origHeight);
+    // Select model
+    let selectedModel;
+    let faceEnhancement = false;
+    if (input.model) {
+        if (!(input.model in ENHANCE_MODELS)) {
+            const available = Object.keys(ENHANCE_MODELS).join(", ");
+            return {
+                content: [
+                    {
+                        type: "text",
+                        text: `Error: Unknown model "${input.model}". Available: ${available}`,
+                    },
+                ],
+                isError: true,
+            };
+        }
+        selectedModel = input.model;
+    }
+    else {
+        const auto = autoSelectModel(origWidth);
+        selectedModel = auto.model;
+        faceEnhancement = auto.faceEnhancement;
+    }
+    // Build API request
+    const req = {
+        model: selectedModel,
+        output_width: targetSize.width,
+        output_height: targetSize.height,
+        output_format: "png",
+        face_enhancement: faceEnhancement || undefined,
+        image: imageBuffer,
+        image_filename: imageFilename,
+    };
+    const startTime = Date.now();
+    try {
+        const { process_id } = await client.enhance(req);
+        const status = await client.pollUntilComplete(process_id);
+        // Download result
+        const outputDir = path.join(os.tmpdir(), "topaz-mcp");
+        const outputPath = await client.downloadToFile(process_id, outputDir);
+        const elapsedSec = ((Date.now() - startTime) / 1000).toFixed(1);
+        const modelInfo = ENHANCE_MODELS[selectedModel];
+        const summary = [
+            `✅ Enhancement complete`,
+            ``,
+            `**Model:** ${selectedModel} (${modelInfo.type})`,
+            `**Input:** ${origWidth}×${origHeight}`,
+            `**Output:** ${status.output_width ?? targetSize.width}×${status.output_height ?? targetSize.height}`,
+            `**Time:** ${elapsedSec}s`,
+            `**File:** ${outputPath}`,
+        ].join("\n");
+        const content = [
+            { type: "text", text: summary },
+        ];
+        // Include enhanced image inline if small enough (<5MB)
+        try {
+            const outputBuffer = fs.readFileSync(outputPath);
+            if (outputBuffer.length < 5 * 1024 * 1024) {
+                content.push({
+                    type: "image",
+                    data: outputBuffer.toString("base64"),
+                    mimeType: "image/png",
+                });
+            }
+        }
+        catch {
+            // Image too large or read failed — skip inline
+        }
+        // Include viewer resource reference
+        const downloadInfo = await client.getDownloadUrl(process_id);
+        content.push({
+            type: "resource",
+            resource: {
+                uri: "ui://topaz/viewer",
+                mimeType: "text/html",
+                text: buildViewerHtml(downloadInfo.input_download_url ?? "", downloadInfo.download_url),
+            },
+        });
+        return { content };
+    }
+    catch (err) {
+        const message = err instanceof Error ? err.message : String(err);
+        return {
+            content: [{ type: "text", text: `Error enhancing image: ${message}` }],
+            isError: true,
+        };
+    }
+}
+/**
+ * Build a minimal viewer HTML that loads the before/after images from URLs.
+ * For the full viewer resource, the compare tool is more appropriate.
+ */
+function buildViewerHtml(beforeUrl, afterUrl) {
+    const data = JSON.stringify({
+        before: beforeUrl,
+        after: afterUrl,
+        labelBefore: "Original",
+        labelAfter: "Enhanced",
+    }).replace(/</g, "\\u003c");
+    return `<!DOCTYPE html><html><head><script>window.__COMPARE_DATA__=${data};</script></head><body><p>Before/after viewer — use the compare tool for the full interactive experience.</p></body></html>`;
+}
+//# sourceMappingURL=enhance.js.map

package/dist/tools/enhance.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"enhance.js","sourceRoot":"","sources":["../../src/tools/enhance.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AAChD,OAAO,EAAe,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAE9D,OAAO,EAAE,eAAe,EAAE,cAAc,EAAE,YAAY,EAAE,MAAM,kBAAkB,CAAC;AAEjF,6DAA6D;AAC7D,MAAM,UAAU,WAAW,CACzB,IAAqE,EACrE,SAAiB,EACjB,UAAkB;IAElB,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5C,kFAAkF;QAClF,IAAI,SAAS,GAAG,YAAY,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,YAAY,GAAG,SAAS,CAAC;YACvC,OAAO;gBACL,KAAK,EAAE,YAAY;gBACnB,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC;aACvC,CAAC;QACJ,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;IAClD,CAAC;IAED,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC7B,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC;YACzC,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC;SAC5C,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC1D,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;IACpD,CAAC;IAED,IAAI,IAAI,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;QAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,GAAG,SAAS,CAAC;QACrC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,GAAG,KAAK,CAAC,EAAE,CAAC;IACvE,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,GAAG,UAAU,CAAC;QACvC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;IACvE,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC;AAClD,CAAC;AAQD,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,KAAmB,EACnB,MAAmB;IAEnB,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC;QACjB,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,2BAA2B,EAAE,CAAC;YAC9D,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,MAAM,KAAK,GACT,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;IAE1E,oCAAoC;IACpC,IAAI,WAAmB,CAAC;IACxB,IAAI,aAAqB,CAAC;IAE1B,IAAI,KAAK,EAAE,CAAC;QACV,WAAW,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC3C,aAAa,GAAG,WAAW,CAAC;IAC9B,CAAC;SAAM,CAAC;QACN,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC1C,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC;IAClC,CAAC;IAED,6BAA6B;IAC7B,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,CAAC;IACrD,MAAM,SAAS,GAAG,QAAQ,CAAC,KAAK,IAAI,CAAC,CAAC;IACtC,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,IAAI,CAAC,CAAC;IAExC,IAAI,SAAS,KAAK,CAAC,IAAI,UAAU,KAAK,CAAC,EAAE,CAAC;QACxC,OAAO;YACL,OAAO,EAAE;gBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,8CAA8C,EAAE;aACvE;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;IAED,sBAAsB;IACtB,MAAM,UAAU,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;IAElE,eAAe;IACf,IAAI,aAA2B,CAAC;IAChC,IAAI,eAAe,GAAG,KAAK,CAAC;IAE5B,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAChB,IAAI,CAAC,CAAC,KAAK,CAAC,KAAK,IAAI,cAAc,CAAC,EAAE,CAAC;YACrC,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzD,OAAO;gBACL,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,yBAAyB,KAAK,CAAC,KAAK,iBAAiB,SAAS,EAAE;qBACvE;iBACF;gBACD,OAAO,EAAE,IAAI;aACd,CAAC;QACJ,CAAC;QACD,aAAa,GAAG,KAAK,CAAC,KAAqB,CAAC;IAC9C,CAAC;SAAM,CAAC;QACN,MAAM,IAAI,GAAG,eAAe,CAAC,SAAS,CAAC,CAAC;QACxC,aAAa,GAAG,IAAI,CAAC,KAAK,CAAC;QAC3B,eAAe,GAAG,IAAI,CAAC,eAAe,CAAC;IACzC,CAAC;IAED,oBAAoB;IACpB,MAAM,GAAG,GAAmB;QAC1B,KAAK,EAAE,aAAa;QACpB,YAAY,EAAE,UAAU,CAAC,KAAK;QAC9B,aAAa,EAAE,UAAU,CAAC,MAAM;QAChC,aAAa,EAAE,KAAK;QACpB,gBAAgB,EAAE,eAAe,IAAI,SAAS;QAC9C,KAAK,EAAE,WAAW;QAClB,cAAc,EAAE,aAAa;KAC9B,CAAC;IAEF,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,CAAC;QACH,MAAM,EAAE,UAAU,EAAE,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;QAE1D,kBAAkB;QAClB,MAAM,SAAS,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,WAAW,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;QACtE,MAAM,UAAU,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QAEhE,MAAM,SAAS,GAAG,cAAc,CAAC,aAAa,CAAC,CAAC;QAEhD,MAAM,OAAO,GAAG;YACd,wBAAwB;YACxB,EAAE;YACF,cAAc,aAAa,KAAK,SAAS,CAAC,IAAI,GAAG;YACjD,cAAc,SAAS,IAAI,UAAU,EAAE;YACvC,eAAe,MAAM,CAAC,YAAY,IAAI,UAAU,CAAC,KAAK,IAAI,MAAM,CAAC,aAAa,IAAI,UAAU,CAAC,MAAM,EAAE;YACrG,aAAa,UAAU,GAAG;YAC1B,aAAa,UAAU,EAAE;SAC1B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAEb,MAAM,OAAO,GAA8B;YACzC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE;SAChC,CAAC;QAEF,uDAAuD;QACvD,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,CAAC,CAAC;YACjD,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,GAAG,IAAI,GAAG,IAAI,EAAE,CAAC;gBAC1C,OAAO,CAAC,IAAI,CAAC;oBACX,IAAI,EAAE,OAAO;oBACb,IAAI,EAAE,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC;oBACrC,QAAQ,EAAE,WAAW;iBACtB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,+CAA+C;QACjD,CAAC;QAED,oCAAoC;QACpC,MAAM,YAAY,GAAG,MAAM,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,CAAC;QAC7D,OAAO,CAAC,IAAI,CAAC;YACX,IAAI,EAAE,UAAmB;YACzB,QAAQ,EAAE;gBACR,GAAG,EAAE,mBAAmB;gBACxB,QAAQ,EAAE,WAAW;gBACrB,IAAI,EAAE,eAAe,CACnB,YAAY,CAAC,kBAAkB,IAAI,EAAE,EACrC,YAAY,CAAC,YAAY,CAC1B;aACF;SACF,CAAC,CAAC;QAEH,OAAO,EAAE,OAAO,EAAE,CAAC;IACrB,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QACjE,OAAO;YACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,0BAA0B,OAAO,EAAE,EAAE,CAAC;YACtE,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAS,eAAe,CAAC,SAAiB,EAAE,QAAgB;IAC1D,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC;QAC1B,MAAM,EAAE,SAAS;QACjB,KAAK,EAAE,QAAQ;QACf,WAAW,EAAE,UAAU;QACvB,UAAU,EAAE,UAAU;KACvB,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IAE5B,OAAO,8DAA8D,IAAI,6HAA6H,CAAC;AACzM,CAAC"}

package/package.json

@@ -0,0 +1,52 @@
+{
+  "name": "@topazlabs/mcp",
+  "version": "0.1.0",
+  "description": "Topaz Labs MCP server — AI image enhancement and comparison for LLMs",
+  "type": "module",
+  "bin": {
+    "topaz-mcp": "./dist/main.js"
+  },
+  "main": "./dist/main.js",
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build:app": "npx --no-install vite build",
+    "build:server": "tsc -p tsconfig.server.json",
+    "build": "npm run build:app && npm run build:server",
+    "dev": "tsc -p tsconfig.server.json --watch",
+    "start": "node dist/main.js"
+  },
+  "keywords": [
+    "mcp",
+    "image",
+    "enhance",
+    "upscale",
+    "compare",
+    "topaz",
+    "ai",
+    "mcp-server"
+  ],
+  "author": "Topaz Labs <hello@topazlabs.com>",
+  "license": "MIT",
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.12.1",
+    "sharp": "^0.33.5",
+    "zod": "^3.25.76"
+  },
+  "devDependencies": {
+    "@types/node": "^22.19.11",
+    "typescript": "5.7",
+    "vite": "^6.1.0",
+    "vite-plugin-singlefile": "^2.0.3"
+  },
+  "engines": {
+    "node": ">=20"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/EricYangTL/topaz-mcp.git",
+    "directory": "packages/mcp"
+  },
+  "homepage": "https://github.com/EricYangTL/topaz-mcp#readme"
+}