This is not what I meant. What is being proved is: a^2-b^2 - (a+b)(a-b) = 0. If you swap a and b you end up with a sign switch on the lhs which is inconsequential.
That is not what the proof proves. The proof proves the equivalence how it was originally stated, and assumes for that b<a.
Your rewriting is of course true for all a,b and might be used in an algebraic proof. But this transformation is not at all shown in the geometric proof.
(You can exchange a and b in, say a^2+b^2, because 2^2+3^2=3^2+2^2)