diff --git a/src/views/ChatPanel/index.tsx b/src/views/ChatPanel/index.tsx index b67c4bf..1add727 100644 --- a/src/views/ChatPanel/index.tsx +++ b/src/views/ChatPanel/index.tsx @@ -1,10 +1,10 @@ import * as React from 'react'; import { useState, useEffect, useRef } from 'react'; -import { Center, Container, Stack, px } from '@mantine/core'; +import { Alert, Center, Container, Stack, px } from '@mantine/core'; import { ScrollArea } from '@mantine/core'; import { Button } from '@mantine/core'; import { useListState, useResizeObserver, useTimeout, useViewportSize } from '@mantine/hooks'; -import { IconPlayerStop, IconRotateDot } from '@tabler/icons-react'; +import { IconAlertCircle, IconPlayerStop, IconRotateDot } from '@tabler/icons-react'; import messageUtil from '../../util/MessageUtil'; import InputMessage from './InputMessage'; @@ -17,7 +17,7 @@ const chatPanel = () => { const [currentMessage, setCurrentMessage] = useState(''); const [generating, setGenerating] = useState(false); const [responsed, setResponsed] = useState(false); - const [hasError, setHasError] = useState(false); + const [hasError, setHasError] = useState(''); const { height, width } = useViewportSize(); const [scrollPosition, onScrollPositionChange] = useState({ x: 0, y: 0 }); const [stopScrolling, setStopScrolling] = useState(false); @@ -42,11 +42,10 @@ const chatPanel = () => { setResponsed(true); }); messageUtil.registerHandler('receiveMessage', (message: { text: string; isError: boolean }) => { - setCurrentMessage(message.text); setGenerating(false); setResponsed(true); if (message.isError) { - setHasError(true); + setHasError(message.text); } }); messageUtil.registerHandler('loadHistoryMessages', (message: { command: string; entries: [{ hash: '', user: '', date: '', request: '', response: '', context: [{ content: '', role: '' }] }] }) => { @@ -121,7 +120,7 @@ const chatPanel = () => { command: 'regeneration' }); messageHandlers.pop(); - setHasError(false); + setHasError(''); setGenerating(true); setResponsed(false); setCurrentMessage(''); @@ -182,6 +181,11 @@ const chatPanel = () => { // onScrollPositionChange={onScrollPositionChange} viewportRef={scrollViewport}> + {hasError && + + {hasError} + + }