Black Mirror, "Be Right Back", Season 2, Episode 1 [1] has this very plot.

[1] https://en.m.wikipedia.org/wiki/Be_Right_Back