Add files via upload

modules/immersive-mode.js

@@ -159,19 +159,28 @@ function toggleImmersiveMode() {
 
 function bindMessageEvents() {
     if (state.messageEventsBound) return;
-
- const refreshOnAI = () => state.isActive && updateMessageDisplay();
-
- messageEvents.on(event_types.MESSAGE_SENT, () => {});
- messageEvents.on(event_types.MESSAGE_RECEIVED, refreshOnAI);
- messageEvents.on(event_types.MESSAGE_DELETED, refreshOnAI);
- messageEvents.on(event_types.MESSAGE_UPDATED, refreshOnAI);
- messageEvents.on(event_types.MESSAGE_SWIPED, refreshOnAI);
- if (event_types.GENERATION_STARTED) {
-   messageEvents.on(event_types.GENERATION_STARTED, () => {});
+    const onUserMessage = () => {
+        if (!state.isActive) return;
+        updateMessageDisplay();
+        scrollToBottom();
+    };
+    const onAIMessage = () => {
+        if (!state.isActive) return;
+        updateMessageDisplay();
+        if (getSettings().autoJumpOnAI) {
+            scrollToBottom();
         }
- messageEvents.on(event_types.GENERATION_ENDED, refreshOnAI);
-
+    };
+    const onMessageChange = () => {
+        if (!state.isActive) return;
+        updateMessageDisplay();
+    };
+    messageEvents.on(event_types.MESSAGE_SENT, onUserMessage);
+    messageEvents.on(event_types.MESSAGE_RECEIVED, onAIMessage);
+    messageEvents.on(event_types.MESSAGE_DELETED, onMessageChange);
+    messageEvents.on(event_types.MESSAGE_UPDATED, onMessageChange);
+    messageEvents.on(event_types.MESSAGE_SWIPED, onMessageChange);
+    messageEvents.on(event_types.GENERATION_ENDED, onAIMessage);
     state.messageEventsBound = true;
 }
 
@@ -269,8 +278,12 @@ function showSingleModeMessages() {
 
         const $prevMessage = $targetAI.prevAll('.mes').first();
         if ($prevMessage.length) {
+
+            const isUserMessage = $prevMessage.attr('is_user') === 'true';
+            if (isUserMessage) {
                 $prevMessage.show();
             }
+        }
 
         $targetAI.nextAll('.mes').show();
 
@@ -377,13 +390,14 @@ function updateSwipesCounter($targetMes) {
  }
  $swipesCounter.html('1​/​1');
 }
-
+function scrollToBottom() {
     const chatContainer = document.getElementById('chat');
-    const chatContainer = document.getElementById('chat');
+    if (!chatContainer) return;
+    
+    chatContainer.scrollTop = chatContainer.scrollHeight;
     requestAnimationFrame(() => {
         chatContainer.scrollTop = chatContainer.scrollHeight;
     });
-        });
 }
 function toggleDisplayMode() {
     if (!state.isActive) return;