Update ChatWindow.js : 문제 있는 표시

Author: minseok128

frontend/src/chat/ChatWindow.js

@@ -52,7 +52,7 @@ function ChatWindow({ messages, setMessages, isGenerated, setIsGenerated }) {
       if (chatEndRef.current) {
         chatEndRef.current.scrollIntoView({ behavior: 'smooth' });
       }
-    }, 100); 
+    }, 100);
 
     return () => clearTimeout(timer);
   }, [imgFile]);
@@ -125,6 +125,17 @@ function ChatWindow({ messages, setMessages, isGenerated, setIsGenerated }) {
               ]);
               setChatStatus(data.chatStatus);
             }
+            else if (data.status === 'moderation') {
+              setMessages((messages) => {
+                const updatedMessages = [...messages];
+                updatedMessages[updatedMessages.length - 1] = {
+                  ...updatedMessages[updatedMessages.length - 1],
+                  flag: 'moderation',
+                };
+                setChatStatus(data.chatStatus);
+                return updatedMessages;
+              });
+            }
             setIsLoading(false);
           })
           .catch((error) => console.error('Error:', error));
@@ -257,14 +268,18 @@ function ChatWindow({ messages, setMessages, isGenerated, setIsGenerated }) {
           if (message.role === 'photo') {
             return (
               <div className={styles.message_assistant}>
-                <PhotoDrop key={index} file={imgFile} setFile={setImgFile} chatEndRef={chatEndRef}/>
+                <PhotoDrop key={index} file={imgFile} setFile={setImgFile} chatEndRef={chatEndRef} />
               </div>
             );
           } else {
             let messageClass =
-              message.role === 'user'
-                ? styles.message_user
-                : styles.message_assistant;
+              message.flag === 'moderation'
+                ? styles.message_moderation
+                : message.role === 'user'
+                  ? styles.message_user
+                  : message.role === 'assistant'
+                    ? styles.message_assistant
+                    : '';
             if (message.role === 'user' || message.role === 'assistant') {
               let timeNow = message.time.slice(0, -3);
               return (