|
|
|
|
@ -5,6 +5,7 @@ from typing_extensions import Literal
|
|
|
|
|
from inflection import underscore
|
|
|
|
|
from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img
|
|
|
|
|
from modules.shared import sd_upscalers, opts, parser
|
|
|
|
|
from typing import List
|
|
|
|
|
|
|
|
|
|
API_NOT_ALLOWED = [
|
|
|
|
|
"self",
|
|
|
|
|
@ -108,12 +109,12 @@ StableDiffusionImg2ImgProcessingAPI = PydanticModelGenerator(
|
|
|
|
|
).generate_model()
|
|
|
|
|
|
|
|
|
|
class TextToImageResponse(BaseModel):
|
|
|
|
|
images: list[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
|
|
|
|
|
images: List[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
|
|
|
|
|
parameters: dict
|
|
|
|
|
info: str
|
|
|
|
|
|
|
|
|
|
class ImageToImageResponse(BaseModel):
|
|
|
|
|
images: list[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
|
|
|
|
|
images: List[str] = Field(default=None, title="Image", description="The generated image in base64 format.")
|
|
|
|
|
parameters: dict
|
|
|
|
|
info: str
|
|
|
|
|
|
|
|
|
|
@ -146,10 +147,10 @@ class FileData(BaseModel):
|
|
|
|
|
name: str = Field(title="File name")
|
|
|
|
|
|
|
|
|
|
class ExtrasBatchImagesRequest(ExtrasBaseRequest):
|
|
|
|
|
imageList: list[FileData] = Field(title="Images", description="List of images to work on. Must be Base64 strings")
|
|
|
|
|
imageList: List[FileData] = Field(title="Images", description="List of images to work on. Must be Base64 strings")
|
|
|
|
|
|
|
|
|
|
class ExtrasBatchImagesResponse(ExtraBaseResponse):
|
|
|
|
|
images: list[str] = Field(title="Images", description="The generated images in base64 format.")
|
|
|
|
|
images: List[str] = Field(title="Images", description="The generated images in base64 format.")
|
|
|
|
|
|
|
|
|
|
class PNGInfoRequest(BaseModel):
|
|
|
|
|
image: str = Field(title="Image", description="The base64 encoded PNG image")
|
|
|
|
|
|