It’s easy to read the PT as Lucas dealing with the fallout of his divorce through the journey of Anakin.
Also, Temple of Doom is a “breakup script” in the same way Phil Collins’ Face Value is a “breakup album.” With both, the dust of the relationship hadn’t fully settled and a lot of raw and ugly feelings had to come out.