summaryrefslogtreecommitdiff
path: root/src/mcp_sd15/tools/generate_guided_image.py
blob: c69286c2b3daa2e21f287193a48ad968db7fe513 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
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)