https://www.google.com/search?q=chess+puzzle+mate+in+4&tbm=i...
It would be surprising if AI couldn't do the same search and produce a realistic drawing out of any one of the result puzzles.
If you want to be trivial about it, you can just have white back-rank mate with a rook, and black has 4 pieces to block with.