#!/usr/bin/env python3
"""Проверка image-to-image: правка существующей картинки по референсу."""
import asyncio, os, sys
from mcp import ClientSession, StdioServerParameters
from mcp.client.stdio import stdio_client

PY = "/usr/local/lib/hermes-agent/venv/bin/python"
SERVER = "/opt/routerai-image-mcp/server.py"
REF = sys.argv[1]


async def main() -> None:
    params = StdioServerParameters(
        command=PY, args=[SERVER],
        env={**os.environ, "IMAGE_MCP_OUT_DIR": "/tmp/mcp-img-test"},
    )
    async with stdio_client(params) as (r, w):
        async with ClientSession(r, w) as s:
            await s.initialize()
            res = await s.call_tool("generate_image", {
                "prompt": "Оставь кота как есть, но поменяй фон на зимний: падает снег, ёлки.",
                "model": "google/gemini-3.1-flash-image",
                "reference_images": [REF],
                "filename": "kot-edit.png",
            })
            print(res.content[0].text)
            print("isError:", res.isError)


asyncio.run(main())