I would like to make changes to the following program, what should I do?

Question

I created a program that draws an additional square on the loaded image and then outputs it. I want to change the structure as follows. Where and how should I change it? Thi