Looking for a niche use case

I'm looking for an AI that can take a floor plan of an apartment building and return a floor plan that has absolutely everything removed except for exterior walls, corridor walls, unit demising walls, and the unit number. Kinda running into roadblocks with this one. Common issues are models saying yes I can do that and ending up with circular answers and no solution