From 251d6920d598c9f91b20ff5a8d8e7d095cedd2cf Mon Sep 17 00:00:00 2001
From: "l.tingting" <l.tingting@pku.edu.cn>
Date: Wed, 19 Jun 2024 16:26:13 +0800
Subject: [PATCH] use a variable to represent the max number of images

---
 app/components/chat.tsx | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/app/components/chat.tsx b/app/components/chat.tsx
index 061192504..bafe51cf6 100644
--- a/app/components/chat.tsx
+++ b/app/components/chat.tsx
@@ -99,6 +99,8 @@ import { getClientConfig } from "../config/client";
 import { useAllModels } from "../utils/hooks";
 import { MultimodalContent } from "../client/api";
 
+const maxImagesNumber = 3;
+
 const Markdown = dynamic(async () => (await import("./markdown")).Markdown, {
   loading: () => <LoadingIcon />,
 });
@@ -1150,8 +1152,8 @@ function _Chat() {
             );
             const imagesLength = images.length;
 
-            if (imagesLength > 3) {
-              images.splice(3, imagesLength - 3);
+            if (imagesLength > maxImagesNumber) {
+              images.splice(maxImagesNumber, imagesLength - maxImagesNumber);
             }
             setAttachImages(images);
           }
@@ -1182,7 +1184,7 @@ function _Chat() {
               .then((dataUrl) => {
                 imagesData.push(dataUrl);
                 if (
-                  imagesData.length === 3 ||
+                  imagesData.length === maxImagesNumber ||
                   imagesData.length === files.length
                 ) {
                   setUploading(false);
@@ -1200,8 +1202,8 @@ function _Chat() {
     );
 
     const imagesLength = images.length;
-    if (imagesLength > 3) {
-      images.splice(3, imagesLength - 3);
+    if (imagesLength > maxImagesNumber) {
+      images.splice(maxImagesNumber, imagesLength - maxImagesNumber);
     }
     setAttachImages(images);
   }