Refactor heatmap event handling to update x-axis labels for clarity and ensure heatmap is plotted after processing search group logic.

heatmap_event.js

@@ -121,12 +121,12 @@ export async function handleHeatmapEvent() {
             [1, ...order.slice(1).map(idx => idx === -1 ? 1 : cosine(refEmbed, groupEmbeddings[idx]))],
             ...order.slice(1).map(i => [cosine(refEmbed, groupEmbeddings[i]), ...order.slice(1).map(j => sim[i][j])])
         ];
-        xLabels = ["Search Group", ...order.slice(1).map(i => (clusterNames && clusterNames[i]) ? clusterNames[i] : `Group ${i + 1}`)];
+        xLabels = ["Search", ...order.slice(1).map(i => (clusterNames && clusterNames[i]) ? clusterNames[i] : `Group ${i + 1}`)];
     } else {
         simOrdered = order.map(i => order.map(j => sim[i][j]));
         xLabels = order.map(i => (clusterNames && clusterNames[i]) ? clusterNames[i] : `Group ${i + 1}`);
     }
-    plotHeatmap(simOrdered, xLabels, xLabels);
+    
 
     // If search group is provided and sorting by line, reorder lines in each group by similarity to search lines
     if (useSearchGroup && search_by_max_search_line && searchEmbeds.length > 0) {
@@ -160,5 +160,7 @@ export async function handleHeatmapEvent() {
         const finalText = buildFinalText(order[0] === -1 ? order.slice(1) : order, sortedLines, clusterNames, n);
         document.getElementById("input").value = finalText;
     }
+
+    plotHeatmap(simOrdered, xLabels, xLabels);
     progressBarInner.style.width = "100%";
 }