1
0
Fork 0

fix thumbnail metadata size

This commit is contained in:
Sean Sube 2024-01-14 12:24:59 -06:00
parent 5e40ba949e
commit 943f0e6f04
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 29 additions and 0 deletions

View File

@ -252,6 +252,30 @@ class ImageMetadata:
return output_size return output_size
def with_args(
self,
params: Optional[ImageParams] = None,
size: Optional[Size] = None,
upscale: Optional[UpscaleParams] = None,
border: Optional[Border] = None,
highres: Optional[HighresParams] = None,
inversions: Optional[List[NetworkMetadata]] = None,
loras: Optional[List[NetworkMetadata]] = None,
models: Optional[List[NetworkMetadata]] = None,
ancestors: Optional[List["ImageMetadata"]] = None,
) -> "ImageMetadata":
return ImageMetadata(
params or self.params,
size or self.size,
upscale=upscale or self.upscale,
border=border or self.border,
highres=highres or self.highres,
inversions=inversions or self.inversions,
loras=loras or self.loras,
models=models or self.models,
ancestors=ancestors or self.ancestors,
)
@staticmethod @staticmethod
def from_exif(input: str) -> "ImageMetadata": def from_exif(input: str) -> "ImageMetadata":
lines = input.splitlines() lines = input.splitlines()

View File

@ -86,6 +86,11 @@ def add_thumbnail_output(
thumbnail = cover.copy() thumbnail = cover.copy()
thumbnail.thumbnail((server.thumbnail_size, server.thumbnail_size)) thumbnail.thumbnail((server.thumbnail_size, server.thumbnail_size))
metadata = images.metadata[0]
metadata = metadata.copy(
size=Size(server.thumbnail_size, server.thumbnail_size)
)
images.insert_image(0, thumbnail, images.metadata[0]) images.insert_image(0, thumbnail, images.metadata[0])