From 7959cef5049430f13616e58d67eaf82221979600 Mon Sep 17 00:00:00 2001
From: shenlan <manbuzhe2009@qq.com>
Date: Sun, 3 Aug 2025 13:52:37 +0800
Subject: [PATCH] Add minimize and chat end features with message limit

---
 ui/homepage/components/AskAIButton.tsx | 31 ++++++++++++++------
 ui/homepage/components/AskAIDialog.tsx | 39 +++++++++++++++++---------
 2 files changed, 48 insertions(+), 22 deletions(-)

diff --git a/ui/homepage/components/AskAIButton.tsx b/ui/homepage/components/AskAIButton.tsx
index a27f7c3..2a7ca72 100644
--- a/ui/homepage/components/AskAIButton.tsx
+++ b/ui/homepage/components/AskAIButton.tsx
@@ -6,18 +6,33 @@ import { AskAIDialog } from './AskAIDialog'
 
 export function AskAIButton() {
   const [open, setOpen] = useState(false)
+  const [minimized, setMinimized] = useState(false)
 
   return (
     <>
-      <button
-        onClick={() => setOpen(true)}
-        className="fixed bottom-6 right-6 z-50 flex items-center gap-2 rounded-full bg-purple-600 text-white px-4 py-3 shadow-lg hover:bg-purple-500"
-      >
-        <Bot className="w-5 h-5" />
-        Ask AI
-      </button>
+      {!open && (
+        <button
+          onClick={() => setOpen(true)}
+          className={`fixed bottom-6 right-6 z-50 flex items-center gap-2 rounded-full bg-purple-600 text-white shadow-lg hover:bg-purple-500 ${
+            minimized ? 'w-12 h-12 justify-center' : 'px-4 py-3'
+          }`}
+        >
+          <Bot className="w-5 h-5" />
+          {!minimized && 'Ask AI'}
+        </button>
+      )}
 
-      <AskAIDialog open={open} onClose={() => setOpen(false)} />
+      <AskAIDialog
+        open={open}
+        onMinimize={() => {
+          setOpen(false)
+          setMinimized(true)
+        }}
+        onEnd={() => {
+          setOpen(false)
+          setMinimized(false)
+        }}
+      />
     </>
   )
 }
diff --git a/ui/homepage/components/AskAIDialog.tsx b/ui/homepage/components/AskAIDialog.tsx
index 3d7d298..a0f2b1b 100644
--- a/ui/homepage/components/AskAIDialog.tsx
+++ b/ui/homepage/components/AskAIDialog.tsx
@@ -4,7 +4,9 @@ import { useState } from 'react'
 import { ChatBubble } from './ChatBubble'
 import { SourceHint } from './SourceHint'
 
-export function AskAIDialog({ open, onClose }: { open: boolean; onClose: () => void }) {
+const MAX_MESSAGES = 20
+
+export function AskAIDialog({ open, onMinimize, onEnd }: { open: boolean; onMinimize: () => void; onEnd: () => void }) {
   const [question, setQuestion] = useState('')
   const [messages, setMessages] = useState<{ sender: 'user' | 'ai'; text: string }[]>([])
 
@@ -12,27 +14,39 @@ export function AskAIDialog({ open, onClose }: { open: boolean; onClose: () => v
     if (!question) return
     const userMessage = { sender: 'user' as const, text: question }
     const apiBase = process.env.NEXT_PUBLIC_API_BASE_URL || ''
+    const trimmedHistory = messages.slice(-MAX_MESSAGES)
     const res = await fetch(`${apiBase}/api/askai`, {
       method: 'POST',
-      body: JSON.stringify({ question, history: messages }),
+      body: JSON.stringify({ question, history: trimmedHistory })
     })
     const data = await res.json()
     const aiMessage = { sender: 'ai' as const, text: data.answer as string }
-    setMessages((prev) => [...prev, userMessage, aiMessage])
+    setMessages(prev => {
+      const newMessages = [...prev, userMessage, aiMessage]
+      return newMessages.slice(-MAX_MESSAGES)
+    })
     setQuestion('')
   }
 
+  function handleEnd() {
+    setMessages([])
+    setQuestion('')
+    onEnd()
+  }
+
   if (!open) return null
 
   return (
     <div className="fixed inset-0 bg-black/40 z-50 flex items-end md:items-center justify-center">
       <div className="bg-white w-full max-w-xl rounded-2xl p-6 m-4 shadow-xl relative">
-        <button
-          onClick={onClose}
-          className="absolute top-3 right-4 text-gray-500 hover:text-gray-800"
-        >
-          ✕
-        </button>
+        <div className="absolute top-3 right-4 flex gap-2 text-gray-500">
+          <button onClick={onMinimize} className="hover:text-gray-800" title="Minimize">
+            –
+          </button>
+          <button onClick={handleEnd} className="hover:text-gray-800" title="End conversation">
+            End
+          </button>
+        </div>
         <h2 className="text-lg font-semibold mb-3">Ask anything about your docs</h2>
 
         <textarea
@@ -40,13 +54,10 @@ export function AskAIDialog({ open, onClose }: { open: boolean; onClose: () => v
           rows={3}
           placeholder="Type your question..."
           value={question}
-          onChange={(e) => setQuestion(e.target.value)}
+          onChange={e => setQuestion(e.target.value)}
         />
 
-        <button
-          onClick={handleAsk}
-          className="bg-indigo-600 text-white px-4 py-2 rounded-lg hover:bg-indigo-700"
-        >
+        <button onClick={handleAsk} className="bg-indigo-600 text-white px-4 py-2 rounded-lg hover:bg-indigo-700">
           Ask
         </button>