If everything off shift is off the record mate, when you both finish work tomorrow keep it professional on shift but as soon as you off shift and off the premises let her know she cannot speak to you like that in no uncertain terms.