notification-page using react and vite

I was able to display the notification content on the notification menu only when the notification type is a message as a dropdown when the message is click. But i find it difficult to display the image when the notification type is comment on a photo.

Feedback on how i can do this is welcome.

Thanks