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