diff options
Diffstat (limited to 'src/mcp_sd15/tools/generate_guided_image.py')
| -rw-r--r-- | src/mcp_sd15/tools/generate_guided_image.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/mcp_sd15/tools/generate_guided_image.py b/src/mcp_sd15/tools/generate_guided_image.py new file mode 100644 index 0000000..c69286c --- /dev/null +++ b/src/mcp_sd15/tools/generate_guided_image.py @@ -0,0 +1,23 @@ +from __future__ import annotations + +from ..schema import GenerationResult, GuidedImageRequest +from ..service import ImageGenerationService + + +async def generate_guided_image( + service: ImageGenerationService, + positive_prompt: str, + control_mode: str, + input_image: str, + negative_prompt: str | None = None, + size: str | None = None, +) -> GenerationResult: + request = GuidedImageRequest.create( + positive_prompt, + control_mode, + input_image, + negative_prompt, + size, + default_size=service.config.generation.default_size, + ) + return await service.generate_guided_image(request) |
