I have a Boss DS1 which is excellent for heavier stuff.
I also have an EHX Soul Food which is really nice for more subtle, less “grainy” drive.
I’d recommend either or both of those but going off your post, it sounds like the EHX would be just the ticket