npm - @vint.tri/report_gen_mcp - Versions diffs - 1.3.9 → 1.4.1 - Mend

@vint.tri/report_gen_mcp 1.3.9 → 1.4.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (10) hide show

package/PUBLICATION_CONFIRMATION_V1.4.0.md +54 -0
package/UPDATED_NEURAL_NETWORK_INSTRUCTIONS.md +4 -1
package/VERSION_1.4.0_RELEASE_NOTES.md +88 -0
package/dist/index.js +93 -267
package/dist/mcp/imageEditingServer.js +294 -0
package/dist/mcp/imageGenerationServer.js +270 -0
package/lizard_wedding.png +0 -0
package/lizard_wedding_bitter.jpg +0 -0
package/moose_image.png +0 -0
package/package.json +2 -1

package/PUBLICATION_CONFIRMATION_V1.4.0.md ADDED Viewed

@@ -0,0 +1,54 @@
+# Publication Confirmation - Version 1.4.0
+Version 1.4.0 of @vint.tri/report_gen_mcp has been successfully published to npm.
+## Published Features
+This release includes significant improvements to image generation and editing capabilities:
+1. **Enhanced Image Generation Tool** (`generate-image`):
+   - Full integration with Python scripts for actual image generation
+   - Proper error handling and response parsing
+   - File path and URL generation for generated images
+   - Configurable timeouts for generation process
+2. **Enhanced Image Editing Tool** (`edit-image`):
+   - Full integration with Python scripts for actual image editing
+   - Proper error handling and response parsing
+   - File path and URL generation for edited images
+   - Configurable timeouts for editing process
+3. **Improved Python Integration**:
+   - Proper communication between Node.js MCP server and Python scripts via stdin/stdout
+   - Environment variable passing (CHUTES_API_TOKEN)
+   - File existence verification
+   - Better error handling for Python script execution
+4. **Version Consistency**:
+   - Updated package version from 1.3.9 to 1.4.0
+   - Updated MCP server version to match package version
+## Verification
+- Package version: 1.4.0 ✓
+- NPM registry confirmation: ✓
+- Build verification: ✓
+- Tool registration testing: ✓
+- Image generation and editing functionality testing: ✓
+## Next Steps
+Users can now install the updated package with:
+```bash
+npm install @vint.tri/report_gen_mcp@1.4.0
+```
+Or globally:
+```bash
+npm install -g @vint.tri/report_gen_mcp@1.4.0
+```
+To use the enhanced image features, users will need to:
+1. Install Python dependencies: `npm run install-python-deps`
+2. Set the CHUTES_API_TOKEN environment variable
+3. Set the REPORTS_DIR environment variable for storing generated reports

package/UPDATED_NEURAL_NETWORK_INSTRUCTIONS.md CHANGED Viewed

@@ -64,7 +64,8 @@
 1. **Путь к файлу**: Абсолютный путь к созданному отчету
 2. **Ссылка на файл**: Кликабельная file:// ссылка для открытия отчета в браузере
-3. **Содержимое файла**: Полный текст HTML содержимого отчета
+3. **HTTP ссылка**: Если сервер запущен, ссылка вида http://localhost:3000/report/filename.html для доступа к отчету через веб-интерфейс
+4. **Содержимое файла**: Полный текст HTML содержимого отчета
 Обязательно покажите пользователю содержимое файла, чтобы он мог сразу ознакомиться с результатом без необходимости открывать файл отдельно.
@@ -76,6 +77,7 @@
 📁 Путь к файлу: /полный/путь/к/отчету.html
 🌐 Ссылка для открытия в браузере: file:///полный/путь/к/отчету.html
+🔗 HTTP ссылка: http://localhost:3000/report/отчет.html
 📄 Содержимое отчета:
 <!DOCTYPE html>
@@ -522,6 +524,7 @@
 📁 Путь к файлу: /path/to/report.html
 🌐 Ссылка для открытия в браузере: file:///path/to/report.html
+🔗 HTTP ссылка: http://localhost:3000/report/report.html
 Содержимое отчета:
 <!DOCTYPE html>

package/VERSION_1.4.0_RELEASE_NOTES.md ADDED Viewed

@@ -0,0 +1,88 @@
+# Version 1.4.0 Release Notes
+## New Features and Enhancements
+### Major Version Update
+- Updated package version from 1.3.9 to 1.4.0
+- Updated MCP server version to match package version
+### Image Generation and Editing Improvements
+Based on the IMAGE_GENERATION_FIX.md documentation, this version includes significant improvements to image handling:
+1. **Fixed Image Generation Tool (`generate-image`)**
+   - Replaced placeholder implementation with actual integration with `src/python/mcp_img_gen.py`
+   - Added proper child_process execution to call Python script
+   - Implemented proper error handling and response parsing
+   - Added file path and URL generation for generated images
+   - Set appropriate timeouts (60 seconds for generation)
+2. **Fixed Image Editing Tool (`edit-image`)**
+   - Replaced placeholder implementation with actual integration with `src/python/mcp_image_edit.py`
+   - Added proper child_process execution to call Python script
+   - Implemented proper error handling and response parsing
+   - Added file path and URL generation for edited images
+   - Set appropriate timeouts (120 seconds for editing)
+3. **Key Improvements**
+   - Both tools now properly communicate with Python scripts via stdin/stdout
+   - Added proper environment variable passing (CHUTES_API_TOKEN)
+   - Implemented file existence verification
+   - Added proper error handling for Python script execution
+   - Return actual file paths and URLs instead of placeholder text
+   - Maintain backward compatibility with existing API
+## Technical Details
+### Communication Flow
+1. MCP client calls `generate-image` or `edit-image` tool
+2. Node.js MCP server receives the call
+3. Server spawns Python process with appropriate script
+4. Server sends tool parameters as JSON to Python script stdin
+5. Python script processes the request and generates image
+6. Python script returns result as JSON to stdout
+7. Node.js server parses response and returns proper content to client
+### Required Setup
+1. Python dependencies must be installed:
+   ```bash
+   npm run install-python-deps
+   ```
+2. CHUTES_API_TOKEN environment variable must be set for actual image generation
+## Files Updated
+- `package.json` - Version bump from 1.3.9 to 1.4.0
+- `src/index.ts` - Updated MCP server version from 1.3.8 to 1.4.0
+- `dist/index.js` - Regenerated with updated version
+- `VERSION_1.4.0_RELEASE_NOTES.md` - This file
+## Verification
+The improvements have been tested and verified to:
+- ✅ Properly integrate with Python image generation scripts
+- ✅ Return actual file paths and URLs instead of placeholder text
+- ✅ Handle errors gracefully
+- ✅ Maintain compatibility with existing API
+- ✅ Support both image generation and editing functionalities
+## Installation
+To upgrade to version 1.4.0, run:
+```bash
+npm install @vint.tri/report_gen_mcp@1.4.0
+```
+Or update your package.json dependency:
+```json
+{
+  "dependencies": {
+    "@vint.tri/report_gen_mcp": "^1.4.0"
+  }
+}
+```
+## Required Environment Variables
+- `REPORTS_DIR` - Directory for storing generated reports and locating Python scripts
+- `CHUTES_API_TOKEN` - Required for actual image generation/editing (optional for testing)
+## Python Dependencies
+If using image generation features, ensure Python dependencies are installed:
+```bash
+npm run install-python-deps

package/dist/index.js CHANGED Viewed

@@ -38,6 +38,10 @@ if (!isStdioMode) {
 const app = express();
 const port = 3000;
 app.use(express.json());
+// Serve static files from the reports directory at /report/ endpoint
+if (reportsDir) {
+    app.use('/report', express.static(reportsDir));
+}
 app.post('/generate-report', async (req, res) => {
     // For HTTP API mode, use the REPORTS_DIR environment variable
     // This endpoint only runs in non-stdio mode where reportsDir is guaranteed to be defined
@@ -103,7 +107,7 @@ if (process.argv.length === 2) {
     // No command specified, run in stdio mode using MCP SDK
     const mcpServer = new McpServer({
         name: "report_gen_mcp",
-        version: "1.3.8",
+        version: "1.4.1",
     }, {
         // Disable health check to prevent automatic calls
         capabilities: {
@@ -263,8 +267,10 @@ if (process.argv.length === 2) {
                 fileUrl: fileUrl,
                 fileContent: fileContent
             };
+            // Generate HTTP link if server is running
+            const httpLink = reportsDir ? `http://localhost:${port}/report/${path.basename(outputPath)}` : null;
             // Return comprehensive information for neural network to use
-            // Neural network should present to user: file path, web link, and file content
+            // Neural network should present to user: file path, web link, HTTP link, and file content
             return {
                 content: [
                     {
@@ -274,8 +280,7 @@ if (process.argv.length === 2) {
 FOR THE NEURAL NETWORK: Please present the following information to the user:
 1. Path to file: ${outputPath}
 2. Web browser link: ${fileUrl}
-3. File content:
-${fileContent}`
+${httpLink ? `3. HTTP link: ${httpLink}\n4. File content:\n${fileContent}` : `3. File content:\n${fileContent}`}`
                     }
                 ]
             };
@@ -386,6 +391,8 @@ ${fileContent}`
             await fs.writeFile(filePath, fileContent, 'utf8');
             // Generate proper file URL
             const fileUrl = pathToFileURL(filePath).href;
+            // Generate HTTP link if server is running
+            const httpLink = reportsDir ? `http://localhost:${port}/report/${path.basename(filePath)}` : null;
             return {
                 content: [
                     {
@@ -395,7 +402,7 @@ ${fileContent}`
 FOR THE NEURAL NETWORK: Please present the following information to the user:
 1. Path to file: ${filePath}
 2. Web browser link: ${fileUrl}
-3. Operation performed: ${operation}`
+${httpLink ? `3. HTTP link: ${httpLink}\n4. Operation performed: ${operation}` : `3. Operation performed: ${operation}`}`
                     }
                 ]
             };
@@ -448,9 +455,8 @@ FOR THE NEURAL NETWORK: Please present the following information to the user:
                     }]
             };
         }
-        // Import child_process for calling Python scripts
-        const { spawn } = await import('child_process');
-        const { promises: fsPromises } = await import('fs');
+        // Import our TypeScript implementation
+        const { ImageGenerator } = await import('./mcp/imageGenerationServer.js');
         const path = await import('path');
         const os = await import('os');
         // Determine the output directory:
@@ -461,7 +467,7 @@ FOR THE NEURAL NETWORK: Please present the following information to the user:
             outputDir = process.env.REPORTS_DIR;
             // Ensure the reports directory exists
             try {
-                await fsPromises.access(outputDir).catch(() => fsPromises.mkdir(outputDir, { recursive: true }));
+                await fs.access(outputDir).catch(() => fs.mkdir(outputDir, { recursive: true }));
             }
             catch (error) {
                 throw new Error(`Cannot create or access the reports directory: ${outputDir}`);
@@ -471,141 +477,44 @@ FOR THE NEURAL NETWORK: Please present the following information to the user:
             outputDir = os.tmpdir();
         }
         // Generate a unique filename if not provided
-        const fileName = outputFile || `generated-image-${Date.now()}.png`;
+        const fileName = outputFile || `generated-image-${Date.now()}.jpeg`;
         const fullPath = path.resolve(outputDir, fileName);
-        // Prepare arguments for the Python script
-        // Use REPORTS_DIR environment variable to locate Python scripts
-        const reportsDir = process.env.REPORTS_DIR || process.cwd();
-        const pythonScriptPath = path.resolve(reportsDir, 'src', 'python', 'mcp_img_gen.py');
-        const pythonArgs = [
-            '-c',
-            `import sys; sys.path.insert(0, '${path.dirname(pythonScriptPath)}'); ` +
-                `import mcp_img_gen; ` +
-                `import asyncio; ` +
-                `asyncio.run(mcp_img_gen.main())`
-        ];
-        // Prepare the tool call arguments as JSON
-        const toolCallArgs = {
-            name: "generate_image_to_file",
-            arguments: {
-                prompt: prompt,
-                directory: outputDir,
-                filename: fileName,
+        // Create image generator instance
+        const imageGen = new ImageGenerator();
+        try {
+            // Generate image using our TypeScript implementation
+            const imageDataUri = await imageGen.generateImage(prompt, {
                 width: width,
                 height: height,
                 guidance_scale: guidanceScale,
                 negative_prompt: negativePrompt,
                 num_inference_steps: numInferenceSteps,
-                seed: seed
-            }
-        };
-        // Execute the Python script
-        return new Promise((resolve, reject) => {
-            const pythonProcess = spawn('python3', pythonArgs, {
-                env: {
-                    ...process.env,
-                    CHUTES_API_TOKEN: process.env.CHUTES_API_TOKEN,
-                },
-                stdio: ['pipe', 'pipe', 'pipe']
-            });
-            let stdoutData = '';
-            let stderrData = '';
-            pythonProcess.stdout.on('data', (data) => {
-                stdoutData += data.toString();
-            });
-            pythonProcess.stderr.on('data', (data) => {
-                stderrData += data.toString();
-            });
-            pythonProcess.on('close', async (code) => {
-                if (code !== 0) {
-                    reject(new Error(`Python script exited with code ${code}. Error: ${stderrData}`));
-                    return;
-                }
-                try {
-                    // Parse the response from the Python script
-                    const responseLines = stdoutData.trim().split('\n');
-                    const jsonResponse = responseLines.find(line => line.startsWith('{') && line.endsWith('}'));
-                    if (jsonResponse) {
-                        const response = JSON.parse(jsonResponse);
-                        if (response.result && response.result.content) {
-                            // Look for success message in the response
-                            const successContent = response.result.content.find((item) => item.type === "text" && item.text.includes("успешно сгенерировано"));
-                            if (successContent) {
-                                // Check if file was created
-                                try {
-                                    await fsPromises.access(fullPath);
-                                    // Generate proper file URL
-                                    const { pathToFileURL } = await import('url');
-                                    const fileUrl = pathToFileURL(fullPath).href;
-                                    resolve({
-                                        content: [{
-                                                type: "text",
-                                                text: `Image successfully generated!\n\nFile saved to: ${fullPath}\nWeb link: ${fileUrl}`
-                                            }]
-                                    });
-                                }
-                                catch (fileError) {
-                                    resolve({
-                                        content: [{
-                                                type: "text",
-                                                text: `Image generation completed according to Python script, but file was not found at expected location: ${fullPath}`
-                                            }]
-                                    });
-                                }
-                            }
-                            else {
-                                // Look for error message
-                                const errorContent = response.result.content.find((item) => item.type === "text" && item.text.includes("Ошибка"));
-                                if (errorContent) {
-                                    reject(new Error(errorContent.text));
-                                }
-                                else {
-                                    resolve({
-                                        content: [{
-                                                type: "text",
-                                                text: `Image generation completed. Response: ${JSON.stringify(response.result.content, null, 2)}`
-                                            }]
-                                    });
-                                }
-                            }
-                        }
-                        else {
-                            resolve({
-                                content: [{
-                                        type: "text",
-                                        text: `Image generation tool executed. Response: ${JSON.stringify(response, null, 2)}`
-                                    }]
-                            });
-                        }
-                    }
-                    else {
-                        // If we can't parse JSON, return the raw output
-                        resolve({
-                            content: [{
-                                    type: "text",
-                                    text: `Image generation completed.\n\nOutput:\n${stdoutData}`
-                                }]
-                        });
-                    }
-                }
-                catch (parseError) {
-                    resolve({
-                        content: [{
-                                type: "text",
-                                text: `Image generation completed.\n\nRaw output:\n${stdoutData}\n\nError parsing response: ${parseError}`
-                            }]
-                    });
-                }
+                seed: seed !== 0 ? seed : null,
+                model: model
             });
-            // Send the tool call to the Python script
-            pythonProcess.stdin.write(JSON.stringify(toolCallArgs) + '\n');
-            pythonProcess.stdin.end();
-            // Set a timeout to prevent hanging
-            setTimeout(() => {
-                pythonProcess.kill();
-                reject(new Error('Image generation timed out after 60 seconds'));
-            }, 60000);
-        });
+            // Extract base64 data
+            let base64Data = imageDataUri;
+            if (imageDataUri.startsWith("data:image/jpeg;base64,")) {
+                base64Data = imageDataUri.substring("data:image/jpeg;base64,".length);
+            }
+            // Save image to file
+            const imageBuffer = Buffer.from(base64Data, 'base64');
+            await fs.writeFile(fullPath, imageBuffer);
+            // Generate proper file URL
+            const { pathToFileURL } = await import('url');
+            const fileUrl = pathToFileURL(fullPath).href;
+            // Generate HTTP link if server is running
+            const httpLink = reportsDir ? `http://localhost:${port}/report/${path.basename(fullPath)}` : null;
+            return {
+                content: [{
+                        type: "text",
+                        text: `Image successfully generated!\n\nFile saved to: ${fullPath}\nWeb link: ${fileUrl}${httpLink ? `\nHTTP link: ${httpLink}` : ''}`
+                    }]
+            };
+        }
+        catch (error) {
+            throw new Error(`Image generation failed: ${error.message}`);
+        }
     });
     // Register image editing tool
     mcpServer.registerTool("edit-image", {
@@ -657,143 +566,60 @@ FOR THE NEURAL NETWORK: Please present the following information to the user:
                     }]
             };
         }
-        // Import child_process for calling Python scripts
-        const { spawn } = await import('child_process');
-        const { promises: fsPromises } = await import('fs');
+        // Import our TypeScript implementation
+        const { ImageEditor } = await import('./mcp/imageEditingServer.js');
         const path = await import('path');
-        // Prepare arguments for the Python script
-        // Use REPORTS_DIR environment variable to locate Python scripts
-        const reportsDir = process.env.REPORTS_DIR || process.cwd();
-        const pythonScriptPath = path.resolve(reportsDir, 'src', 'python', 'mcp_image_edit.py');
-        const pythonArgs = [
-            '-c',
-            `import sys; sys.path.insert(0, '${path.dirname(pythonScriptPath)}'); ` +
-                `import mcp_image_edit; ` +
-                `import asyncio; ` +
-                `asyncio.run(mcp_image_edit.main())`
-        ];
-        // Prepare the tool call arguments as JSON
-        const toolCallArgs = {
-            name: "edit_image_file",
-            arguments: {
-                prompt: prompt,
-                image_path: imagePath,
-                output_path: output_path,
+        // Create image editor instance
+        const imageEditor = new ImageEditor();
+        try {
+            // Обрабатываем путь к исходному файлу с учетом базового каталога
+            const baseSaveDirectory = process.env.IMG_SAVE_BASE_DIR || "";
+            const fullImagePath = baseSaveDirectory && !path.isAbsolute(imagePath)
+                ? path.normalize(path.join(baseSaveDirectory, imagePath))
+                : path.normalize(imagePath);
+            // Проверяем существование входного файла
+            if (!await fs.pathExists(fullImagePath)) {
+                throw new Error(`Файл изображения не найден: ${fullImagePath}`);
+            }
+            // Загружаем изображение и конвертируем в base64
+            const imageBuffer = await fs.readFile(fullImagePath);
+            const imageB64 = imageBuffer.toString('base64');
+            // Редактируем изображение через base64 метод
+            const editedImageB64 = await imageEditor.editImage(prompt, imageB64, {
                 width: width,
                 height: height,
                 true_cfg_scale: cfgScale,
                 negative_prompt: negativePrompt,
                 num_inference_steps: numInferenceSteps,
-                seed: seed
-            }
-        };
-        // Execute the Python script
-        return new Promise((resolve, reject) => {
-            const pythonProcess = spawn('python3', pythonArgs, {
-                env: {
-                    ...process.env,
-                    CHUTES_API_TOKEN: process.env.CHUTES_API_TOKEN,
-                },
-                stdio: ['pipe', 'pipe', 'pipe']
-            });
-            let stdoutData = '';
-            let stderrData = '';
-            pythonProcess.stdout.on('data', (data) => {
-                stdoutData += data.toString();
+                seed: seed || null
             });
-            pythonProcess.stderr.on('data', (data) => {
-                stderrData += data.toString();
-            });
-            pythonProcess.on('close', async (code) => {
-                if (code !== 0) {
-                    reject(new Error(`Python script exited with code ${code}. Error: ${stderrData}`));
-                    return;
-                }
-                try {
-                    // Parse the response from the Python script
-                    const responseLines = stdoutData.trim().split('\n');
-                    const jsonResponse = responseLines.find(line => line.startsWith('{') && line.endsWith('}'));
-                    if (jsonResponse) {
-                        const response = JSON.parse(jsonResponse);
-                        if (response.result && response.result.content) {
-                            // Look for success message in the response
-                            const successContent = response.result.content.find((item) => item.type === "text" && item.text.includes("успешно отредактировано"));
-                            if (successContent) {
-                                // Check if output file was created
-                                try {
-                                    await fsPromises.access(output_path);
-                                    // Generate proper file URL
-                                    const { pathToFileURL } = await import('url');
-                                    const fileUrl = pathToFileURL(path.resolve(output_path)).href;
-                                    resolve({
-                                        content: [{
-                                                type: "text",
-                                                text: `Image successfully edited!\n\nOutput file: ${output_path}\nWeb link: ${fileUrl}`
-                                            }]
-                                    });
-                                }
-                                catch (fileError) {
-                                    resolve({
-                                        content: [{
-                                                type: "text",
-                                                text: `Image editing completed according to Python script, but output file was not found at expected location: ${output_path}`
-                                            }]
-                                    });
-                                }
-                            }
-                            else {
-                                // Look for error message
-                                const errorContent = response.result.content.find((item) => item.type === "text" && item.text.includes("Ошибка"));
-                                if (errorContent) {
-                                    reject(new Error(errorContent.text));
-                                }
-                                else {
-                                    resolve({
-                                        content: [{
-                                                type: "text",
-                                                text: `Image editing completed. Response: ${JSON.stringify(response.result.content, null, 2)}`
-                                            }]
-                                    });
-                                }
-                            }
-                        }
-                        else {
-                            resolve({
-                                content: [{
-                                        type: "text",
-                                        text: `Image editing tool executed. Response: ${JSON.stringify(response, null, 2)}`
-                                    }]
-                            });
-                        }
-                    }
-                    else {
-                        // If we can't parse JSON, return the raw output
-                        resolve({
-                            content: [{
-                                    type: "text",
-                                    text: `Image editing completed.\n\nOutput:\n${stdoutData}`
-                                }]
-                        });
-                    }
-                }
-                catch (parseError) {
-                    resolve({
-                        content: [{
-                                type: "text",
-                                text: `Image editing completed.\n\nRaw output:\n${stdoutData}\n\nError parsing response: ${parseError}`
-                            }]
-                    });
-                }
-            });
-            // Send the tool call to the Python script
-            pythonProcess.stdin.write(JSON.stringify(toolCallArgs) + '\n');
-            pythonProcess.stdin.end();
-            // Set a timeout to prevent hanging
-            setTimeout(() => {
-                pythonProcess.kill();
-                reject(new Error('Image editing timed out after 120 seconds'));
-            }, 120000);
-        });
+            // Декодируем и сохраняем результат
+            const editedImageBuffer = Buffer.from(editedImageB64, 'base64');
+            // Обрабатываем путь к выходному файлу с учетом базового каталога
+            const fullOutputPath = baseSaveDirectory && !path.isAbsolute(output_path)
+                ? path.normalize(path.join(baseSaveDirectory, output_path))
+                : path.normalize(output_path);
+            // Создаем папку для выходного файла если нужно
+            const outputDir = path.dirname(fullOutputPath);
+            if (outputDir && !await fs.pathExists(outputDir)) {
+                await fs.ensureDir(outputDir);
+            }
+            await fs.writeFile(fullOutputPath, editedImageBuffer);
+            // Создаем file URI для выходного файла
+            const { pathToFileURL } = await import('url');
+            const fileUrl = pathToFileURL(path.resolve(fullOutputPath)).href;
+            // Generate HTTP link if server is running
+            const httpLink = reportsDir ? `http://localhost:${port}/report/${path.basename(fullOutputPath)}` : null;
+            return {
+                content: [{
+                        type: "text",
+                        text: `Image successfully edited!\n\nOutput file: ${fullOutputPath}\nWeb link: ${fileUrl}${httpLink ? `\nHTTP link: ${httpLink}` : ''}`
+                    }]
+            };
+        }
+        catch (error) {
+            throw new Error(`Image editing failed: ${error.message}`);
+        }
     });
     async function main() {
         const transport = new StdioServerTransport();

package/dist/mcp/imageEditingServer.js ADDED Viewed

@@ -0,0 +1,294 @@
+/**
+ * MCP Server для редактирования изображений с использованием Chutes AI API
+ * Реализация на TypeScript без использования Python
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import axios from 'axios';
+import fs from 'fs-extra';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { z } from 'zod';
+// Получаем __dirname в ES модуле
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+class ImageEditor {
+    apiToken;
+    defaultSettings;
+    constructor() {
+        this.apiToken = process.env.CHUTES_API_TOKEN || "";
+        // Настройки по умолчанию
+        this.defaultSettings = {
+            width: parseInt(process.env.EDIT_WIDTH || "1024"),
+            height: parseInt(process.env.EDIT_HEIGHT || "1024"),
+            true_cfg_scale: parseFloat(process.env.EDIT_CFG_SCALE || "4"),
+            negative_prompt: process.env.EDIT_NEGATIVE_PROMPT || "",
+            num_inference_steps: parseInt(process.env.EDIT_INFERENCE_STEPS || "50"),
+            seed: null
+        };
+    }
+    async editImage(prompt, imageB64, additionalParams = {}) {
+        if (!this.apiToken) {
+            throw new Error("API токен не настроен. Установите переменную окружения CHUTES_API_TOKEN");
+        }
+        if (!prompt.trim()) {
+            throw new Error("Пустой промпт недопустим");
+        }
+        if (!imageB64.trim()) {
+            throw new Error("Изображение для редактирования не предоставлено");
+        }
+        // Объединяем настройки по умолчанию с переданными параметрами
+        const settings = {
+            ...this.defaultSettings,
+            ...additionalParams
+        };
+        const headers = {
+            "Authorization": `Bearer ${this.apiToken}`,
+            "Content-Type": "application/json",
+        };
+        const body = {
+            seed: settings.seed,
+            width: settings.width,
+            height: settings.height,
+            prompt: prompt.trim(),
+            image_b64: imageB64,
+            true_cfg_scale: settings.true_cfg_scale,
+            negative_prompt: settings.negative_prompt,
+            num_inference_steps: settings.num_inference_steps
+        };
+        try {
+            const response = await axios.post("https://chutes-qwen-image-edit.chutes.ai/generate", body, {
+                headers,
+                responseType: 'arraybuffer',
+                timeout: 120000 // Редактирование может занимать больше времени
+            });
+            if (response.status !== 200) {
+                throw new Error(`API Error ${response.status}: ${response.statusText}`);
+            }
+            // Проверяем content-type ответа
+            const contentType = response.headers['content-type'] || '';
+            if (contentType.includes('application/json')) {
+                // Если JSON ответ
+                const result = response.data;
+                // Извлекаем отредактированное изображение из ответа
+                if (result.image) {
+                    return result.image;
+                }
+                else if (result.data) {
+                    return result.data;
+                }
+                else {
+                    throw new Error(`Неожиданный формат ответа API: ${JSON.stringify(result)}`);
+                }
+            }
+            else if (contentType.includes('image/')) {
+                // Если изображение напрямую
+                const imageData = response.data;
+                if (!imageData || imageData.length < 2) {
+                    throw new Error("Пустые или некорректные данные изображения");
+                }
+                // Конвертируем в base64
+                return imageData.toString('base64');
+            }
+            else {
+                throw new Error(`Неожиданный content-type: ${contentType}`);
+            }
+        }
+        catch (error) {
+            if (axios.isAxiosError(error)) {
+                if (error.code === 'ECONNABORTED') {
+                    throw new Error("Таймаут при редактировании изображения");
+                }
+                if (error.response) {
+                    throw new Error(`API Error ${error.response.status}: ${error.response.data}`);
+                }
+            }
+            throw new Error(`Ошибка сетевого соединения: ${error.message}`);
+        }
+    }
+}
+// Создаем экземпляр сервера
+const app = new McpServer({
+    name: "image-editor-ts",
+    version: "1.0.0",
+}, {
+    capabilities: {
+        tools: {}
+    }
+});
+const imageEditor = new ImageEditor();
+// Регистрируем инструменты
+app.registerTool("edit_image", {
+    description: "Редактирует изображение на основе текстового описания желаемых изменений. ВАЖНО: промпт и негативный промпт должны быть СТРОГО на английском языке!",
+    inputSchema: {
+        prompt: z.string().describe("Текстовое описание желаемых изменений к изображению НА АНГЛИЙСКОМ ЯЗЫКЕ"),
+        image_b64: z.string().describe("Исходное изображение в формате base64 (без data URI префикса)"),
+        width: z.number().optional().describe("Ширина результата (512-2048)"),
+        height: z.number().optional().describe("Высота результата (512-2048)"),
+        true_cfg_scale: z.number().optional().describe("Сила следования промпту (1.0-10.0)"),
+        negative_prompt: z.string().optional().describe("Негативный промпт (что НЕ должно быть в результате)"),
+        num_inference_steps: z.number().optional().describe("Количество шагов обработки (10-100)"),
+        seed: z.number().optional().describe("Seed для воспроизводимости (оставьте пустым для случайного)")
+    },
+}, async (args) => {
+    try {
+        // Извлекаем параметры
+        const prompt = args.prompt || "";
+        const imageB64 = args.image_b64 || "";
+        if (!prompt) {
+            throw new Error("Параметр 'prompt' обязателен");
+        }
+        if (!imageB64) {
+            throw new Error("Параметр 'image_b64' обязателен");
+        }
+        // Дополнительные параметры (необязательные)
+        const editParams = {};
+        if (args.width !== undefined)
+            editParams.width = args.width;
+        if (args.height !== undefined)
+            editParams.height = args.height;
+        if (args.true_cfg_scale !== undefined)
+            editParams.true_cfg_scale = args.true_cfg_scale;
+        if (args.negative_prompt !== undefined)
+            editParams.negative_prompt = args.negative_prompt;
+        if (args.num_inference_steps !== undefined)
+            editParams.num_inference_steps = args.num_inference_steps;
+        if (args.seed !== undefined)
+            editParams.seed = args.seed;
+        // Редактируем изображение
+        const editedImageB64 = await imageEditor.editImage(prompt, imageB64, editParams);
+        // Возвращаем результат
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `✅ Изображение успешно отредактировано по промпту: '${prompt}'`
+                },
+                {
+                    type: "image",
+                    data: editedImageB64,
+                    mimeType: "image/jpeg"
+                }
+            ]
+        };
+    }
+    catch (error) {
+        // Возвращаем ошибку как текстовый контент
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `❌ Ошибка при редактировании изображения: ${error.message}`
+                }
+            ]
+        };
+    }
+});
+app.registerTool("edit_image_file", {
+    description: "Редактирует изображение из файла на основе текстового описания желаемых изменений. ВАЖНО: промпт и негативный промпт должны быть СТРОГО на английском языке!",
+    inputSchema: {
+        prompt: z.string().describe("Текстовое описание желаемых изменений к изображению НА АНГЛИЙСКОМ ЯЗЫКЕ"),
+        image_path: z.string().describe("Путь к исходному файлу изображения"),
+        output_path: z.string().describe("Путь для сохранения отредактированного изображения"),
+        width: z.number().optional().describe("Ширина результата (512-2048)"),
+        height: z.number().optional().describe("Высота результата (512-2048)"),
+        true_cfg_scale: z.number().optional().describe("Сила следования промпту (1.0-10.0)"),
+        negative_prompt: z.string().optional().describe("Негативный промпт (что НЕ должно быть в результате)"),
+        num_inference_steps: z.number().optional().describe("Количество шагов обработки (10-100)"),
+        seed: z.number().optional().describe("Seed для воспроизводимости (оставьте пустым для случайного)")
+    },
+}, async (args) => {
+    try {
+        // Извлекаем параметры
+        const prompt = args.prompt || "";
+        const imagePath = args.image_path || "";
+        const outputPath = args.output_path || "";
+        if (!prompt) {
+            throw new Error("Параметр 'prompt' обязателен");
+        }
+        if (!imagePath) {
+            throw new Error("Параметр 'image_path' обязателен");
+        }
+        if (!outputPath) {
+            throw new Error("Параметр 'output_path' обязателен");
+        }
+        // Обрабатываем путь к исходному файлу с учетом базового каталога
+        const baseSaveDirectory = process.env.IMG_SAVE_BASE_DIR || "";
+        const fullImagePath = baseSaveDirectory && !path.isAbsolute(imagePath)
+            ? path.normalize(path.join(baseSaveDirectory, imagePath))
+            : path.normalize(imagePath);
+        // Проверяем существование входного файла
+        if (!await fs.pathExists(fullImagePath)) {
+            throw new Error(`Файл изображения не найден: ${fullImagePath}`);
+        }
+        // Загружаем изображение и конвертируем в base64
+        const imageBuffer = await fs.readFile(fullImagePath);
+        const imageB64 = imageBuffer.toString('base64');
+        // Дополнительные параметры (необязательные)
+        const editParams = {};
+        if (args.width !== undefined)
+            editParams.width = args.width;
+        if (args.height !== undefined)
+            editParams.height = args.height;
+        if (args.true_cfg_scale !== undefined)
+            editParams.true_cfg_scale = args.true_cfg_scale;
+        if (args.negative_prompt !== undefined)
+            editParams.negative_prompt = args.negative_prompt;
+        if (args.num_inference_steps !== undefined)
+            editParams.num_inference_steps = args.num_inference_steps;
+        if (args.seed !== undefined)
+            editParams.seed = args.seed;
+        // Редактируем изображение через base64 метод
+        const editedImageB64 = await imageEditor.editImage(prompt, imageB64, editParams);
+        // Декодируем и сохраняем результат
+        const editedImageBuffer = Buffer.from(editedImageB64, 'base64');
+        // Обрабатываем путь к выходному файлу с учетом базового каталога
+        const fullOutputPath = baseSaveDirectory && !path.isAbsolute(outputPath)
+            ? path.normalize(path.join(baseSaveDirectory, outputPath))
+            : path.normalize(outputPath);
+        // Создаем папку для выходного файла если нужно
+        const outputDir = path.dirname(fullOutputPath);
+        if (outputDir && !await fs.pathExists(outputDir)) {
+            await fs.ensureDir(outputDir);
+        }
+        await fs.writeFile(fullOutputPath, editedImageBuffer);
+        // Создаем file URI для выходного файла
+        const fileUrl = new URL(`file://${path.resolve(fullOutputPath)}`).href;
+        // Возвращаем результат
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `✅ Изображение успешно отредактировано: '${fullImagePath}' -> '${fullOutputPath}' (URI: ${fileUrl})`
+                }
+            ]
+        };
+    }
+    catch (error) {
+        // Возвращаем ошибку как текстовый контент
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `❌ Ошибка при редактировании файла изображения: ${error.message}`
+                }
+            ]
+        };
+    }
+});
+async function main() {
+    try {
+        const transport = new StdioServerTransport();
+        await app.connect(transport);
+        console.log("MCP Image Editing Server (TypeScript) is running...");
+    }
+    catch (error) {
+        console.error("Server error:", error);
+        process.exit(1);
+    }
+}
+// Запускаем сервер если этот файл запущен напрямую
+if (import.meta.url === `file://${process.argv[1]}`) {
+    main();
+}
+export { main, ImageEditor };

package/dist/mcp/imageGenerationServer.js ADDED Viewed

@@ -0,0 +1,270 @@
+/**
+ * MCP Server для генерации изображений с использованием Chutes AI API
+ * Реализация на TypeScript без использования Python
+ */
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import axios from 'axios';
+import fs from 'fs-extra';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { z } from 'zod';
+// Получаем __dirname в ES модуле
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+class ImageGenerator {
+    apiToken;
+    defaultSettings;
+    constructor() {
+        this.apiToken = process.env.CHUTES_API_TOKEN || "";
+        // Настройки по умолчанию
+        this.defaultSettings = {
+            model: process.env.SD_MODEL || "JuggernautXL",
+            width: this.validateRange(parseInt(process.env.SD_WIDTH || "1024"), 128, 2048),
+            height: this.validateRange(parseInt(process.env.SD_HEIGHT || "1024"), 128, 2048),
+            guidance_scale: this.validateRange(parseFloat(process.env.SD_GUIDANCE_SCALE || "7.5"), 1.0, 20.0),
+            negative_prompt: process.env.SD_NEGATIVE_PROMPT || "",
+            num_inference_steps: this.validateRange(parseInt(process.env.SD_NUM_INFERENCE_STEPS || "25"), 1, 50),
+            seed: Math.max(0, parseInt(process.env.SD_SEED || "0")) || null
+        };
+    }
+    validateRange(value, minVal, maxVal) {
+        return Math.max(minVal, Math.min(value, maxVal));
+    }
+    async generateImage(prompt, additionalParams = {}) {
+        if (!this.apiToken) {
+            throw new Error("API токен не настроен. Установите переменную окружения CHUTES_API_TOKEN");
+        }
+        if (!prompt.trim()) {
+            throw new Error("Пустой промпт недопустим");
+        }
+        // Объединяем настройки по умолчанию с переданными параметрами
+        const settings = {
+            ...this.defaultSettings,
+            ...additionalParams
+        };
+        const headers = {
+            "Authorization": `Bearer ${this.apiToken}`,
+            "Content-Type": "application/json",
+        };
+        const body = {
+            model: settings.model,
+            prompt: prompt.trim(),
+            width: settings.width,
+            height: settings.height,
+            guidance_scale: settings.guidance_scale,
+            negative_prompt: settings.negative_prompt,
+            num_inference_steps: settings.num_inference_steps,
+            seed: settings.seed !== 0 ? settings.seed : null
+        };
+        try {
+            const response = await axios.post("https://image.chutes.ai/generate", body, {
+                headers,
+                responseType: 'arraybuffer',
+                timeout: 60000
+            });
+            if (response.status !== 200) {
+                throw new Error(`API Error ${response.status}: ${response.statusText}`);
+            }
+            const imageData = response.data;
+            if (!imageData || imageData.length < 2) {
+                throw new Error("Пустые или некорректные данные изображения");
+            }
+            // Проверяем JPEG signature
+            if (imageData[0] !== 0xFF || imageData[1] !== 0xD8) {
+                console.warn("Предупреждение: Данные могут не быть JPEG");
+            }
+            const base64Image = imageData.toString('base64');
+            return `data:image/jpeg;base64,${base64Image}`;
+        }
+        catch (error) {
+            if (axios.isAxiosError(error)) {
+                if (error.code === 'ECONNABORTED') {
+                    throw new Error("Таймаут при генерации изображения");
+                }
+                if (error.response) {
+                    throw new Error(`API Error ${error.response.status}: ${error.response.data}`);
+                }
+            }
+            throw new Error(`Ошибка сетевого соединения: ${error.message}`);
+        }
+    }
+}
+// Создаем экземпляр сервера
+const app = new McpServer({
+    name: "image-generator-ts",
+    version: "1.0.0",
+}, {
+    capabilities: {
+        tools: {}
+    }
+});
+const imageGen = new ImageGenerator();
+// Регистрируем инструменты
+app.registerTool("generate_image", {
+    description: "Генерирует изображения. ВАЖНО: промпт и негативный промпт должны быть СТРОГО на английском языке!",
+    inputSchema: {
+        prompt: z.string().describe("Текстовый промпт для генерации изображения НА АНГЛИЙСКОМ ЯЗЫКЕ"),
+        width: z.number().optional().describe("Ширина изображения (128-2048)"),
+        height: z.number().optional().describe("Высота изображения (128-2048)"),
+        guidance_scale: z.number().optional().describe("Сила следования промпту (1.0-20.0)"),
+        negative_prompt: z.string().optional().describe("Негативный промпт (что НЕ включать в изображение)"),
+        num_inference_steps: z.number().optional().describe("Количество шагов генерации (1-50)"),
+        seed: z.number().optional().describe("Seed для воспроизводимости (0 = случайный)")
+    },
+}, async (args) => {
+    try {
+        // Извлекаем параметры
+        const prompt = args.prompt || "";
+        if (!prompt) {
+            throw new Error("Параметр 'prompt' обязателен");
+        }
+        // Дополнительные параметры (необязательные)
+        const generationParams = {};
+        if (args.width !== undefined)
+            generationParams.width = args.width;
+        if (args.height !== undefined)
+            generationParams.height = args.height;
+        if (args.guidance_scale !== undefined)
+            generationParams.guidance_scale = args.guidance_scale;
+        if (args.negative_prompt !== undefined)
+            generationParams.negative_prompt = args.negative_prompt;
+        if (args.num_inference_steps !== undefined)
+            generationParams.num_inference_steps = args.num_inference_steps;
+        if (args.seed !== undefined)
+            generationParams.seed = args.seed;
+        // Генерируем изображение
+        const imageDataUri = await imageGen.generateImage(prompt, generationParams);
+        // Извлекаем только base64 данные без префикса
+        let base64Data = imageDataUri;
+        if (imageDataUri.startsWith("data:image/jpeg;base64,")) {
+            base64Data = imageDataUri.substring("data:image/jpeg;base64,".length);
+        }
+        // Возвращаем результат
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `✅ Изображение успешно сгенерировано по промпту: '${prompt}'`
+                },
+                {
+                    type: "image",
+                    data: base64Data,
+                    mimeType: "image/jpeg"
+                }
+            ]
+        };
+    }
+    catch (error) {
+        // Возвращаем ошибку как текстовый контент
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `❌ Ошибка при генерации изображения: ${error.message}`
+                }
+            ]
+        };
+    }
+});
+app.registerTool("generate_image_to_file", {
+    description: "Генерирует изображение и сохраняет его в указанный файл. ВАЖНО: промпт и негативный промпт должны быть СТРОГО на английском языке!",
+    inputSchema: {
+        prompt: z.string().describe("Текстовый промпт для генерации изображения НА АНГЛИЙСКОМ ЯЗЫКЕ"),
+        directory: z.string().describe("Каталог для сохранения изображения"),
+        filename: z.string().describe("Имя файла для сохранения (с расширением)"),
+        width: z.number().optional().describe("Ширина изображения (128-2048)"),
+        height: z.number().optional().describe("Высота изображения (128-2048)"),
+        guidance_scale: z.number().optional().describe("Сила следования промпту (1.0-20.0)"),
+        negative_prompt: z.string().optional().describe("Негативный промпт (что НЕ включать в изображение)"),
+        num_inference_steps: z.number().optional().describe("Количество шагов генерации (1-50)"),
+        seed: z.number().optional().describe("Seed для воспроизводимости (0 = случайный)")
+    },
+}, async (args) => {
+    try {
+        // Извлекаем параметры
+        const prompt = args.prompt || "";
+        const directory = args.directory || "";
+        const filename = args.filename || "";
+        if (!prompt) {
+            throw new Error("Параметр 'prompt' обязателен");
+        }
+        if (!directory) {
+            throw new Error("Параметр 'directory' обязателен");
+        }
+        if (!filename) {
+            throw new Error("Параметр 'filename' обязателен");
+        }
+        // Дополнительные параметры (необязательные)
+        const generationParams = {};
+        if (args.width !== undefined)
+            generationParams.width = args.width;
+        if (args.height !== undefined)
+            generationParams.height = args.height;
+        if (args.guidance_scale !== undefined)
+            generationParams.guidance_scale = args.guidance_scale;
+        if (args.negative_prompt !== undefined)
+            generationParams.negative_prompt = args.negative_prompt;
+        if (args.num_inference_steps !== undefined)
+            generationParams.num_inference_steps = args.num_inference_steps;
+        if (args.seed !== undefined)
+            generationParams.seed = args.seed;
+        // Генерируем изображение
+        const imageDataUri = await imageGen.generateImage(prompt, generationParams);
+        // Извлекаем только base64 данные без префикса
+        let base64Data = imageDataUri;
+        if (imageDataUri.startsWith("data:image/jpeg;base64,")) {
+            base64Data = imageDataUri.substring("data:image/jpeg;base64,".length);
+        }
+        // Проверяем базовый каталог из переменной окружения
+        const baseSaveDirectory = process.env.IMG_SAVE_BASE_DIR || "";
+        const fullDirectory = baseSaveDirectory
+            ? path.normalize(path.join(baseSaveDirectory, directory))
+            : path.normalize(directory);
+        // Создаем каталог если не существует
+        await fs.ensureDir(fullDirectory);
+        // Формируем полный путь к файлу
+        const outputPath = path.join(fullDirectory, filename);
+        // Сохраняем изображение
+        const imageBuffer = Buffer.from(base64Data, 'base64');
+        await fs.writeFile(outputPath, imageBuffer);
+        // Создаем file URI
+        const fileUrl = new URL(`file://${path.resolve(outputPath)}`).href;
+        // Возвращаем результат
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `✅ Изображение успешно сгенерировано и сохранено в '${outputPath}' (URI: ${fileUrl})`
+                }
+            ]
+        };
+    }
+    catch (error) {
+        // Возвращаем ошибку как текстовый контент
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: `❌ Ошибка при генерации изображения в файл: ${error.message}`
+                }
+            ]
+        };
+    }
+});
+async function main() {
+    try {
+        const transport = new StdioServerTransport();
+        await app.connect(transport);
+        console.log("MCP Image Generation Server (TypeScript) is running...");
+    }
+    catch (error) {
+        console.error("Server error:", error);
+        process.exit(1);
+    }
+}
+// Запускаем сервер если этот файл запущен напрямую
+if (import.meta.url === `file://${process.argv[1]}`) {
+    main();
+}
+export { main, ImageGenerator };

package/lizard_wedding.png ADDED Viewed

Binary file

package/lizard_wedding_bitter.jpg ADDED Viewed

Binary file

package/moose_image.png ADDED Viewed

Binary file

package/package.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "name": "@vint.tri/report_gen_mcp",
-  "version": "1.3.9",
+  "version": "1.4.1",
   "description": "CLI tool for generating HTML reports with embedded charts and images",
   "main": "dist/index.js",
   "bin": {
@@ -16,6 +16,7 @@
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.0.3",
+    "axios": "^1.7.9",
     "chart.js": "^4.4.4",
     "commander": "^12.1.0",
     "ejs": "^3.1.10",