Yep - If it was me, I would try a new battery with that controller, if that didn't work, then buy a matching controller/display/ pedal sensor
There is lots of options for 36V batteries - bag battery for lightest weight , rack battery if you have rack mounting points, seat post battery, bottle...