From 30997c20d707a0b70c91c97de5758e0d346d66dd Mon Sep 17 00:00:00 2001
From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com>
Date: Wed, 8 Apr 2026 15:29:18 -0500
Subject: [PATCH] Render anchor badge and similarity scores in chat results

---
 .../chat/ChatEventThumbnailsRow.tsx           | 79 +++++++++++++------
 web/src/pages/Chat.tsx                        | 12 +++
 2 files changed, 67 insertions(+), 24 deletions(-)

diff --git a/web/src/components/chat/ChatEventThumbnailsRow.tsx b/web/src/components/chat/ChatEventThumbnailsRow.tsx
index bf2c5e88f..da26f8dd6 100644
--- a/web/src/components/chat/ChatEventThumbnailsRow.tsx
+++ b/web/src/components/chat/ChatEventThumbnailsRow.tsx
@@ -1,42 +1,73 @@
 import { useApiHost } from "@/api";
+import { useTranslation } from "react-i18next";
+import { cn } from "@/lib/utils";
+
+type ChatEvent = { id: string; score?: number };
 
 type ChatEventThumbnailsRowProps = {
-  events: { id: string }[];
+  events: ChatEvent[];
+  anchor?: { id: string } | null;
 };
 
 /**
- * Horizontal scroll row of event thumbnail images for chat (e.g. after search_objects).
- * Renders nothing when events is empty.
+ * Horizontal scroll row of event thumbnail images for chat.
+ * Optionally renders an anchor thumbnail with a "reference" badge above the
+ * results, and per-event similarity scores when provided.
+ * Renders nothing when there is nothing to show.
  */
 export function ChatEventThumbnailsRow({
   events,
+  anchor = null,
 }: ChatEventThumbnailsRowProps) {
   const apiHost = useApiHost();
+  const { t } = useTranslation(["views/chat"]);
 
-  if (events.length === 0) return null;
+  if (events.length === 0 && !anchor) return null;
+
+  const renderThumb = (event: ChatEvent, isAnchor = false) => (
+    <a
+      key={event.id}
+      href={`/explore?event_id=${event.id}`}
+      target="_blank"
+      rel="noopener noreferrer"
+      className={cn(
+        "relative aspect-square size-32 shrink-0 overflow-hidden rounded-lg",
+        isAnchor && "ring-2 ring-primary",
+      )}
+    >
+      <img
+        className="size-full object-cover"
+        src={`${apiHost}api/events/${event.id}/thumbnail.webp`}
+        alt=""
+        loading="lazy"
+      />
+      {typeof event.score === "number" && !isAnchor && (
+        <span className="absolute bottom-1 right-1 rounded bg-black/60 px-1 text-[10px] text-white">
+          {Math.round(event.score * 100)}%
+        </span>
+      )}
+      {isAnchor && (
+        <span className="absolute left-1 top-1 rounded bg-primary px-1 text-[10px] text-primary-foreground">
+          {t("anchor")}
+        </span>
+      )}
+    </a>
+  );
 
   return (
-    <div className="flex min-w-0 max-w-full flex-col gap-1 self-start">
-      <div className="scrollbar-container min-w-0 overflow-x-auto">
-        <div className="flex w-max gap-2">
-          {events.map((event) => (
-            <a
-              key={event.id}
-              href={`/explore?event_id=${event.id}`}
-              target="_blank"
-              rel="noopener noreferrer"
-              className="relative aspect-square size-32 shrink-0 overflow-hidden rounded-lg"
-            >
-              <img
-                className="size-full object-cover"
-                src={`${apiHost}api/events/${event.id}/thumbnail.webp`}
-                alt=""
-                loading="lazy"
-              />
-            </a>
-          ))}
+    <div className="flex min-w-0 max-w-full flex-col gap-2 self-start">
+      {anchor && (
+        <div className="scrollbar-container min-w-0 overflow-x-auto">
+          <div className="flex w-max gap-2">{renderThumb(anchor, true)}</div>
         </div>
-      </div>
+      )}
+      {events.length > 0 && (
+        <div className="scrollbar-container min-w-0 overflow-x-auto">
+          <div className="flex w-max gap-2">
+            {events.map((event) => renderThumb(event))}
+          </div>
+        </div>
+      )}
     </div>
   );
 }
diff --git a/web/src/pages/Chat.tsx b/web/src/pages/Chat.tsx
index bd154f1a4..bab238fa1 100644
--- a/web/src/pages/Chat.tsx
+++ b/web/src/pages/Chat.tsx
@@ -12,6 +12,7 @@ import { ChatStartingState } from "@/components/chat/ChatStartingState";
 import type { ChatMessage } from "@/types/chat";
 import {
   getEventIdsFromSearchObjectsToolCalls,
+  getFindSimilarObjectsFromToolCalls,
   streamChatCompletion,
 } from "@/utils/chatUtil";
 
@@ -161,6 +162,17 @@ export default function ChatPage() {
                     {msg.role === "assistant" &&
                       isComplete &&
                       (() => {
+                        const similar = getFindSimilarObjectsFromToolCalls(
+                          msg.toolCalls,
+                        );
+                        if (similar) {
+                          return (
+                            <ChatEventThumbnailsRow
+                              events={similar.results}
+                              anchor={similar.anchor}
+                            />
+                          );
+                        }
                         const events = getEventIdsFromSearchObjectsToolCalls(
                           msg.toolCalls,
                         );