“I am okay with me,” Khalid wrote after a seeming former partner of Khalid’s shared a selfie with the singer