AI assisted keystone

I LOVE the auto keystone, when it works. The manual keystone process is painful and slow. What if there was a middle ground. When auto fails, it picks the wrong strong edges to straighten off of. I would love to loosely identify which straight lines to align based off of ... thinking just like AI uses subject detection, it could use edge detection ... or I could loosely "paint a selection" over the edge I want it to use.

Bonus points, this would give your auto alignment AI more training data to be better.