#!/bin/bash
echo "Testing upscaling functionality..."
# Test upscaling with the MCP server
docker exec mcp-comfyui-flux-mcp-server-1 node -e "
const { connectToComfyUI, queuePrompt, getHistory, disconnect } = require('./src/comfyui-client.js');
const { getUpscalingWorkflow } = require('./src/workflows/upscaling.js');
async function test() {
try {
await connectToComfyUI('comfyui', 8188);
console.log('Connected to ComfyUI');
const workflow = getUpscalingWorkflow('flux_output_00001_.png', '4x-UltraSharp.safetensors', 1.0);
const promptId = await queuePrompt(workflow);
console.log('Queued upscaling job:', promptId);
// Wait for completion
let completed = false;
let attempts = 0;
while (!completed && attempts < 30) {
await new Promise(r => setTimeout(r, 1000));
const history = await getHistory(promptId);
if (history && history[promptId] && history[promptId].outputs) {
const outputs = history[promptId].outputs;
if (outputs['5'] && outputs['5'].images && outputs['5'].images.length > 0) {
console.log('Upscaling completed!');
console.log('Output:', outputs['5'].images[0].filename);
completed = true;
}
}
attempts++;
}
if (!completed) {
console.log('Timeout waiting for upscaling');
}
} catch (error) {
console.error('Error:', error.message);
} finally {
disconnect();
}
}
test();
"
echo "Test complete!"
