I think it can be banned if you also don't criminalize it and implement some kind of harm-reduction system.

This way being caught with a cigarette won't land you in jail. At most you'll be fined (though ideally confiscation should be enough). And if you want to quit there are resources.

Heavy penalties should only be for trafficking, which you can achieve by stipulating an amount that counts as trafficking.