diff --git a/index.js b/index.js index aed6db1..78880bc 100644 --- a/index.js +++ b/index.js @@ -197,14 +197,22 @@ function processToolCallMessages() { // Create a container for the rendered markdown const renderContainer = $('
'); - // Add the markdown content (render the image) - renderContainer.html(tool.result); + // Convert markdown to actual HTML img tag + const markdownImgRegex = /!\[\]\(([^)]+)\)/; + const match = tool.result.match(markdownImgRegex); + if (match && match[1]) { + renderContainer.html(`Chat Bubble`); + } else { + renderContainer.html(tool.result); + } // Remove any existing rendered images $(this).find('.sillybubble-rendered-image').remove(); - // Add this image after the summary element - summaryElement.after(renderContainer); + // Add this image after the summary element, but only if details is present + if (summaryElement.length) { + summaryElement.after(renderContainer); + } // Mark this message as processed $(this).attr('data-sb-processed', 'true');