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