Absent minded in salaah;talking to girl one is interested in marrying

As Salaamu Alaikum

Hope you are keeping well Mufti Saab.
I have two Questions

My problem is that I am very, very absent minded in Salaah. I almost can never remember how many rakaats I have read towards the end of the Salaah and I end up having to read Sajadh Sahw almost always and sometimes I cant remember if I read Sajdah Sahw and I read it again. Is this the correct way or is there some concession in the matter as this frustrates me sometimes.