That's exactly the idea, it wouldn't kill them to just leave frogs and fireflies both in the game, but not have frogs eat the fireflies. 🧠 They did it with parrots and cookies back in like 2016.
And if you're thinking that this would make fireflies useless, it's just a really cool aesthetic and it sucks that Mojang cancelled it.