Option to have wezterm only flatten if in the same pane

Would it be possible to have instances in wezterm only flatten if the neovim instance is in the same pane?

I really like the flatten feature for when I am using splits, but when I am using tabs it's quite confusing. It took me a while to realise that it was the flatten feature that made my neovim instances instantly close becasue I had one open in a different tab :sweat_smile: I could of course disable wezterm flattening, but it would be neat to have it for just splits, when the neovim instance is visible, and it is obvious what is happening.

Definitely should be doable by customizing the callbacks. I'll look into this soon!