From a19b1025890eb0f71b44a9d1bb948cb084d4106a Mon Sep 17 00:00:00 2001
From: Junjie <DELL@qq.com>
Date: 星期五, 12 十二月 2025 16:50:44 +0800
Subject: [PATCH] #

---
 src/main/webapp/views/ai/diagnosis.html |  199 +++++++++++++++++++++++++++++++++++++++++++++----
 1 files changed, 180 insertions(+), 19 deletions(-)

diff --git a/src/main/webapp/views/ai/diagnosis.html b/src/main/webapp/views/ai/diagnosis.html
index 70867ec..612b817 100644
--- a/src/main/webapp/views/ai/diagnosis.html
+++ b/src/main/webapp/views/ai/diagnosis.html
@@ -3,40 +3,71 @@
 <head>
   <meta charset="UTF-8" />
   <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-  <title>WCS AI 璇婃柇</title>
+  <title>WCS AI 鍔╂墜</title>
   <link rel="stylesheet" href="../../static/vue/element/element.css" />
   <style>
     body { background: #f5f7fa; }
     .container { max-width: 1100px; margin: 24px auto; }
     .actions { display: flex; gap: 12px; align-items: center; }
-    .output { min-height: 360px; }
+    .output { height: 60vh; }
     .markdown-body { font-size: 14px; line-height: 1.4; white-space: pre-wrap; word-break: break-word; }
     .markdown-body p { margin: 4px 0; }
     .markdown-body ul, .markdown-body ol { margin: 4px 0 4px 16px; padding: 0; }
     .markdown-body h1, .markdown-body h2, .markdown-body h3 { margin-top: 8px; }
     .markdown-body pre { background: #f6f8fa; padding: 12px; border-radius: 6px; overflow: auto; }
     .status { color: #909399; }
+    .chat { display: flex; flex-direction: column; gap: 10px; height: 100%; overflow-y: auto; padding-right: 8px; }
+    .msg { display: flex; align-items: flex-start; }
+    .msg.user { justify-content: flex-end; }
+    .msg.assistant { justify-content: flex-start; }
+    .bubble { max-width: 72%; padding: 10px 12px; border-radius: 16px; line-height: 1.5; white-space: pre-wrap; word-break: break-word; }
+    .assistant .bubble { background: #ffffff; border: 1px solid #ebeef5; color: #303133; }
+    .user .bubble { background: #409EFF; color: #ffffff; }
+    .composer { display: flex; gap: 10px; align-items: center; margin-top: 12px; }
+    .avatar { width: 24px; height: 24px; display: flex; align-items: center; margin-right: 8px; }
+    .time { font-size: 12px; color: #909399; text-align: right; margin-top: 6px; }
+    .output .el-card__body { height: 100%; padding: 0; }
   </style>
 </head>
 <body>
   <div id="app" class="container">
     <el-card shadow="hover">
-      <div slot="header" class="clearfix">
-        <span>WCS AI 璇婃柇</span>
+      <div slot="header" class="clearfix" style="display: flex; align-items: center;">
+        <div v-html="headerIcon" style="margin-right: 10px; display: flex;"></div>
+        <span>WCS AI 鍔╂墜</span>
       </div>
 
-      <div class="actions">
-        <el-button type="primary" :loading="loading" :disabled="streaming" @click="start">寮�濮嬭瘖鏂�</el-button>
+      <div class="actions" style="flex-wrap: wrap;">
+        <el-button type="primary" :loading="loading" :disabled="streaming" @click="start">涓�閿瘖鏂郴缁�</el-button>
         <el-button type="warning" :disabled="!streaming" @click="stop">鍋滄</el-button>
-        <el-button @click="clear">娓呯┖</el-button>
-        <span class="status">{{ statusText }}</span>
+        <el-button @click="clear">娓呯┖褰撳墠鑱婂ぉ</el-button>
+        <span class="status" style="margin-right: 12px;">{{ statusText }}</span>
+        <el-select v-model="currentChatId" placeholder="閫夋嫨浼氳瘽" style="min-width:240px;" @change="switchChat" :disabled="streaming">
+          <el-option v-for="c in chats" :key="c.chatId" :label="(c.title||('浼氳瘽 '+c.chatId)) + '锛�'+(c.size||0)+'锛�'" :value="c.chatId" />
+        </el-select>
+        <el-button type="success" plain icon="el-icon-plus" @click="newChat" :disabled="streaming">鏂颁細璇�</el-button>
+        <el-button type="danger" plain icon="el-icon-delete" @click="deleteChat" :disabled="!currentChatId || streaming">鍒犻櫎浼氳瘽</el-button>
       </div>
 
       <el-divider></el-divider>
 
       <el-card class="output" shadow="never">
-        <div class="markdown-body" v-html="renderedHtml"></div>
+        <div ref="chat" class="chat">
+          <div v-for="(m,i) in messages" :key="i" class="msg" :class="m.role">
+            <div class="avatar" v-html="m.role === 'assistant' ? assistantIcon : userIcon"></div>
+            <div class="bubble">
+              <div v-if="m.role === 'assistant'" class="markdown-body" v-html="m.html"></div>
+              <div v-else v-text="m.text"></div>
+              <div class="time">{{ m.ts }}</div>
+            </div>
+          </div>
+        </div>
       </el-card>
+
+      <div class="composer">
+        <el-input v-model="userInput" placeholder="鍚� AI 鍔╂墜鎻愰棶" clearable :disabled="streaming" @keyup.enter.native="ask"></el-input>
+        <el-button type="success" :disabled="sendDisabled" @click="ask">鍙戦��</el-button>
+      </div>
     </el-card>
   </div>
 
@@ -51,20 +82,35 @@
       breaks: true
     });
 
+    function getUserIconHtml(width, height) {
+      width = width || 24; height = height || 24;
+      return '<svg width="'+width+'" height="'+height+'" viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg">\n'
+        + '<circle cx="12" cy="7" r="4" fill="#909399"/>\n'
+        + '<path d="M4 20c0-4 4-6 8-6s8 2 8 6" fill="#909399" opacity="0.35"/>\n'
+        + '</svg>';
+    }
+
     new Vue({
       el: '#app',
       data: function() {
         return {
+          headerIcon: getAiIconHtml(50, 50),
+          assistantIcon: getAiIconHtml(24, 24),
+          userIcon: getUserIconHtml(24, 24),
           loading: false,
           streaming: false,
           source: null,
-          markdownBuffer: '',
-          renderedHtml: '',
+          messages: [],
           pendingText: '',
           typingTimer: null,
           lastRenderTs: 0,
           renderIntervalMs: 120,
-          stepChars: 6
+          stepChars: 6,
+          userInput: '',
+          autoScrollThreshold: 80,
+          chats: [],
+          currentChatId: '',
+          resetting: false
         };
       },
       computed: {
@@ -72,9 +118,105 @@
           if (this.streaming) return '璇婃柇杩涜涓�';
           if (this.loading) return '杩炴帴涓�';
           return '绌洪棽';
+        },
+        sendDisabled: function() {
+          var t = (this.userInput || '').trim();
+          return this.streaming || t.length === 0;
         }
       },
       methods: {
+        loadChats: function() {
+          var self = this;
+          fetch(baseUrl + '/ai/diagnose/chats', { headers: { 'token': localStorage.getItem('token') } })
+            .then(function(r){ return r.json(); })
+            .then(function(arr){ if (Array.isArray(arr)) { self.chats = arr; } });
+        },
+        switchChat: function() {
+          var self = this;
+          if (!self.currentChatId) { self.clear(); return; }
+          fetch(baseUrl + '/ai/diagnose/chats/' + encodeURIComponent(self.currentChatId) + '/history', { headers: { 'token': localStorage.getItem('token') } })
+            .then(function(r){ return r.json(); })
+            .then(function(arr){
+              if (!Array.isArray(arr)) return;
+              var msgs = [];
+              for (var i=0;i<arr.length;i++) {
+                var m = arr[i];
+                if (m.role === 'assistant') msgs.push({ role: 'assistant', md: m.content || '', html: DOMPurify.sanitize(marked.parse((m.content||'').replace(/\\n/g,'\n'))), ts: self.nowStr() });
+                else msgs.push({ role: 'user', text: m.content || '', ts: self.nowStr() });
+              }
+              self.messages = msgs;
+              self.$nextTick(function(){ self.scrollToBottom(true); });
+            });
+        },
+        newChat: function() {
+          var id = Date.now() + '_' + Math.random().toString(36).substr(2,8);
+          this.currentChatId = id;
+          this.resetting = true;
+          this.clear();
+        },
+        deleteChat: function() {
+          var self = this;
+          if (!self.currentChatId) return;
+          fetch(baseUrl + '/ai/diagnose/chats/' + encodeURIComponent(self.currentChatId), { method: 'DELETE', headers: { 'token': localStorage.getItem('token') } })
+            .then(function(r){ return r.json(); })
+            .then(function(ok){ if (ok === true) { self.currentChatId = ''; self.clear(); self.loadChats(); self.newChat(); } });
+        },
+        shouldAutoScroll: function() {
+          var el = this.$refs.chat;
+          if (!el) return false;
+          return (el.scrollHeight - el.scrollTop - el.clientHeight) <= this.autoScrollThreshold;
+        },
+        scrollToBottom: function(force) {
+          var el = this.$refs.chat;
+          if (!el) return;
+          if (force || this.streaming || this.shouldAutoScroll()) {
+            el.scrollTop = el.scrollHeight;
+          }
+        },
+        nowStr: function() {
+          var d = new Date();
+          function pad(n) { return (n<10?'0':'') + n; }
+          var y = d.getFullYear();
+          var m = pad(d.getMonth() + 1);
+          var day = pad(d.getDate());
+          var hh = pad(d.getHours());
+          var mm = pad(d.getMinutes());
+          var ss = pad(d.getSeconds());
+          return y + '-' + m + '-' + day + ' ' + hh + ':' + mm + ':' + ss;
+        },
+        ask: function() {
+          if (this.streaming) return;
+          var msg = (this.userInput || '').trim();
+          if (!msg) return;
+          this.loading = true;
+          this.streaming = true;
+          this.messages.push({ role: 'user', text: msg, ts: this.nowStr() });
+          this.messages.push({ role: 'assistant', md: '', html: '', ts: this.nowStr() });
+          this.scrollToBottom(true);
+          var url = baseUrl + '/ai/diagnose/askStream?prompt=' + encodeURIComponent(msg);
+          if (this.currentChatId) url += '&chatId=' + encodeURIComponent(this.currentChatId);
+          if (this.resetting) url += '&reset=true';
+          this.source = new EventSource(url);
+          var self = this;
+          this.source.onopen = function() {
+            self.loading = false;
+          };
+          this.source.onmessage = function(e) {
+            if (!e || !e.data) return;
+            var chunk = (e.data || '').replace(/\\n/g, '\n');
+            if (!chunk) return;
+            var normalized = chunk.replace(/\r/g, '');
+            if (/^\n+$/.test(normalized)) return;
+            self.pendingText += chunk;
+            self.ensureTyping();
+            self.scrollToBottom(true);
+          };
+          this.source.onerror = function() {
+            self.stop();
+          };
+          this.userInput = '';
+          this.resetting = false;
+        },
         start: function() {
           if (this.streaming) return;
           this.clear();
@@ -82,6 +224,8 @@
           this.streaming = true;
           var url = baseUrl + '/ai/diagnose/runAiStream';
           this.source = new EventSource(url);
+          this.messages.push({ role: 'assistant', md: '', html: '', ts: this.nowStr() });
+          this.scrollToBottom(true);
           var self = this;
           this.source.onopen = function() {
             self.loading = false;
@@ -96,6 +240,7 @@
             if (/^\n+$/.test(normalized)) return;
             self.pendingText += chunk;
             self.ensureTyping();
+            self.scrollToBottom(true);
           };
           this.source.onerror = function() {
             self.stop();
@@ -112,14 +257,22 @@
             }
             if (self.pendingText.length > 0) {
               var n = Math.min(self.stepChars, self.pendingText.length);
-              self.markdownBuffer += self.pendingText.slice(0, n);
+              var part = self.pendingText.slice(0, n);
               self.pendingText = self.pendingText.slice(n);
+              var last = self.messages.length > 0 ? self.messages[self.messages.length - 1] : null;
+              if (last && last.role === 'assistant') {
+                last.md = (last.md || '') + part;
+              }
             }
             var now = Date.now();
             if (now - self.lastRenderTs > self.renderIntervalMs) {
               self.lastRenderTs = now;
-              var renderSource = self.markdownBuffer.replace(/\\n/g, '\n');
-              self.renderedHtml = DOMPurify.sanitize(marked.parse(renderSource));
+              var last = self.messages.length > 0 ? self.messages[self.messages.length - 1] : null;
+              if (last && last.role === 'assistant') {
+                var renderSource = (last.md || '').replace(/\\n/g, '\n');
+                last.html = DOMPurify.sanitize(marked.parse(renderSource));
+                self.$nextTick(function() { self.scrollToBottom(true); });
+              }
             }
           }, 50);
         },
@@ -134,15 +287,23 @@
             clearInterval(this.typingTimer);
             this.typingTimer = null;
           }
-          var renderSource = this.markdownBuffer.replace(/\\n/g, '\n');
-          this.renderedHtml = DOMPurify.sanitize(marked.parse(renderSource));
+          var last = this.messages.length > 0 ? this.messages[this.messages.length - 1] : null;
+          if (last && last.role === 'assistant') {
+            var renderSource = (last.md || '').replace(/\\n/g, '\n');
+            last.html = DOMPurify.sanitize(marked.parse(renderSource));
+          }
+          this.$nextTick(function() { this.scrollToBottom(true); }.bind(this));
+          this.loadChats();
         },
         clear: function() {
-          this.markdownBuffer = '';
-          this.renderedHtml = '';
+          this.messages = [];
           this.pendingText = '';
         }
       }
+      ,mounted: function() {
+        this.loadChats();
+        this.newChat();
+      }
     });
   </script>
 </body>

--
Gitblit v1.9.1