Q: so I wonder if you can think about non-diatonic chords functionally. Do you think about them functionally? How do you know which chromatic chord is appropriate when?
A: the golden rule is, try it, and listen to the whole song and see if it makes it better or worse. this is a classic trick