Sorry, it wouldn't have happened if you had done what you were asked with the phone and not made a big deal about others.
Was the FA wrong? Perhaps, but each of us should take responsibility to do everything that WE can to make for a better flight and environment.