From 27fd4b3a1490beba343eca4f73d10de35c7fbfca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=8C=97=E6=9E=B3?=
 <7854742+wang_rumeng@user.noreply.gitee.com>
Date: Sun, 24 Aug 2025 17:03:20 +0800
Subject: [PATCH] =?UTF-8?q?chatbox=20=E4=BF=AE=E5=A4=8D=E8=87=AA=E5=8A=A8?=
 =?UTF-8?q?=E6=BB=9A=E5=8A=A8=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 components/SmartChatBox/InputBar.tsx       |  7 +++-
 components/SmartChatBox/LoadMoreButton.tsx |  2 +-
 components/SmartChatBox/useMessages.ts     | 45 +++++++++++++---------
 3 files changed, 34 insertions(+), 20 deletions(-)

diff --git a/components/SmartChatBox/InputBar.tsx b/components/SmartChatBox/InputBar.tsx
index 40b358b..c95d197 100644
--- a/components/SmartChatBox/InputBar.tsx
+++ b/components/SmartChatBox/InputBar.tsx
@@ -44,14 +44,19 @@ export function InputBar({ onSend }: InputBarProps) {
 
   const onFileChange = async (e: React.ChangeEvent<HTMLInputElement>) => {
     const file = e.target.files?.[0];
+    // 保存对 input 元素的引用
+    const inputElement = e.currentTarget;
+    
     if (file) {
       if (!file.type.startsWith('image/')) {
         alert('请选择图片文件');
+        inputElement.value = ""; // 重置 input
         return;
       }
       await handleFileUpload(file);
+      // 使用保存的引用重置 input
+      inputElement.value = "";
     }
-    e.currentTarget.value = ""; // reset
   };
 
   const removeImage = () => {
diff --git a/components/SmartChatBox/LoadMoreButton.tsx b/components/SmartChatBox/LoadMoreButton.tsx
index 45884b7..6e98e98 100644
--- a/components/SmartChatBox/LoadMoreButton.tsx
+++ b/components/SmartChatBox/LoadMoreButton.tsx
@@ -11,7 +11,7 @@ export function LoadMoreButton({ onClick, loading = false }: LoadMoreButtonProps
     <button
       onClick={onClick}
       disabled={loading}
-      className="w-full flex items-center justify-center gap-2 text-sm text-gray-400 hover:text-gray-300 hover:bg-gray-800/30 transition-colors disabled:opacity-50"
+      className="w-full flex items-center justify-center gap-2 text-sm text-gray-400 hover:text-gray-300 hover:bg-gray-800/30 transition-colors disabled:opacity-50 disabled:cursor-not-allowed"
       data-alt="load-more-button"
     >
       {loading ? (
diff --git a/components/SmartChatBox/useMessages.ts b/components/SmartChatBox/useMessages.ts
index 50319fd..0699e15 100644
--- a/components/SmartChatBox/useMessages.ts
+++ b/components/SmartChatBox/useMessages.ts
@@ -143,6 +143,7 @@ export function useMessages({ config, onMessagesUpdate }: UseMessagesProps): [Me
     if (timeoutIdRef.current) {
       clearTimeout(timeoutIdRef.current);
     }
+    isPollingRef.current = true; // 设置轮询标志，防止其他轮询启动
     
     try {
       // 立即添加用户消息（临时显示）
@@ -157,13 +158,19 @@ export function useMessages({ config, onMessagesUpdate }: UseMessagesProps): [Me
 
       // 无论是否在查看历史，发送消息后都切换到最新视图
       isViewingHistoryRef.current = false;
-      setDisplayMessages(prev => mergeMessages(prev, [userMessage]));
+      setDisplayMessages(prev => {
+        const newMessages = mergeMessages(prev, [userMessage]);
+        // 在 setState 的回调中触发滚动，确保消息已更新到界面
+        setTimeout(() => onMessagesUpdate?.(true), 0);
+        return newMessages;
+      });
 
       // 发送到服务器
       await sendMessage(blocks, configRef.current);
       
       // 立即获取最新的消息列表
       await updateMessages(false);
+      onMessagesUpdate?.(true);
     } catch (err) {
       console.error("发送消息失败:", err);
       setError(err instanceof Error ? err : new Error("发送消息失败"));
@@ -172,9 +179,10 @@ export function useMessages({ config, onMessagesUpdate }: UseMessagesProps): [Me
       
       // 恢复轮询
       if (systemPush) {
+        isPollingRef.current = false;
+        // 延迟一个轮询间隔后再开始轮询
         timeoutIdRef.current = setTimeout(() => {
-          isPollingRef.current = false;
-          updateMessages(false);
+          poll();
         }, POLLING_INTERVAL);
       }
     }
@@ -223,24 +231,25 @@ export function useMessages({ config, onMessagesUpdate }: UseMessagesProps): [Me
     });
   }, [latestMessages, filterMessages]);
 
+  // 定义轮询函数
+  const poll = useCallback(async () => {
+    if (isPollingRef.current) return;
+    
+    try {
+      isPollingRef.current = true;
+      await updateMessages(false);
+    } finally {
+      isPollingRef.current = false;
+      if (systemPush) {
+        timeoutIdRef.current = setTimeout(poll, POLLING_INTERVAL);
+      }
+    }
+  }, [systemPush, updateMessages]);
+
   // 轮询获取最新消息
   useEffect(() => {
     if (!systemPush) return;
 
-    const poll = async () => {
-      if (isPollingRef.current) return;
-      
-      try {
-        isPollingRef.current = true;
-        await updateMessages(false);
-      } finally {
-        isPollingRef.current = false;
-        if (systemPush) {
-          timeoutIdRef.current = setTimeout(poll, POLLING_INTERVAL);
-        }
-      }
-    };
-
     poll();
 
     return () => {
@@ -248,7 +257,7 @@ export function useMessages({ config, onMessagesUpdate }: UseMessagesProps): [Me
         clearTimeout(timeoutIdRef.current);
       }
     };
-  }, [systemPush, updateMessages]);
+  }, [systemPush, poll]);
 
   // 初始加载
   useEffect(() => {