It's not great yet, but at least is not using a single color to paint all the image.
The AI will always have a favorite color, in this examples is blue, but I'm trying to give it a default warmer color. After that make another AI that let you select a pallet to paint it.
Will also train a model more focused on HQ after this start working better.