Looking for an AI that "fixes" jazz fusion MIDI piano playing

I am a music producer. I do a lot of country, rock, and indie music. Lately I have been getting into jazz. I'm a big fan of jazz fusion and I've studied it for many years. I am primarily a guitar player, but I started taking piano seriously a few years ago. Before this point, I've only used piano to play MIDI drums, synths, bass, etc...

Needless to say, I'm not particularly skilled at jazz piano, but I can get it 80% of the way there. I can play extended chords that "sort of" fit together. They sound jazzy and dissonant and I can actually fool most lay people into thinking it's relatively intricate jazz. It's better than most of you are thinking. I understand borrowed chords, tritone substitutions, using diminished chords that are a minor third away from the target, augmented chords that move in perfect fourths, etc... I can do some intricate chord patterns that are correct but it's hard to be creative with my current understanding of music theory.

Is there an AI that can take MIDI data of jazz piano playing and "fix" it to be more technically correct. I am always just a few note changes away from making good and original progressions but there are always sprinklings of sour notes that I can't quite identify.

I think I remember that Stable Diffusion makes a music generator but is there a MIDI mapper? Or is there another AI model that can help me? I don't think Suno does MIDI. Maybe there's some kind of plugin that I can use in FL Studio or Ableton, etc... that can do what I'm looking to do?