Auto resize while also remaining within optimal SDXL resolutions

[lamb570]

I wrote a shortcode script to automatically resize while also staying at a total pixel height + width of 2048. This means that if you batch up a bunch of images and some are upscaled, photos, or whatever would lead to them not being an ideal SDXL resolution you can still utilize the automatic resize... It will convert something like 4096x4096 down to 1024x1024... This way you'll more likely sweet spot the image generation.

If @ThereforeGames or anyone else sees this, unprompt is awesome, and I would love to get this working in Forge since it's far faster.

I simply replaced my own autoresize python with this one... Happy to make a PR but I'm not sure this repo is active so wanted to share.

class Shortcode():
    def __init__(self, Unprompted):
        self.Unprompted = Unprompted
        self.description = "Automatically resizes images so that width + height = 2048 while maintaining aspect ratio."

    def run_atomic(self, pargs, kwargs, context):
        checking_image = self.Unprompted.parse_image_kwarg("input")
        if checking_image:
            this_width = checking_image.size[0]
            this_height = checking_image.size[1]
        elif "init_images" in self.Unprompted.shortcode_user_vars:
            this_width = self.Unprompted.shortcode_user_vars["init_images"][0].width
            this_height = self.Unprompted.shortcode_user_vars["init_images"][0].height
        else:
            self.log.error(f"Could not find initial image! Printing user vars for reference: {dir(self.Unprompted.shortcode_user_vars)}")
            return ""

        sd_unit = self.Unprompted.parse_advanced(kwargs["unit"], context) if "unit" in kwargs else 64
        only_full_res = self.Unprompted.parse_advanced(kwargs["only_full_res"], context) if "only_full_res" in kwargs else False

        if not only_full_res or "inpaint_full_res" not in self.Unprompted.shortcode_user_vars or not self.Unprompted.shortcode_user_vars["inpaint_full_res"]:
            # Calculate aspect ratio
            aspect_ratio = this_width / this_height

            # Solve for width and height such that width + height = 2048
            new_height = round(2048 / (1 + aspect_ratio))
            new_width = 2048 - new_height

            # Round to nearest sd_unit
            new_width = int(round(new_width / sd_unit) * sd_unit)
            new_height = int(round(new_height / sd_unit) * sd_unit)

            # Ensure dimensions do not exceed the original image (prevent upscaling)
            if new_width > this_width or new_height > this_height:
                scale_factor = min(this_width / new_width, this_height / new_height)
                new_width = int(new_width * scale_factor)
                new_height = int(new_height * scale_factor)

            self.Unprompted.shortcode_user_vars["width"] = new_width
            self.Unprompted.shortcode_user_vars["height"] = new_height

            self.log.debug(f"Resized image to: {new_width}x{new_height}")

        return ""

    def ui(self, gr):
        return [
            gr.Number(label="Rounding multiplier of output resolution (must be divisible by 8) 🡢 unit", value=64, interactive=True),
            gr.Checkbox(label="Only run this shortcode if using full resolution inpainting mode 🡢 only_full_res"),
        ]