Before you start replacing components, try one thing, if you haven't already. Finding a video on how to remove the final drive cog assembly will help here, but it's quite easy to do.
Remove crank arms and pedals, remove chainring, remove the plastic gear cover and unscrew the axle locking nuts (long needle nose pliers work here!)
Pull out the gear assembly, making sure you don't lose the thrust bearings on either end.
Clean the PAS sensor on the motor body (grease from the final drive can get trapped here)
Clean the area of the PAS disc (big plastic disc on back of large cog)
Put back together, making sure you tighten the axle locking nuts just enough to stop any free play. Too tight and you risk crushing the thrust bearings (whoops, wonder who did that recently
)