diff --git a/src/views/pages/login.vue b/src/views/pages/login.vue
index 8cdd0c7..ee7879f 100644
--- a/src/views/pages/login.vue
+++ b/src/views/pages/login.vue
@@ -20,7 +20,6 @@
             type="password"
             placeholder="密码"
             v-model="param.password"
-            @keyup.enter="submitForm(login)"
           >
             <template #prepend>
               <el-icon>
@@ -56,7 +55,7 @@
   </div>
 </template>
 
-<script setup>
+<script setup lang="ts">
 import { ref, reactive, inject } from "vue";
 import { useRouter } from "vue-router";
 import { ElMessage } from "element-plus";
@@ -115,17 +114,17 @@ const onLogin = async () => {
   loginData.value.password = param.password;
   let result = await loginService(loginData);
   console.log("login result:", result);
-  if (result.code !== 0) {
+  if (result["code"] !== 0) {
     //alert(result.message);
     ElMessage.error("登录失败！用户名或密码错误");
     return;
   }
-  globalData.token = result.data;
+  globalData["token"] = result.data;
   localStorage.setItem("token", result.data.token);
   localStorage.setItem("userId", result.data.id);
   localStorage.setItem("username", result.data.username);
   let now = new Date();
-  localStorage.setItem("end_time", now.setDate(now.getHours() + 12)); //过期时间
+  localStorage.setItem("end_time", (now.setDate(now.getHours() + 12)).toString()); //过期时间
   await getMyUserInfo(result.data.id);
   //token.value= result.data;
 
@@ -139,18 +138,18 @@ const getMyUserInfo = async (id) => {
       id: id,
     };
     result = await GetUserInfoService(tokenData);
-    if (result.code === 0) {
+    if (result["code"] === 0) {
       //console.log("token data:",this.tokenData)
-      localStorage.setItem("video_func", result.data.VideoFunc);
-      localStorage.setItem("device_func", result.data.DeviceFunc);
-      localStorage.setItem("cid_func", result.data.CIDFunc);
-      localStorage.setItem("role", result.data.Role);
+      // localStorage.setItem("video_func", result.data.VideoFunc);
+      // localStorage.setItem("device_func", result.data.DeviceFunc);
+      // localStorage.setItem("cid_func", result.data.CIDFunc);
+      // localStorage.setItem("role", result.data.Role);
 
       ElMessage.success("登录成功");
-        localStorage.setItem("ms_username", result.data.Name);
+        localStorage.setItem("ms_username", result["data"]["Name"]);
         const keys =
-            permiss.defaultList[result.data.Role == "admin" ? "admin" : "user"];
-            localStorage.setItem("ms_imgurl", result.data.Avatar);
+            permiss.defaultList[result["data"]["Role"] == "admin" ? "admin" : "user"];
+            localStorage.setItem("ms_imgurl", result["data"]["Avatar"]);
         permiss.handleSet(keys);
         localStorage.setItem("ms_keys", JSON.stringify(keys));
         router.push("/");
diff --git a/src/views/pages/register.vue b/src/views/pages/register.vue
index 8f7869f..20d0380 100644
--- a/src/views/pages/register.vue
+++ b/src/views/pages/register.vue
@@ -27,8 +27,7 @@
                     </el-input>
                 </el-form-item>
                 <el-form-item prop="password">
-                    <el-input type="password" placeholder="密码" v-model="param.password"
-                        @keyup.enter="submitForm(register)">
+                    <el-input type="password" placeholder="密码" v-model="param.password">
                         <template #prepend>
                             <el-icon>
                                 <Lock />
@@ -37,8 +36,7 @@
                     </el-input>
                 </el-form-item>
                 <el-form-item prop="password">
-                    <el-input type="password" placeholder="密码" v-model="param.repassword"
-                        @keyup.enter="submitForm(register)">
+                    <el-input type="password" placeholder="密码" v-model="param.repassword">
                         <template #prepend>
                             <el-icon>
                                 <Lock />
@@ -53,13 +51,15 @@
     </div>
 </template>
 
-<script setup>
+<script setup lang="ts">
 import { ref, reactive,inject } from 'vue';
 import { useRouter } from 'vue-router';
 import { ElMessage } from 'element-plus';
 import { registerService } from "@/api/user";
 import {GetUserInfoService} from "@/api/user";
+import { usePermissStore } from "@/store/permiss";
 const router = useRouter();
+const permiss = usePermissStore();
 
 const globalData = inject("globalData");
 const param = reactive({
@@ -98,17 +98,17 @@ const getMyUserInfo = async (id) => {
       id: id,
     }
     result = await GetUserInfoService(tokenData);
-    if (result.code === 0) {
+    if (result["code"] === 0) {
       //console.log("token data:",this.tokenData)
-      localStorage.setItem("video_func", result.data.VideoFunc);
-      localStorage.setItem("device_func", result.data.DeviceFunc);
-      localStorage.setItem("cid_func", result.data.CIDFunc);
-      localStorage.setItem("role", result.data.Role);
+    //   localStorage.setItem("video_func", result.data.VideoFunc);
+    //   localStorage.setItem("device_func", result.data.DeviceFunc);
+    //   localStorage.setItem("cid_func", result.data.CIDFunc);
+    //   localStorage.setItem("role", result.data.Role);
 
       ElMessage.success("注册成功");
-        localStorage.setItem("ms_username", result.data.Name);
+        localStorage.setItem("ms_username", result["data"]["Name"]);
         const keys =
-            permiss.defaultList[result.data.Role == "admin" ? "admin" : "user"];
+            permiss.defaultList[result["data"]["Role"] == "admin" ? "admin" : "user"];
         permiss.handleSet(keys);
         localStorage.setItem("ms_keys", JSON.stringify(keys));
         router.push("/");
@@ -140,12 +140,12 @@ const onRegister = async () => {
 
   let result = await registerService(registerData);
   if (result !== null) {
-    globalData.token = result.data;
+    globalData["token"] = result.data;
     localStorage.setItem("token", result.data.token);
     localStorage.setItem("userId", result.data.id);
     localStorage.setItem("username", result.data.username);
     let now = new Date();
-    localStorage.setItem("end_time", now.setDate(now.getHours() + 12)); //过期时间
+    localStorage.setItem("end_time", (now.setDate(now.getHours() + 12)).toString()); //过期时间
     //token.value= result.data;
     await getMyUserInfo(result.data.id);
     router.push("/home");
diff --git a/src/views/pages/reset-pwd.vue b/src/views/pages/reset-pwd.vue
index f78aa62..9bb258d 100644
--- a/src/views/pages/reset-pwd.vue
+++ b/src/views/pages/reset-pwd.vue
@@ -49,7 +49,7 @@
     </div>
 </template>
 
-<script setup >
+<script setup lang="ts">
 import { ref,inject } from 'vue';
 import { ElMessage } from 'element-plus';
 import {genResetPassword} from "@/api/user";
@@ -90,11 +90,11 @@ const resetPassword  = async () =>{
 
     try{
         let result = await genResetPassword(req);
-        if (result.code === 0) {
+        if (result["code"] === 0) {
             //重置成功，返回新token，直接登录
             if (result.data.token) {
                 localStorage.setItem('token', result.data.token);
-                globalData.token = result.data.token;
+                globalData["token"] = result.data.token;
                 ElMessage.success('重置密码成功');
                 router.push('/home');
             } else {
@@ -102,7 +102,7 @@ const resetPassword  = async () =>{
                 router.push('/login');
             }
         } else {
-            ElMessage.error(result.msg);
+            ElMessage.error(result["message"]);
         }
         
     }catch(err){
@@ -122,11 +122,11 @@ const sendCodeMail = async () => {
     
     try{
         let result = await genResetPassword(req);
-        if (result.code === 0) {
+        if (result["code"] === 0) {
             //重置成功，返回新token，直接登录
             ElMessage.success('邮件已成功发送，请查收');
         } else {
-            ElMessage.error(result.message);
+            ElMessage.error(result["message"]);
             return;
         }
         
diff --git a/src/views/pages/ucenter.vue b/src/views/pages/ucenter.vue
index eb60de0..5e4bc5b 100644
--- a/src/views/pages/ucenter.vue
+++ b/src/views/pages/ucenter.vue
@@ -75,7 +75,7 @@
 </template>
 
 <script setup lang="ts" name="ucenter">
-import { reactive, ref } from 'vue';
+import { reactive, ref,inject } from 'vue';
 import { VueCropper } from 'vue-cropper';
 import 'vue-cropper/dist/index.css';
 import avatar from '@/assets/img/img.jpg';
@@ -96,9 +96,9 @@ const form = reactive({
     new: '',
     old: '',
 });
-const onSubmit = () => {};
-const userInfo = ref<UserInfo>({});
+const userInfo = ref<UserInfo>();
 const isUserInfoLoaded = ref(false);
+const globalData = inject("globalData");
 
 const activeName = ref('label2');
 const router = useRouter();
@@ -147,7 +147,7 @@ const updateUserInfo = async (data: any) => {
             role:data.Role
         };
         result = await updateUserInfoService(req)
-        if (result.code === 0) {
+        if (result["code"] === 0) {
           ElMessage.success("更新成功");
         } else {
           ElMessage.error("更新失败");
@@ -167,15 +167,15 @@ const resetPassword = async () =>{
     }
     try{
         let result = await genResetPassword(req);
-        if (result.code === 0) {
+        if (result["code"] === 0) {
             //重置成功，返回新token
             if (result.data.token) {
                 localStorage.setItem('token', result.data.token);
-                globalData.token = result.data.token;
+                globalData["token"] = result.data.token;
                 ElMessage.success('重置密码成功');
             }
         } else {
-            ElMessage.error(result.msg);
+            ElMessage.error(result["msg"]);
         }
     }catch(e){
         console.log(e)
@@ -210,13 +210,13 @@ const GetMyUserInfo = async () => {
     };
     try{
         let result = await GetUserInfoService(req);
-        if (result.code == 0) {
+        if (result["code"] == 0) {
             avatarImg.value = result.data.Avatar == '' ? avatar : result.data.Avatar;
             imgSrc.value = avatarImg.value;
             userInfo.value = result.data;
             userInfo.value.Password = '**********';
         }else{
-            ElMessage.error(result.msg);
+            ElMessage.error(result["msg"]);
         }
         isUserInfoLoaded.value = true;
     }catch(e){
@@ -259,7 +259,7 @@ const saveAvatar =async () => {
 
         
         let result = await UploadFileService(formData, token);
-        if (result.code !== 0) {
+        if (result["code"] !== 0) {
             ElMessage.error('上传文件失败，请稍后再试');
             return;
         }
diff --git a/src/views/system/copy-test.vue b/src/views/system/copy-test.vue
index 072304a..e69de29 100644
--- a/src/views/system/copy-test.vue
+++ b/src/views/system/copy-test.vue
@@ -1,467 +0,0 @@
-<template>
-    <div class="chat-app">
-      <!-- 历史会话侧边栏 -->
-      <div class="history-sessions" v-if="sessionIsShow">
-          <div >
-            <el-button type="primary" @click="clearCurrent">新会话</el-button>
-          </div>
-        <el-card class="session-card">
-          <template #header>
-            <h3>当前会话</h3>
-          </template>
-          <ul>
-            <li>{{ getShortenedName(sessionName) }}</li>
-          </ul>
-        </el-card>
-        <el-card class="session-card2">
-          <template #header>
-            <h3>历史会话</h3>
-          </template>
-          <el-scrollbar style="height: 400px; scrollbar-width: none">
-            <ul>
-              <li
-                v-for="(session, index) in historySessions"
-                :key="index"
-                @click="loadSession(session.ID)"
-              >
-                {{ getShortenedName(session.Name) }}
-              </li>
-            </ul>
-          </el-scrollbar>
-        </el-card>
-      </div>
-  
-      <!-- 原有的聊天区域 -->
-      <div class="chat-container">
-        <div>
-          <div @click="showSession" style="cursor: pointer">
-            <el-icon v-if="sessionIsShow">
-              <Expand />
-            </el-icon>
-            <el-icon v-else>
-              <Fold />
-            </el-icon>
-          </div>
-        </div>
-        <!-- 消息列表 -->
-        <el-card class="chat-messages" shadow="never" ref="messagesContainer" v-if="messages.length > 0">
-          <div
-            v-for="(message, index) in messages"
-            :key="index"
-            :class="['message', message.role]"
-          >
-            <div class="message-avatar">
-              <span v-if="message.role === 'assistant'">💬</span>
-              <span v-else>👤</span>
-            </div>
-            <div class="message-content">
-              <div v-html="renderMarkdown(message.content)"></div>
-              <!-- 添加复制 -->
-              <div>
-              <el-button type="text" :icon="DocumentCopy" @click="copyMessage(message.content)"></el-button>
-            </div>
-            </div>
-          </div>
-          <div v-if="loading" class="loading-indicator">Loading...</div>
-        </el-card>
-  
-        <!-- 输入区域 -->
-        <el-card class="chat-input" shadow="never">
-          <el-row :gutter="10">
-            <el-col :span="20">
-              <el-input
-                v-model="inputMessage"
-                type="textarea"
-                :rows="2"
-                placeholder="输入消息..."
-                @keyup.enter="sendMessage"
-              />
-              <el-text v-model="inputMessage" aria-placeholder="输入信息...."></el-text>
-            </el-col>
-            <el-col :span="4" style="text-align: center">
-              <el-button  @click="sendMessage" type="success" :icon="Check" round :disabled="loading">发送</el-button>
-            </el-col>
-          </el-row>
-        </el-card>
-      </div>
-    </div>
-  </template>
-  
-  <script setup lang="ts">
-  import { ref, onMounted, onUnmounted, reactive, nextTick } from "vue";
-  import { ElCard, ElInput, ElButton } from "element-plus";
-  import { WSMessage, AIQMessage, OllamaMessage } from "@/types/im";
-  import { ElMessage } from "element-plus";
-  import { GetMessageService } from "@/api/im";
-  import {Check, Loading, DocumentCopy} from '@element-plus/icons-vue'
-  import MarkdownIt from "markdown-it";
-  import markdownItMermaid from "markdown-it-mermaid";
-  import hljs from "highlight.js";
-  import { Session } from "@/types/session";
-  import bash from "highlight.js/lib/languages/bash";
-  import { FindSessionService } from "@/api/session";
-  import javascript from "highlight.js/lib/languages/javascript";
-  import typescript from "highlight.js/lib/languages/typescript";
-  import java from "highlight.js/lib/languages/java";
-  import sql from "highlight.js/lib/languages/sql";
-  import nginx from "highlight.js/lib/languages/nginx";
-  import json from "highlight.js/lib/languages/json";
-  import yaml from "highlight.js/lib/languages/yaml";
-  import xml from "highlight.js/lib/languages/xml";
-  import shell from "highlight.js/lib/languages/shell";
-  import "katex/dist/katex.min.css";
-  interface Message {
-    role: "user" | "assistant";
-    content: string;
-    finished?: boolean;
-  }
-  
-  hljs.registerLanguage("bash", bash);
-  hljs.registerLanguage("javascript", javascript);
-  hljs.registerLanguage("typescript", typescript);
-  hljs.registerLanguage("java", java);
-  hljs.registerLanguage("sql", sql);
-  hljs.registerLanguage("nginx", nginx);
-  hljs.registerLanguage("json", json);
-  hljs.registerLanguage("yaml", yaml);
-  hljs.registerLanguage("xml", xml);
-  hljs.registerLanguage("shell", shell);
-  const md = new MarkdownIt({
-    html: true,
-    linkify: true,
-    breaks: true,
-    xhtmlOut: true,
-    typographer: true,
-    highlight: (str, lang) => {
-      if (lang && hljs.getLanguage(lang)) {
-        try {
-          return `<pre class="hljs"><code>${
-            hljs.highlight(str, { language: lang, ignoreIllegals: true }).value
-          }</code></pre>`;
-        } catch (__) {}
-      }
-      return `<pre class="hljs"><code>${md.utils.escapeHtml(str)}</code></pre>`;
-    },
-  });
-  
-  const historySessions = ref<Session[]>([]);
-  const loading = ref(false);
-  const isUserScrolling = ref<boolean>(false);
-  const messages = reactive([]);
-  const inputMessage = ref("");
-  const socket = ref(null);
-  const currentAIMessage = ref("");
-  const sessionID = ref(0);
-  const messagesContainer = ref<HTMLDivElement | null>(null);
-  const sessionIsShow = ref(false);
-  const sessionName = ref("");
-  
-  const renderMarkdown = (content: string) => {
-    return md.render(content);
-  };
-  
-  const scrollToBottom = () => {
-    let x = document.getElementsByClassName("chat-messages")[0];
-    x.scrollTop = x.scrollHeight; //将滚轮置底
-  };
-  
-  onMounted(() => {
-    // if (typeof window !== 'undefined') {
-    //   // 浏览器环境
-    //   md.use(markdownItMermaid);
-    // }
-    let url =
-      "wss://pm.ljsea.top/im/ai_chat_ws?" +
-      "token=" +
-      localStorage.getItem("token");
-    socket.value = new WebSocket(url);
-    socket.value.onopen = () => {
-      console.log("WebSocket 连接已建立");
-      ElMessage.success("连接成功");
-    };
-    //getMessage(125);
-    messagesContainer.value = document.querySelector(".chat-messages");
-  
-    socket.value.onmessage = (event) => {
-      let msg: WSMessage = JSON.parse(event.data);
-      const existingMessage = messages.find(
-        (msg) => msg.role === "assistant" && !msg.finished
-      );
-      if (existingMessage) {
-        // 追加内容
-        existingMessage.content += msg.msg.msg.response;
-      } else {
-        // 新消息
-        messages.push({
-          role: "assistant",
-          content: msg.msg.msg.response,
-          finished: false,
-        });
-      }
-      //console.log("resp:", msg);
-      sessionID.value = msg.session_id;
-      currentAIMessage.value += msg.msg.msg.response;
-      if (msg.msg.msg.done) {
-        const assistantMessage = messages[messages.length - 1];
-        assistantMessage.finished = true;
-        loading.value = false;
-      }
-      nextTick(() => {
-        scrollToBottom(); // 新增滚动调用
-      });
-    };
-  
-    socket.value.onclose = () => {
-      console.log("WebSocket 连接已关闭");
-      ElMessage.error("连接已关闭");
-      //重新连接
-      socket.value = new WebSocket(url);
-    };
-  
-    socket.value.onerror = (error) => {
-      console.error("WebSocket 发生错误:", error);
-    };
-  });
-  
-  onUnmounted(() => {
-    if (socket.value) {
-      socket.value.close();
-    }
-  });
-  
-  const sendMessage = () => {
-    if (inputMessage.value.trim() === "") {
-      ElMessage.warning("消息不能为空");
-      return;
-    }
-    let msg = {
-      msg: inputMessage.value,
-      type: "ollama",
-      function: "gen-ai-chat",
-      session_id: sessionID.value,
-    };
-    messages.push({ role: "user", content: inputMessage.value, finished: true });
-    socket.value.send(JSON.stringify(msg));
-    inputMessage.value = "";
-    scrollToBottom();
-    loading.value = true;
-    if (sessionID.value == 0) {
-      sessionName.value = msg.msg;
-    }
-  };
-  
-  const loadSession = async (session_id: number) => {
-    sessionID.value = session_id;
-    messages.length = 0; // 清空消息
-    sessionName.value = historySessions.value.find(
-      (session) => session.ID == session_id
-    )?.Name;
-    await getMessage(session_id);
-    scrollToBottom();
-  };
-  
-  const clearCurrent = () => {
-    sessionID.value = 0;
-    messages.length = 0; // 清空消息
-    sessionName.value = "新会话";
-    ElMessage.success("新会话已创建!可以开始聊天了");
-  };
-  
-  const showSession = async () => {
-    //获取历史会话
-    let req = {
-      token: localStorage.getItem("token"),
-      type: "UserID",
-    };
-    let result = await FindSessionService(req);
-    historySessions.value = result.data;
-    sessionIsShow.value = !sessionIsShow.value;
-  };
-  const getShortenedName = (name: string) => {
-    if (name.length > 10) {
-      return name.slice(0, 10) + "...";
-    }
-    return name;
-  };
-  
-  const getMessage = async (session_id: number) => {
-    let result = {};
-    try {
-      let req = {
-        token: localStorage.getItem("token"),
-        session_id: session_id,
-      };
-      result = await GetMessageService(req);
-      if (result["code"] === 0) {
-        console.log(result["data"]);
-        let data = result["data"];
-        for (let i = 0; i < data.length; i++) {
-          if (data[i]["Type"] === 3) {
-            messages.push({
-              role: "user",
-              content: data[i]["Msg"],
-              finished: true,
-            });
-          } else if (data[i]["Type"] === 4) {
-            messages.push({
-              role: "assistant",
-              content: data[i]["Msg"],
-              finished: true,
-            });
-          } else {
-            console.log("未知消息类型");
-          }
-        }
-      }
-    } catch (e) {
-      console.log(e);
-    }
-    return {};
-  };
-  const copyMessage = (content: string) => {
-    navigator.clipboard.writeText(content).then(() => {
-      ElMessage.success('复制成功');
-    }).catch((error) => {
-      ElMessage.error('复制失败: ' + error);
-    });
-  };
-  </script>
-  <style scoped>
-  .chat-app {
-    display: flex;
-    height: 100%;
-  }
-  
-  .history-sessions {
-    width: 15%;
-    padding: 20px;
-    height: 100%;
-    overflow-y: auto;
-    border-right: 1px solid #e1e1e1;
-    scrollbar-width: none;
-  }
-  
-  .history-sessions ul {
-    list-style-type: none;
-    padding: 0;
-  }
-  
-  .history-sessions li {
-    cursor: pointer;
-    padding: 10px;
-    border-bottom: 1px solid #e1e1e1;
-  }
-  
-  .history-sessions li:hover {
-    background-color: #f0f0f0;
-  }
-  
-  .chat-container {
-    flex: 1;
-    display: flex;
-    flex-direction: column;
-    padding: 20px;
-    box-sizing: border-box;
-  }
-  
-  .chat-messages {
-    flex: 1;
-    overflow-y: auto; /* 允许垂直滚动 */
-    padding: 10px;
-    margin-bottom: 20px;
-    scrollbar-width: 20px;
-    position: relative;
-    display: flex;
-    flex-direction: column;
-  }
-  
-  .messages-wrapper {
-    flex: 1;
-    display: flex;
-    flex-direction: column;
-    overflow-y: auto; /* 确保内部可以滚动 */
-    height: 100%;
-  }
-  
-  .message {
-    display: flex;
-    margin-bottom: 15px;
-  }
-  
-  .message.user {
-    justify-content: flex-end;
-  }
-  
-  .message-avatar {
-    width: 40px;
-    text-align: center;
-    line-height: 40px;
-    font-size: 20px;
-    margin-right: 10px;
-  }
-  
-  .message.assistant .message-avatar {
-    margin-right: 0;
-    margin-left: 10px;
-  }
-  
-  .message-content {
-    background-color: #f0f9eb;
-    border: 1px solid #e1f3d8;
-    border-radius: 4px;
-    padding: 10px 15px;
-    white-space: pre-wrap;
-    max-width: 70%;
-  }
-  
-  .message.assistant .message-content {
-    background-color: #f3f4f6;
-    border: 1px solid #dce4eb;
-    max-width: 70%;
-  }
-  
-  .loading-indicator {
-    text-align: center;
-    color: #999;
-    font-size: 16px;
-  }
-  
-  .chat-input {
-    width: 100%;
-  }
-  
-  .el-input {
-    width: 100%;
-  }
-  
-  .session-card {
-    margin-bottom: 20px;
-    height: 20%;
-  }
-  .session-card2 {
-    margin-bottom: 20px;
-    height: 70%;
-  }
-  
-  .session-card ul {
-    list-style-type: none;
-    padding: 0;
-    margin: 0;
-  }
-  
-  .session-card li {
-    cursor: pointer;
-    padding: 10px;
-    border-bottom: 1px solid #e1e1e1;
-  }
-  
-  .session-card li:hover {
-    background-color: #f0f0f0;
-  }
-  
-  .scroll-wrapper {
-    height: 100%;
-    overflow-y: auto;
-  }
-  </style>
-  
\ No newline at end of file
diff --git a/src/views/system/gen-chat.vue b/src/views/system/gen-chat.vue
index 9f642a3..24d64cc 100644
--- a/src/views/system/gen-chat.vue
+++ b/src/views/system/gen-chat.vue
@@ -95,25 +95,27 @@ import { ElMessage } from "element-plus";
 import { GetMessageService } from "@/api/im";
 import {Check, Loading, DocumentCopy} from '@element-plus/icons-vue'
 import MarkdownIt from "markdown-it";
-import markdownItMermaid from "markdown-it-mermaid";
 import hljs from "highlight.js";
 import { Session } from "@/types/session";
-import bash from "highlight.js/lib/languages/bash";
 import { FindSessionService } from "@/api/session";
 import markdownItHighlightjs from 'markdown-it-highlightjs';
+import mermaid from 'mermaid';
+import markdownItMermaid from 'markdown-it-mermaid';
 import "katex/dist/katex.min.css";
 interface Message {
   role: "user" | "assistant";
   content: string;
   finished?: boolean;
 }
+mermaid.initialize({ startOnLoad: false });
 
 const md = new MarkdownIt();
-md.use(markdownItHighlightjs,{
+md.use(markdownItHighlightjs, {
   hljs,
   auto: true,
-  code: true
-})
+  code: true,
+});
+md.use(markdownItMermaid);
 
 const historySessions = ref<Session[]>([]);
 const loading = ref(false);
@@ -145,19 +147,44 @@ const copyCode = (code: string) => {
 const doButtonD = () => {
   const codeBlocks = document.querySelectorAll('pre code');
   codeBlocks.forEach((codeBlock) => {
+    // 获取代码类型
+    const codeType = codeBlock.className.replace('hljs ', '');
+    // 创建代码类型显示元素
+    const codeTypeElement = document.createElement('span');
+    codeTypeElement.textContent = codeType.split('-')[1];
+    codeTypeElement.setAttribute("background-color", "rgba(0, 0, 0, 0.1)");
+    codeTypeElement.setAttribute("padding", "3px 6px");
+    codeTypeElement.setAttribute("border-radius", "4px");
+    codeTypeElement.setAttribute("font-size", "0.9em");
+
     // 创建复制按钮
     const copyButton = document.createElement('button');
-    copyButton.textContent = '复制代码';
-    copyButton.classList.add('copy-code-button');
+    copyButton.setAttribute("background-color", "dodgerblue");
+    copyButton.setAttribute("display", "flex");
+    copyButton.setAttribute("align-items", "center");
+    copyButton.setAttribute("padding", "5px 10px");
+    copyButton.setAttribute("cursor", "pointer");
+    copyButton.setAttribute("border-radius", "4px");
+    copyButton.textContent = '复制';
+    copyButton.classList.add();
     copyButton.addEventListener('click', () => {
       copyCode(codeBlock.textContent);
     });
+
     // 设置代码块父元素的定位，以便按钮定位
     const pre = codeBlock.parentNode;
-    pre.style.position = 'relative';
-    // 将复制按钮添加到代码块父元素中
-    pre.appendChild(copyButton);
+    // pre.style.position = 'relative';
+
+    // 创建一个容器用于放置代码类型和复制按钮
+    const controlsContainer = document.createElement('div');
+    controlsContainer.classList.add('code-controls');
+    controlsContainer.appendChild(codeTypeElement);
+    controlsContainer.appendChild(copyButton);
+
+    // 将容器添加到代码块父元素中
+    pre.insertBefore(controlsContainer, codeBlock);
   });
+
 };
 
 onMounted(() => {
@@ -253,7 +280,7 @@ const loadSession = async (session_id: number) => {
   )?.Name;
   await getMessage(session_id);
   scrollToBottom();
-  //doButtonD();
+  doButtonD();
 };
 
 const clearCurrent = () => {
@@ -460,14 +487,27 @@ const copyMessage = (content: string) => {
   overflow-y: auto;
 }
 
-.copy-code-button {
+.code-controls {
   position: absolute;
   top: 5px;
   right: 5px;
-  background-color: #f0f0f0;
-  border: 1px solid #ccc;
-  padding: 5px 10px;
-  cursor: pointer;
-  border-radius: 4px;
+  display: flex;
+  gap: 10px;
 }
+
+.copy-code-button {
+  background-color:dodgerblue;
+  color:white;
+  width: 30px;
+  height: 20px;
+  border:0;
+  display: flex;
+  align-items: center;
+}
+
+
+.el-icon-copy {
+  margin-right: 5px;
+}
+
 </style>
diff --git a/src/views/system/reco-topic.vue b/src/views/system/reco-topic.vue
index 7eca35b..61b33bf 100644
--- a/src/views/system/reco-topic.vue
+++ b/src/views/system/reco-topic.vue
@@ -4,10 +4,3 @@
     </div>
 
 </template>
-
-<script lang="ts">
-import { ref, reactive } from 'vue';
-</script>
-
-<style lang="scss" scoped>
-</style>
\ No newline at end of file
diff --git a/src/views/system/simu-defense.vue b/src/views/system/simu-defense.vue
index 9273afa..db8dd42 100644
--- a/src/views/system/simu-defense.vue
+++ b/src/views/system/simu-defense.vue
@@ -6,6 +6,11 @@
 </template>
 
 <script lang="ts">
+import { ref } from 'vue';
+
+const data = ref({});
+
+
 </script>
 
 <style lang="scss" scoped>