feat: enhance message rendering with rich markdown and styled code blocks

- Add styled code block headers with language badge (LANGUAGE: LUA)
- Add copy indicator and fold support for code blocks
- Add markdown highlighting (headers, bold, italic, lists, blockquotes)
- Add diff highlighting for +/- lines in diff blocks
- Add sender indicators (You/Assistant) before messages
- Add message dividers between answers
- Add inline code, links, task lists, strikethrough styling
- Remove context display from settings panel (now using @refs)
- Restore add_context/add_project_context for backwards compatibility
- Update README with new features and keybindings

Author: jackMort

README.md

@@ -32,6 +32,14 @@ test addition, code optimization, summarization, bug fixing, code explanation,
 Roxygen editing, and code readability analysis. Additionally, you can define
 your own custom actions using a JSON file.
 
+- **Rich Message Rendering**: Enhanced chat display with styled code blocks
+(language headers, copy indicators, foldable), markdown formatting (headers,
+bold, italic, lists, blockquotes, links), diff highlighting, and sender indicators.
+
+- **Inline Context References**: Use `@` to add context from LSP definitions or
+project files directly in your prompts. References are displayed inline and
+expanded when sending to the API.
+
 For a comprehensive understanding of the extension's functionality, you can watch
 a plugin showcase [video](https://www.youtube.com/watch?v=7k0KZsheLP4)
 
@@ -349,6 +357,9 @@ keybindings are available:
 - `<C-i>` [Edit Window] use response as input.
 - `<C-d>` [Edit Window] view the diff between left and right panes and use diff-mode
   commands
+- `y` [Chat] Copy code block at cursor.
+- `za` [Chat] Toggle fold for code block.
+- `@` [Chat] Trigger context autocomplete (LSP definitions, project context).
 
 The settings window (`<C-o>`) displays current configuration (model, temperature,
 max_tokens, session name). To change settings, modify your `setup()` configuration.

lua/chatgpt.lua

@@ -24,6 +24,29 @@ M.setup = function(options)
 
   vim.api.nvim_set_hl(0, "ChatGPTContextRef", { fg = "#89b4fa", bg = "#1e1e2e", bold = true, default = true })
 
+  vim.api.nvim_set_hl(0, "ChatGPTInlineCode", { fg = "#f38ba8", bg = "#1e1e2e", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTCodeBlock", { bg = "#181825", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTCodeBlockHeader", { fg = "#6c7086", bg = "#181825", italic = true, default = true })
+
+  vim.api.nvim_set_hl(0, "ChatGPTLink", { fg = "#89b4fa", underline = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTBold", { bold = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTItalic", { italic = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTHeader", { fg = "#cba6f7", bold = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTDivider", { fg = "#313244", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTListMarker", { fg = "#f9e2af", bold = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTBlockquote", { fg = "#a6adc8", italic = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTHRule", { fg = "#45475a", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTTaskDone", { fg = "#a6e3a1", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTTaskPending", { fg = "#f9e2af", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTStrikethrough", { strikethrough = true, fg = "#6c7086", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTLinkText", { fg = "#89dceb", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTLinkUrl", { fg = "#6c7086", underline = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTCodeLang", { fg = "#ffffff", bg = "#45475a", bold = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTDiffAdd", { fg = "#a6e3a1", bg = "#1e3a2f", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTDiffDel", { fg = "#f38ba8", bg = "#3a1e2f", default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTSenderUser", { fg = "#89b4fa", bold = true, default = true })
+  vim.api.nvim_set_hl(0, "ChatGPTSenderAssistant", { fg = "#a6e3a1", bold = true, default = true })
+
   vim.cmd("highlight default link ChatGPTSelectedMessage ColorColumn")
 
   config.setup(options)
@@ -57,4 +80,28 @@ end
 
 M.complete_code = module.complete_code
 
+-- Context APIs (deprecated but kept for backwards compatibility)
+M.add_context = function()
+  local lsp_context = require("chatgpt.context.lsp")
+  lsp_context.get_context(function(item)
+    if item then
+      local Context = require("chatgpt.context")
+      local ref = Context.make_ref(item)
+      Context.add(ref, item)
+      vim.notify(string.format("Context added: %s", ref), vim.log.levels.INFO)
+    end
+  end)
+end
+
+M.add_project_context = function()
+  local project_context = require("chatgpt.context.project")
+  local item = project_context.get_context()
+  if item then
+    local Context = require("chatgpt.context")
+    local ref = Context.make_ref(item)
+    Context.add(ref, item)
+    vim.notify(string.format("Context added: %s", ref), vim.log.levels.INFO)
+  end
+end
+
 return M

lua/chatgpt/config.lua

@@ -81,7 +81,8 @@ function M.defaults()
         toggle_message_role = "<C-r>",
         toggle_system_role_open = "<C-s>",
         stop_generating = "<C-x>",
-        delete_context_item = "d",
+        yank_code = "y",
+        toggle_fold = "za",
       },
     },
     popup_layout = {
@@ -107,6 +108,8 @@ function M.defaults()
         wrap = true,
         linebreak = true,
         foldcolumn = "1",
+        foldmethod = "manual",
+        foldenable = true,
         winhighlight = "Normal:Normal,FloatBorder:FloatBorder",
       },
       buf_options = {