summaryrefslogtreecommitdiff
path: root/src/mcp_sd15/tools/inpaint_image.py
blob: e1232ceddf9bd612a2dd19d365dfa4309869ef2c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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,
        base_positive=service.config.prompt.positive,
        base_negative=service.config.prompt.negative,
    )
    return await service.inpaint_image(request)