Run custom code after every generation?

[monophthongal]

I’m looking at building a tool that integrates with this UI and it needs to detect images as they’re generated.

It’s easy to run a standalone program that watches the output folder, but I was wondering if there is a standard way to do this within the automatic1111 UI?

(My understanding is only one custom script can run per generation, and I want to detect all generated images.)

Thank you for any help or advice!

[GeorgLegato]

you can do this by your own custom script, you just need to call proc.process_images(p)
then you can afterprocess either in memory iterating over images[] list or over stored files.
please check my txt2vectorgraphics script.
but chaining of scripts is not a concept yet.

[grexzen]

you can do this by your own custom script, you just need to call proc.process_images(p) then you can afterprocess either in memory iterating over images[] list or over stored files. please check my txt2vectorgraphics script. but chaining of scripts is not a concept yet.

Hey I checked out your script and it is awesome. Question: do you know how to create a function for SD that repeats generation after a conditional statement?

Like generate first image > then check condition > if condition not met > generate another image > condition met > save image > then generate next image > keep going until interrupt.

Seed is random, so it is just doing the same prompt and settings over and over.

Thanks!

[GeorgLegato]

you can script that, yes.
typically, every script calls "process_image(p)".
Where p contains all necessary "rendering" parameters, like seed, cfg, etc.
after calling "process_image" you get an list of images[].
Now, put a loop around that like

// pseudocode, only checking 1st image
while (not condition(image[0]) {
p.someAttribute = new value
process_image(p)
// and maybe delete stored images on disk, process_image does store to disk already
}

[cmp-nct]

I think chaining of scripts should become a concept, it's already needed and every week that passes makes it a more difficult topic.

"prompts from file -> prompt-expansion -> SD upscale" just as an example

But that would require a complete rewrite of the scripting interface with a lot more complexity than the current 'overwrite' approach.