If you are looking for more light, then look for an LED replacement bulb. To replace the whole assembly, well, I'd look at an aftermarket off road lamp in similar size. Or if you have a cycle salvage yard in the area you would have quite a few options.
Neither of my bikes came with lamps originally, so I just used off the shelf lamps for head/tail lamps.