diff --git a/src/pages/chats/[id]/index.vue b/src/pages/chats/[id]/index.vue
index a0abfc1..fc7a6bb 100644
--- a/src/pages/chats/[id]/index.vue
+++ b/src/pages/chats/[id]/index.vue
@@ -14,6 +14,7 @@
           <div v-else-if="message.role == 'system'">系统</div>
 
           <div v-else-if="message.role == 'file'" class="text-right">文件</div>
+          <div v-else-if="message.role == 'image'" class="text-right">图片</div>
           <div v-else-if="message.role == 'user'" class="text-right">用户</div>
 
           <div v-else>
@@ -35,10 +36,15 @@
         </div>
         <div v-else-if="message.role == 'image'" class="text-right">
           <img
-            :src="imageBaseUrl + '/' + message.content + '/download'"
+            :src="fileBaseUrl + '/' + message.content + '/download'"
             width="30%"
           />
         </div>
+        <div v-else-if="message.role == 'file'" class="text-right">
+          其他文件：<a :href="fileBaseUrl + '/' + message.content + '/download'"
+            >下载</a
+          >
+        </div>
         <div v-else>
           {{ message.content }}
         </div>
@@ -107,7 +113,7 @@ const toolError = ref(false);
 const toolCalling = ref(false);
 const fileUpload = ref();
 const uploading = ref(false);
-const imageBaseUrl = config.backend + "/api/v1/files";
+const fileBaseUrl = config.backend + "/api/v1/files";
 
 document.addEventListener("paste", function (event) {
   const items = event.clipboardData && event.clipboardData.items;
@@ -243,7 +249,16 @@ const getMessages = () => {
   api.ChatMessage.apiV1ChatsIdMessagesGet(chatId).then((res) => {
     messages.value.data = [];
     res.data.data?.forEach((message: any) => {
-      if (message.role === "assistant" || message.role === "user") {
+      if (message.role === "file") {
+        // 如果 mime_type 是 image/
+        if (message.file.mime_type.startsWith("image/")) {
+          // 设置 content
+          message.role = "image";
+        }
+
+        message.content = message.file.id;
+        messages.value.data?.push(message);
+      } else if (message.role === "assistant" || message.role === "user") {
         messages.value.data?.push(message);
       }
     });
diff --git a/src/pages/public_chat/index.vue b/src/pages/public_chat/index.vue
index f7a82fc..502fbda 100644
--- a/src/pages/public_chat/index.vue
+++ b/src/pages/public_chat/index.vue
@@ -272,7 +272,16 @@ const getMessages = () => {
   ).then((res) => {
     messages.value.data = [];
     res.data.data?.forEach((message: any) => {
-      if (message.role === "assistant" || message.role === "user") {
+      if (message.role === "file") {
+        // 如果 mime_type 是 image/
+        if (message.file.mime_type.startsWith("image/")) {
+          // 设置 content
+          message.role = "image";
+        }
+
+        message.content = message.file.id;
+        messages.value.data?.push(message);
+      } else if (message.role === "assistant" || message.role === "user") {
         messages.value.data?.push(message);
       }
     });