Seeking Microcontroller Upgrade for 3D Printer with More Flash Memory

Currently engaged in refurbishing my 3D printer, I accidentally damaged the ATMEGA1284P-PU microcontroller. Despite the availability of this microcontroller and the open-source Marlin firmware it uses, its 128K of Flash memory has me contemplating an upgrade. Given the chance to replace the microcontroller, I'm seeking an equivalent alternative that aligns perfectly in terms of pin configuration but offers increased Flash memory. Any recommendations for such a microcontroller would be greatly appreciated. This inquiry is in the context of a RepRapPro Huxley 3D printer utilizing a Sanguinololu version 1.3a control board. Thanks in advance for any advice.