Fix: floating widget match style with original one (#10317)

### What problem does this PR solve?

These changes are intended to implement the remaining functionalities of
the fullscreen widget.

The question arises: how to display document prieview of PDFs in this
floating widget?
- simply enlarge the widget window
- implement zoom in/out
- render outside the iframe?

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

web/src/components/floating-chat-widget-markdown.less

@@ -0,0 +1,58 @@
+/* floating-chat-widget-markdown.less */
+
+.widget-citation-popover {
+  max-width: 90vw;
+  /* Use viewport width for better responsiveness */
+  width: max-content;
+
+  .ant-popover-inner {
+    max-height: 400px;
+    overflow-y: auto;
+  }
+
+  .ant-popover-inner-content {
+    padding: 12px;
+  }
+}
+
+/* Responsive breakpoints for popover width */
+@media (min-width: 480px) {
+  .widget-citation-popover {
+    max-width: 360px;
+  }
+}
+
+.widget-citation-content {
+
+  p,
+  div,
+  span,
+  button {
+    word-break: break-word;
+    overflow-wrap: break-word;
+    white-space: normal;
+  }
+}
+
+.floating-chat-widget {
+
+  /* General styles for markdown content within the widget */
+  p,
+  div,
+  ul,
+  ol,
+  blockquote {
+    line-height: 1.6;
+  }
+
+  /* Enhanced image styles */
+  img,
+  .ant-image,
+  .ant-image-img {
+    max-width: 100% !important;
+    height: auto !important;
+    border-radius: 8px;
+    margin: 8px 0 !important;
+    display: inline-block !important;
+  }
+}

web/src/components/floating-chat-widget-markdown.tsx

@@ -0,0 +1,199 @@
+import Image from '@/components/image';
+import SvgIcon from '@/components/svg-icon';
+import { useFetchDocumentThumbnailsByIds, useGetDocumentUrl } from '@/hooks/document-hooks';
+import { IReference, IReferenceChunk } from '@/interfaces/database/chat';
+import { preprocessLaTeX, replaceThinkToSection, showImage } from '@/utils/chat';
+import { getExtension } from '@/utils/document-util';
+import { InfoCircleOutlined } from '@ant-design/icons';
+import { Button, Flex, Popover, Tooltip } from 'antd';
+import classNames from 'classnames';
+import DOMPurify from 'dompurify';
+import { omit } from 'lodash';
+import { pipe } from 'lodash/fp';
+import 'katex/dist/katex.min.css';
+import { useCallback, useEffect, useMemo } from 'react';
+import { useTranslation } from 'react-i18next';
+import Markdown from 'react-markdown';
+import reactStringReplace from 'react-string-replace';
+import SyntaxHighlighter from 'react-syntax-highlighter';
+import { oneDark, oneLight } from 'react-syntax-highlighter/dist/esm/styles/prism';
+import rehypeKatex from 'rehype-katex';
+import rehypeRaw from 'rehype-raw';
+import remarkGfm from 'remark-gfm';
+import remarkMath from 'remark-math';
+import { visitParents } from 'unist-util-visit-parents';
+import { currentReg, replaceTextByOldReg } from '../pages/next-chats/utils';
+import styles from './floating-chat-widget-markdown.less';
+import { useIsDarkTheme } from './theme-provider';
+
+const getChunkIndex = (match: string) => Number(match.replace(/\[|\]/g, ''));
+
+const FloatingChatWidgetMarkdown = ({
+  reference,
+  clickDocumentButton,
+  content,
+}: {
+  content: string;
+  loading: boolean;
+  reference: IReference;
+  clickDocumentButton?: (documentId: string, chunk: IReferenceChunk) => void;
+}) => {
+  const { t } = useTranslation();
+  const { setDocumentIds, data: fileThumbnails } = useFetchDocumentThumbnailsByIds();
+  const getDocumentUrl = useGetDocumentUrl();
+  const isDarkTheme = useIsDarkTheme();
+
+  const contentWithCursor = useMemo(() => {
+    let text = content === '' ? t('chat.searching') : content;
+    const nextText = replaceTextByOldReg(text);
+    return pipe(replaceThinkToSection, preprocessLaTeX)(nextText);
+  }, [content, t]);
+
+  useEffect(() => {
+    const docAggs = reference?.doc_aggs;
+    const docList = Array.isArray(docAggs) ? docAggs : Object.values(docAggs ?? {});
+    setDocumentIds(docList.map((x: any) => x.doc_id).filter(Boolean));
+  }, [reference, setDocumentIds]);
+
+  const handleDocumentButtonClick = useCallback((documentId: string, chunk: IReferenceChunk, isPdf: boolean, documentUrl?: string) => () => {
+    if (!documentId) return;
+    if (!isPdf && documentUrl) {
+      window.open(documentUrl, '_blank');
+    } else if (clickDocumentButton) {
+      clickDocumentButton(documentId, chunk);
+    }
+  }, [clickDocumentButton]);
+
+  const rehypeWrapReference = () => (tree: any) => {
+    visitParents(tree, 'text', (node, ancestors) => {
+      const latestAncestor = ancestors[ancestors.length - 1];
+      if (latestAncestor.tagName !== 'custom-typography' && latestAncestor.tagName !== 'code') {
+        node.type = 'element';
+        node.tagName = 'custom-typography';
+        node.properties = {};
+        node.children = [{ type: 'text', value: node.value }];
+      }
+    });
+  };
+
+  const getReferenceInfo = useCallback((chunkIndex: number) => {
+    const chunkItem = reference?.chunks?.[chunkIndex];
+    if (!chunkItem) return null;
+    const docAggsArray = Array.isArray(reference?.doc_aggs) ? reference.doc_aggs : Object.values(reference?.doc_aggs ?? {});
+    const document = docAggsArray.find((x: any) => x?.doc_id === chunkItem?.document_id) as any;
+    const documentId = document?.doc_id;
+    const documentUrl = document?.url ?? (documentId ? getDocumentUrl(documentId) : undefined);
+    const fileThumbnail = documentId ? fileThumbnails[documentId] : '';
+    const fileExtension = documentId ? getExtension(document?.doc_name ?? '') : '';
+    return { documentUrl, fileThumbnail, fileExtension, imageId: chunkItem.image_id, chunkItem, documentId, document };
+  }, [fileThumbnails, reference, getDocumentUrl]);
+
+  const getPopoverContent = useCallback((chunkIndex: number) => {
+    const info = getReferenceInfo(chunkIndex);
+
+    if (!info) {
+      return <div className="p-2 text-xs text-red-500">Error: Missing document information.</div>;
+    }
+
+    const { documentUrl, fileThumbnail, fileExtension, imageId, chunkItem, documentId, document } = info;
+
+    return (
+      <div key={`popover-content-${chunkItem.id}`} className="flex gap-2 widget-citation-content">
+        {imageId && (
+          <Popover placement="left" content={<Image id={imageId} className="max-w-[80vw] max-h-[60vh] rounded" />}>
+            <Image id={imageId} className="w-24 h-24 object-contain rounded m-1 cursor-pointer" />
+          </Popover>
+        )}
+        <div className="space-y-2 flex-1 min-w-0">
+          <div
+            dangerouslySetInnerHTML={{ __html: DOMPurify.sanitize(chunkItem?.content ?? '') }}
+            className="max-h-[250px] overflow-y-auto text-xs leading-relaxed p-2 bg-gray-50 dark:bg-gray-800 rounded prose-sm"
+          ></div>
+          {documentId && (
+            <Flex gap={'small'} align="center">
+              {fileThumbnail ? (
+                <img src={fileThumbnail} alt={document?.doc_name} className="w-6 h-6 rounded" />
+              ) : (
+                <SvgIcon name={`file-icon/${fileExtension}`} width={20} />
+              )}
+              <Tooltip title={!documentUrl && fileExtension !== 'pdf' ? 'Document link unavailable' : document.doc_name}>
+                <Button
+                  type="link"
+                  size="small"
+                  className="p-0 text-xs break-words h-auto text-left flex-1"
+                  onClick={handleDocumentButtonClick(documentId, chunkItem, fileExtension === 'pdf', documentUrl)}
+                  disabled={!documentUrl && fileExtension !== 'pdf'}
+                  style={{ whiteSpace: 'normal' }}
+                >
+                  <span className="truncate">{document?.doc_name ?? 'Unnamed Document'}</span>
+                </Button>
+              </Tooltip>
+            </Flex>
+          )}
+        </div>
+      </div>
+    );
+  }, [getReferenceInfo, handleDocumentButtonClick]);
+
+  const renderReference = useCallback((text: string) => {
+    return reactStringReplace(text, currentReg, (match, i) => {
+      const chunkIndex = getChunkIndex(match);
+      const info = getReferenceInfo(chunkIndex);
+
+      if (!info) {
+        return <Tooltip key={`err-tooltip-${i}`} title="Reference unavailable"><InfoCircleOutlined className={styles.referenceIcon} /></Tooltip>;
+      }
+
+      const { imageId, chunkItem, documentId, fileExtension, documentUrl } = info;
+
+      if (showImage(chunkItem?.doc_type)) {
+        return <Image key={`img-${i}`} id={imageId} className="block object-contain max-w-full max-h-48 rounded my-2 cursor-pointer" onClick={handleDocumentButtonClick(documentId, chunkItem, fileExtension === 'pdf', documentUrl)} />;
+      }
+
+      return (
+        <Popover
+          content={getPopoverContent(chunkIndex)}
+          key={`popover-${i}`}
+        >
+          <InfoCircleOutlined className={styles.referenceIcon} />
+        </Popover>
+      );
+    });
+  }, [getPopoverContent, getReferenceInfo, handleDocumentButtonClick]);
+
+  return (
+    <div className="floating-chat-widget">
+      <Markdown
+        rehypePlugins={[rehypeWrapReference, rehypeKatex, rehypeRaw]}
+        remarkPlugins={[remarkGfm, remarkMath]}
+        className="text-sm leading-relaxed space-y-2 prose-sm max-w-full"
+        components={{
+          'custom-typography': ({ children }: { children: string }) => renderReference(children),
+          code(props: any) {
+            const { children, className, node, ...rest } = props;
+            const match = /language-(\w+)/.exec(className || '');
+            return match ? (
+              <SyntaxHighlighter
+                {...omit(rest, 'inline')}
+                PreTag="div"
+                language={match[1]}
+                style={isDarkTheme ? oneDark : oneLight}
+                wrapLongLines
+              >
+                {String(children).replace(/\n$/, '')}
+              </SyntaxHighlighter>
+            ) : (
+              <code {...rest} className={classNames(className, 'text-wrap text-xs bg-gray-200 dark:bg-gray-700 px-1 py-0.5 rounded')}>
+                {children}
+              </code>
+            );
+          },
+        } as any}
+      >
+        {contentWithCursor}
+      </Markdown>
+    </div>
+  );
+};
+
+export default FloatingChatWidgetMarkdown;

web/src/components/floating-chat-widget.tsx

@@ -2,8 +2,10 @@ import { MessageType, SharedFrom } from '@/constants/chat';
 import { useFetchNextConversationSSE } from '@/hooks/chat-hooks';
 import { useFetchFlowSSE } from '@/hooks/flow-hooks';
 import { useFetchExternalChatInfo } from '@/hooks/use-chat-request';
+import { useClickDrawer } from '@/components/pdf-drawer/hooks';
 import i18n from '@/locales/config';
 import { MessageCircle, Minimize2, Send, X } from 'lucide-react';
+import PdfDrawer from '@/components/pdf-drawer';
 import React, {
   useCallback,
   useEffect,
@@ -15,6 +17,7 @@ import {
   useGetSharedChatSearchParams,
   useSendSharedMessage,
 } from '../pages/next-chats/hooks/use-send-shared-message';
+import FloatingChatWidgetMarkdown from './floating-chat-widget-markdown';
 
 const FloatingChatWidget = () => {
   const [isOpen, setIsOpen] = useState(false);
@@ -63,6 +66,14 @@ const FloatingChatWidget = () => {
 
   const { data: avatarData } = useFetchAvatar();
 
+  const { visible, hideModal, documentId, selectedChunk, clickDocumentButton } =
+    useClickDrawer();
+
+  // PDF drawer state tracking
+  useEffect(() => {
+    // Drawer state management
+  }, [visible, documentId, selectedChunk]);
+
   // Play sound when opening
   const playNotificationSound = useCallback(() => {
     try {
@@ -223,7 +234,7 @@ const FloatingChatWidget = () => {
     const syntheticEvent = {
       target: { value: inputValue },
       currentTarget: { value: inputValue },
-      preventDefault: () => {},
+      preventDefault: () => { },
     } as any;
 
     handleInputChange(syntheticEvent);
@@ -314,9 +325,8 @@ const FloatingChatWidget = () => {
               '*',
             );
           }}
-          className={`w-14 h-14 bg-blue-600 hover:bg-blue-700 text-white rounded-full transition-all duration-300 flex items-center justify-center group ${
+          className={`w-14 h-14 bg-blue-600 hover:bg-blue-700 text-white rounded-full transition-all duration-300 flex items-center justify-center group ${isOpen ? 'scale-95' : 'scale-100 hover:scale-105'
-            isOpen ? 'scale-95' : 'scale-100 hover:scale-105'
+            }`}
-          }`}
         >
           <div
             className={`transition-transform duration-300 ${isOpen ? 'rotate-45' : 'rotate-0'}`}
@@ -343,9 +353,8 @@ const FloatingChatWidget = () => {
       >
         <button
           onClick={toggleChat}
-          className={`w-14 h-14 bg-blue-600 hover:bg-blue-700 text-white rounded-full transition-all duration-300 flex items-center justify-center group ${
+          className={`w-14 h-14 bg-blue-600 hover:bg-blue-700 text-white rounded-full transition-all duration-300 flex items-center justify-center group ${isOpen ? 'scale-95' : 'scale-100 hover:scale-105'
-            isOpen ? 'scale-95' : 'scale-100 hover:scale-105'
+            }`}
-          }`}
         >
           <div
             className={`transition-transform duration-300 ${isOpen ? 'rotate-45' : 'rotate-0'}`}
@@ -367,127 +376,143 @@ const FloatingChatWidget = () => {
   if (mode === 'window') {
     // Only render the chat window (always open)
     return (
-      <div
+      <>
-        className={`fixed top-0 left-0 z-50 bg-blue-600 rounded-2xl transition-all duration-300 ease-out h-[500px] w-[380px] overflow-hidden ${isLoaded ? 'opacity-100' : 'opacity-0'}`}
+        <div
-      >
+          className={`fixed top-0 left-0 z-50 bg-blue-600 rounded-2xl transition-all duration-300 ease-out h-[500px] w-[380px] overflow-hidden ${isLoaded ? 'opacity-100' : 'opacity-0'}`}
-        {/* Header */}
+        >
-        <div className="flex items-center justify-between p-4 bg-gradient-to-r from-blue-600 to-blue-700 text-white rounded-t-2xl">
+          {/* Header */}
-          <div className="flex items-center space-x-3">
+          <div className="flex items-center justify-between p-4 bg-gradient-to-r from-blue-600 to-blue-700 text-white rounded-t-2xl">
-            <div className="w-8 h-8 bg-white bg-opacity-20 rounded-full flex items-center justify-center">
+            <div className="flex items-center space-x-3">
-              <MessageCircle size={18} />
+              <div className="w-8 h-8 bg-white bg-opacity-20 rounded-full flex items-center justify-center">
-            </div>
+                <MessageCircle size={18} />
-            <div>
+              </div>
-              <h3 className="font-semibold text-sm">
+              <div>
-                {chatInfo?.title || 'Chat Support'}
+                <h3 className="font-semibold text-sm">
-              </h3>
+                  {chatInfo?.title || 'Chat Support'}
-              <p className="text-xs text-blue-100">
+                </h3>
-                We typically reply instantly
+                <p className="text-xs text-blue-100">
-              </p>
+                  We typically reply instantly
+                </p>
+              </div>
             </div>
           </div>
-        </div>
 
-        {/* Messages and Input */}
+          {/* Messages and Input */}
-        <div
-          className="flex flex-col h-[436px] bg-white"
-          style={{ borderRadius: '0 0 16px 16px' }}
-        >
           <div
-            className="flex-1 overflow-y-auto p-4 space-y-4"
+            className="flex flex-col h-[436px] bg-white"
-            onWheel={(e) => {
+            style={{ borderRadius: '0 0 16px 16px' }}
-              const element = e.currentTarget;
-              const isAtTop = element.scrollTop === 0;
-              const isAtBottom =
-                element.scrollTop + element.clientHeight >=
-                element.scrollHeight - 1;
-
-              // Allow scroll to pass through to parent when at boundaries
-              if ((isAtTop && e.deltaY < 0) || (isAtBottom && e.deltaY > 0)) {
-                e.preventDefault();
-                // Let the parent handle the scroll
-                window.parent.postMessage(
-                  {
-                    type: 'SCROLL_PASSTHROUGH',
-                    deltaY: e.deltaY,
-                  },
-                  '*',
-                );
-              }
-            }}
           >
-            {displayMessages?.map((message, index) => (
+            <div
-              <div
+              className="flex-1 overflow-y-auto p-4 space-y-4"
-                key={index}
+              onWheel={(e) => {
-                className={`flex ${message.role === MessageType.User ? 'justify-end' : 'justify-start'}`}
+                const element = e.currentTarget;
-              >
+                const isAtTop = element.scrollTop === 0;
+                const isAtBottom =
+                  element.scrollTop + element.clientHeight >=
+                  element.scrollHeight - 1;
+
+                // Allow scroll to pass through to parent when at boundaries
+                if ((isAtTop && e.deltaY < 0) || (isAtBottom && e.deltaY > 0)) {
+                  e.preventDefault();
+                  // Let the parent handle the scroll
+                  window.parent.postMessage(
+                    {
+                      type: 'SCROLL_PASSTHROUGH',
+                      deltaY: e.deltaY,
+                    },
+                    '*',
+                  );
+                }
+              }}
+            >
+              {displayMessages?.map((message, index) => (
                 <div
-                  className={`max-w-[280px] px-4 py-2 rounded-2xl ${
+                  key={index}
-                    message.role === MessageType.User
+                  className={`flex ${message.role === MessageType.User ? 'justify-end' : 'justify-start'}`}
+                >
+                  <div
+                    className={`max-w-[280px] px-4 py-2 rounded-2xl ${message.role === MessageType.User
                       ? 'bg-blue-600 text-white rounded-br-md'
                       : 'bg-gray-100 text-gray-800 rounded-bl-md'
-                  }`}
+                      }`}
+                  >
+                    {message.role === MessageType.User ? (
+                      <p className="text-sm leading-relaxed whitespace-pre-wrap">
+                        {message.content}
+                      </p>
+                    ) : (
+                      <FloatingChatWidgetMarkdown
+                        loading={false}
+                        content={message.content}
+                        reference={message.reference || { doc_aggs: [], chunks: [], total: 0 }}
+                        clickDocumentButton={clickDocumentButton}
+                      />
+                    )}
+                  </div>
+                </div>
+              ))}
+
+              {/* Clean Typing Indicator */}
+              {sendLoading && !enableStreaming && (
+                <div className="flex justify-start pl-4">
+                  <div className="flex space-x-1">
+                    <div className="w-2 h-2 bg-blue-500 rounded-full animate-bounce"></div>
+                    <div
+                      className="w-2 h-2 bg-blue-500 rounded-full animate-bounce"
+                      style={{ animationDelay: '0.1s' }}
+                    ></div>
+                    <div
+                      className="w-2 h-2 bg-blue-500 rounded-full animate-bounce"
+                      style={{ animationDelay: '0.2s' }}
+                    ></div>
+                  </div>
+                </div>
+              )}
+
+              <div ref={messagesEndRef} />
+            </div>
+
+            {/* Input Area */}
+            <div className="border-t border-gray-200 p-4">
+              <div className="flex items-end space-x-3">
+                <div className="flex-1">
+                  <textarea
+                    value={inputValue}
+                    onChange={(e) => {
+                      const newValue = e.target.value;
+                      setInputValue(newValue);
+                      handleInputChange(e);
+                    }}
+                    onKeyPress={handleKeyPress}
+                    placeholder="Type your message..."
+                    rows={1}
+                    className="w-full resize-none border border-gray-300 rounded-2xl px-4 py-3 text-sm focus:outline-none focus:ring-2 focus:ring-blue-500 focus:border-transparent"
+                    style={{ minHeight: '44px', maxHeight: '120px' }}
+                    disabled={hasError || sendLoading}
+                  />
+                </div>
+                <button
+                  onClick={handleSendMessage}
+                  disabled={!inputValue.trim() || sendLoading}
+                  className="p-3 bg-blue-600 text-white rounded-full hover:bg-blue-700 disabled:opacity-50 disabled:cursor-not-allowed transition-colors"
                 >
-                  <p className="text-sm leading-relaxed whitespace-pre-wrap">
+                  <Send size={18} />
-                    {message.content}
+                </button>
-                  </p>
-                </div>
               </div>
-            ))}
-
-            {/* Clean Typing Indicator */}
-            {sendLoading && !enableStreaming && (
-              <div className="flex justify-start pl-4">
-                <div className="flex space-x-1">
-                  <div className="w-2 h-2 bg-blue-500 rounded-full animate-bounce"></div>
-                  <div
-                    className="w-2 h-2 bg-blue-500 rounded-full animate-bounce"
-                    style={{ animationDelay: '0.1s' }}
-                  ></div>
-                  <div
-                    className="w-2 h-2 bg-blue-500 rounded-full animate-bounce"
-                    style={{ animationDelay: '0.2s' }}
-                  ></div>
-                </div>
-              </div>
-            )}
-
-            <div ref={messagesEndRef} />
-          </div>
-
-          {/* Input Area */}
-          <div className="border-t border-gray-200 p-4">
-            <div className="flex items-end space-x-3">
-              <div className="flex-1">
-                <textarea
-                  value={inputValue}
-                  onChange={(e) => {
-                    const newValue = e.target.value;
-                    setInputValue(newValue);
-                    handleInputChange(e);
-                  }}
-                  onKeyPress={handleKeyPress}
-                  placeholder="Type your message..."
-                  rows={1}
-                  className="w-full resize-none border border-gray-300 rounded-2xl px-4 py-3 text-sm focus:outline-none focus:ring-2 focus:ring-blue-500 focus:border-transparent"
-                  style={{ minHeight: '44px', maxHeight: '120px' }}
-                  disabled={hasError || sendLoading}
-                />
-              </div>
-              <button
-                onClick={handleSendMessage}
-                disabled={!inputValue.trim() || sendLoading}
-                className="p-3 bg-blue-600 text-white rounded-full hover:bg-blue-700 disabled:opacity-50 disabled:cursor-not-allowed transition-colors"
-              >
-                <Send size={18} />
-              </button>
             </div>
           </div>
         </div>
-      </div>
+        <PdfDrawer
+          visible={visible}
+          hideModal={hideModal}
+          documentId={documentId}
+          chunk={selectedChunk}
+          width={'100vw'}
+          height={'100vh'}
+        />
+      </>
     );
-  }
+  }  // Full mode - render everything together (original behavior)
-
-  // Full mode - render everything together (original behavior)
   return (
     <div
       className={`transition-opacity duration-300 ${isLoaded ? 'opacity-100' : 'opacity-0'}`}
@@ -495,9 +520,8 @@ const FloatingChatWidget = () => {
       {/* Chat Widget Container */}
       {isOpen && (
         <div
-          className={`fixed bottom-24 right-6 z-50 bg-blue-600 rounded-2xl transition-all duration-300 ease-out ${
+          className={`fixed bottom-24 right-6 z-50 bg-blue-600 rounded-2xl transition-all duration-300 ease-out ${isMinimized ? 'h-16' : 'h-[500px]'
-            isMinimized ? 'h-16' : 'h-[500px]'
+            } w-[380px] overflow-hidden`}
-          } w-[380px] overflow-hidden`}
         >
           {/* Header */}
           <div className="flex items-center justify-between p-4 bg-gradient-to-r from-blue-600 to-blue-700 text-white rounded-t-2xl">
@@ -568,15 +592,23 @@ const FloatingChatWidget = () => {
                     className={`flex ${message.role === MessageType.User ? 'justify-end' : 'justify-start'}`}
                   >
                     <div
-                      className={`max-w-[280px] px-4 py-2 rounded-2xl ${
+                      className={`max-w-[280px] px-4 py-2 rounded-2xl ${message.role === MessageType.User
-                        message.role === MessageType.User
+                        ? 'bg-blue-600 text-white rounded-br-md'
-                          ? 'bg-blue-600 text-white rounded-br-md'
+                        : 'bg-gray-100 text-gray-800 rounded-bl-md'
-                          : 'bg-gray-100 text-gray-800 rounded-bl-md'
+                        }`}
-                      }`}
                     >
-                      <p className="text-sm leading-relaxed whitespace-pre-wrap">
+                      {message.role === MessageType.User ? (
-                        {message.content}
+                        <p className="text-sm leading-relaxed whitespace-pre-wrap">
-                      </p>
+                          {message.content}
+                        </p>
+                      ) : (
+                        <FloatingChatWidgetMarkdown
+                          loading={false}
+                          content={message.content}
+                          reference={message.reference || { doc_aggs: [], chunks: [], total: 0 }}
+                          clickDocumentButton={clickDocumentButton}
+                        />
+                      )}
                     </div>
                   </div>
                 ))}
@@ -641,9 +673,8 @@ const FloatingChatWidget = () => {
       <div className="fixed bottom-6 right-6 z-50">
         <button
           onClick={toggleChat}
-          className={`w-14 h-14 bg-blue-600 hover:bg-blue-700 text-white rounded-full transition-all duration-300 flex items-center justify-center group ${
+          className={`w-14 h-14 bg-blue-600 hover:bg-blue-700 text-white rounded-full transition-all duration-300 flex items-center justify-center group ${isOpen ? 'scale-95' : 'scale-100 hover:scale-105'
-            isOpen ? 'scale-95' : 'scale-100 hover:scale-105'
+            }`}
-          }`}
         >
           <div
             className={`transition-transform duration-300 ${isOpen ? 'rotate-45' : 'rotate-0'}`}
@@ -659,6 +690,12 @@ const FloatingChatWidget = () => {
           </div>
         )}
       </div>
+      <PdfDrawer
+        visible={visible}
+        hideModal={hideModal}
+        documentId={documentId}
+        chunk={selectedChunk}
+      />
     </div>
   );
 };

web/src/components/pdf-drawer/index.tsx

@@ -7,6 +7,8 @@ import DocumentPreviewer from '../pdf-previewer';
 interface IProps extends IModalProps<any> {
   documentId: string;
   chunk: IChunk | IReferenceChunk;
+  width?: string | number;
+  height?: string | number;
 }
 
 export const PdfDrawer = ({
@@ -14,13 +16,16 @@ export const PdfDrawer = ({
   hideModal,
   documentId,
   chunk,
+  width = '50vw',
+  height,
 }: IProps) => {
   return (
     <Drawer
       title="Document Previewer"
       onClose={hideModal}
       open={visible}
-      width={'50vw'}
+      width={width}
+      height={height}
     >
       <DocumentPreviewer
         documentId={documentId}