The prophethood did come to an end. The job of the imam
Is not to receive divine message but to interpret the message of the prophet and the Quran. There is no issue against Imamate with this.
Yes angels comes to lady fatima doesn’t mean the revelation of the Quran is still coming down. The angels used to speak to maryam too, and she is not a prophet.