DocsGPT/frontend/src/conversation/ConversationBubble.tsx

import { forwardRef } from 'react';
import Avatar from '../Avatar';
import { MESSAGE_TYPE } from './conversationModels';
import Alert from './../assets/alert.svg';

const ConversationBubble = forwardRef<
  HTMLDivElement,
  {
    message: string;
    type: MESSAGE_TYPE;
    className: string;
  }
>(function ConversationBubble({ message, type, className }, ref) {
  return (
    <div
      ref={ref}
      className={`flex rounded-3xl ${
        type === 'ANSWER'
          ? 'self-start bg-gray-1000'
          : type === 'ERROR'
          ? 'self-start bg-red-1000'
          : 'flex-row-reverse self-end bg-blue-1000 text-white'
      } py-7 px-5 ${className}`}
    >
      <Avatar avatar={type === 'QUESTION' ? '👤' : '🦖'}></Avatar>
      <div
        className={`${
          type === 'QUESTION' ? 'mr-5' : 'ml-5'
        } flex items-center ${
          type === 'ERROR'
            ? 'rounded-lg border border-red-2000 p-2 text-red-3000'
            : ''
        }`}
      >
        {type === 'ERROR' && (
          <img src={Alert} alt="alert" className="mr-2 inline" />
        )}
        <p className="whitespace-pre-wrap break-words">{message}</p>
      </div>
    </div>
  );
});

export default ConversationBubble;

// TODO : split question and answer into two diff JSX