Defensive Training

Hi!
First at all, thanks to all the Lichess team for this great community.
I love solving puzzles and after 5000 I missed some defensive problems.
I've read this post ( lichess.org/blog/U4sjakQAAEAAhH9d/how-training-puzzles-are-generated ) on how the puzzles are being generated and I want to suggest a way to get some defensive puzzles:
* Defend Against Mate in N:
To generate the "Mate in N" puzzles, the blog says:
"It's simple, a pre-processor examines a game from the lichess database, highlights where a game went from a centipawn evaluation to a "Mate in N" evaluation; then sends the position to the main program, which in turn generates a puzzle."
Could be possible to start in some moves before and evaluate if there are many options to avoid entering the mate line? If there is only one line to avoid mate, this could be a defensive puzzle?
What do you think about? Is this computable?

Another ¿¿possible?? way:
Find games where taking a piece leads to a "Checkmate is now unavoidable" position.
Send the previous position (before taking the piece) to Stockfish to evaluate alternative moves.
List the moves that avoid mate, and sort them by centipawn evaluation.
Positions that have only one move that avoids mate or when the list has only one move that keeps or increments the previous centipawn evaluation could be "Defensive puzzles" or, at least, "Trap avoiding puzzles".