From 37ef8919455cde79e0ba3fe6d683a8fa845aecbb Mon Sep 17 00:00:00 2001
From: Haileyesus <118998054+blackmammoth@users.noreply.github.com>
Date: Mon, 29 Jun 2026 22:27:59 +0300
Subject: [PATCH] fix(chat): hide load all prompt after final page

---
 .../chat/hooks/useChatSessionState.ts          | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/src/components/chat/hooks/useChatSessionState.ts b/src/components/chat/hooks/useChatSessionState.ts
index 3330e21b..9103e28a 100644
--- a/src/components/chat/hooks/useChatSessionState.ts
+++ b/src/components/chat/hooks/useChatSessionState.ts
@@ -344,12 +344,11 @@ export function useChatSessionState({
             setHasMoreMessages(false);
             allMessagesLoadedRef.current = true;
             setAllMessagesLoaded(true);
-            if (!loadAllOverlayTimerRef.current) {
-              loadAllOverlayTimerRef.current = setTimeout(() => {
-                setShowLoadAllOverlay(false);
-                loadAllOverlayTimerRef.current = null;
-              }, 2500);
+            if (loadAllOverlayTimerRef.current) {
+              clearTimeout(loadAllOverlayTimerRef.current);
+              loadAllOverlayTimerRef.current = null;
             }
+            setShowLoadAllOverlay(false);
           }
           return false;
         }
@@ -361,12 +360,11 @@ export function useChatSessionState({
         if (!slot.hasMore) {
           allMessagesLoadedRef.current = true;
           setAllMessagesLoaded(true);
-          if (!loadAllOverlayTimerRef.current) {
-            loadAllOverlayTimerRef.current = setTimeout(() => {
-              setShowLoadAllOverlay(false);
-              loadAllOverlayTimerRef.current = null;
-            }, 2500);
+          if (loadAllOverlayTimerRef.current) {
+            clearTimeout(loadAllOverlayTimerRef.current);
+            loadAllOverlayTimerRef.current = null;
           }
+          setShowLoadAllOverlay(false);
         }
         return true;
       } finally {