We should use the circuit with the diode not the "non-working " very problematic circuit.
As I said it isn't a good idea to use the output of the 555 for "auto-timing" purposes,like it is done here.
About your suggestion ,
we could ,up to a point.
As they say "there are many ways to skin a...