Earlier we were using codelens to display the suggestions in in VS-code extension, this was displaying outputs above every line which needed changes as we saw in the demo

but imo its creating more clutters in VS-code editor. So i am thinking of shifting to vscode diagnostics feature, this will just print a wavy line below the code suggesting and on hovering over them we can see the suggested changes, demo is provided in the attached video