@stack-spot/ai-chat-widget 1.38.2-beta.2 → 2.0.0-beta.1

package/src/views/KnowledgeSources.tsx

@@ -1,16 +1,12 @@
-import { Button } from '@citric/core'
-import { Search } from '@citric/icons'
+import { Button, Tab } from '@stack-spot/citric-react'
 import { Placeholder } from '@stack-spot/portal-components/Placeholder'
 import { aiClient, dataIntegrationClient, workspaceAiClient } from '@stack-spot/portal-network'
-import { KnowledgeSourceItemResponse, VisibilityLevelEnum } from '@stack-spot/portal-network/api/ai'
+import { VisibilityLevelEnum } from '@stack-spot/portal-network/api/ai'
 import { WorkspaceResponse } from '@stack-spot/portal-network/api/workspace-ai'
 import { Dictionary, useTranslate } from '@stack-spot/portal-translate'
-import { difference, uniqBy } from 'lodash'
-import React, { ReactElement, useCallback, useEffect, useMemo, useRef, useState } from 'react'
-import { ButtonFavorite } from '../components/ButtonFavorite'
+import React, { useCallback, useEffect, useMemo, useRef } from 'react'
 import { NavigationComponent } from '../components/ComponentNavigator'
 import { DescribedCheckboxGroup } from '../components/form/DescribedCheckboxGroup'
-import { IconInput } from '../components/IconInput'
 import { RightPanelContentList } from '../components/RightPanelContentList'
 import { RightPanelTabs } from '../components/RightPanelTabs'
 import { WorkspaceTabNavigator } from '../components/WorkspaceTabNavigator'
@@ -28,8 +24,6 @@ export interface TabProps {
   onSubmit?: () => void,
 }
 
-type TabElement = { title: string, content: ReactElement }
-
 export const KnowledgeSources = () => {
   const t = useTranslate(dictionary)
   const panel = useWidgetState('panel')
@@ -69,15 +63,15 @@ const KnowledgeSourcesPanel = () => {
     allKS.current = chat.get('knowledgeSources') ?? []
   }, [chat])
 
-  const allTabsMap: Partial<Record<Scope, TabElement>> = {
-    favorite: { title: t.favorites, 
+  const allTabsMap: Partial<Record<Scope, Omit<Tab, 'key'>>> = {
+    favorite: { label: t.favorites, 
       content: <KnowledgeSourcesTab key="favorite" visibility="favorite" allKS={allKS} onSubmit={onSubmit} /> },
-    personal: { title: t.personal, 
+    personal: { label: t.personal, 
       content: <KnowledgeSourcesTab key="personal" visibility="personal" allKS={allKS} onSubmit={onSubmit} /> },
-    shared: { title: t.shared, content: <KnowledgeSourcesTab key="shared" visibility="shared" allKS={allKS} onSubmit={onSubmit} /> },
-    workspace: { title: t.spots,
+    shared: { label: t.shared, content: <KnowledgeSourcesTab key="shared" visibility="shared" allKS={allKS} onSubmit={onSubmit} /> },
+    workspace: { label: t.spots,
       content: <KnowledgeSourcesTabWorkspace key="workspace" visibility="workspace" allKS={allKS} onSubmit={onSubmit} /> },
-    account: { title: t.account, content: <KnowledgeSourcesTab key="account" visibility="account" allKS={allKS} onSubmit={onSubmit} /> },
+    account: { label: t.account, content: <KnowledgeSourcesTab key="account" visibility="account" allKS={allKS} onSubmit={onSubmit} /> },
   }
 
   const defaultScopes: Scope[] = ['favorite', 'personal', 'shared', 'workspace', 'account']
@@ -89,8 +83,8 @@ const KnowledgeSourcesPanel = () => {
     : rawScopes
 
   const tabs = scopesToRender
-    .map(scope => allTabsMap[scope])
-    .filter(Boolean) as TabElement[]
+    .map(scope => scope in allTabsMap ? { key: scope, ...allTabsMap[scope] } : undefined)
+    .filter(Boolean) as Tab[]
 
   return (isGroupResourcesByScope ? (
     <RightPanelTabs key={chat.id} tabs={tabs} />
@@ -101,14 +95,15 @@ const KnowledgeSourcesPanel = () => {
 
 export const KnowledgeSourcesTab = ({ visibility, allKS, onSubmit, workspaceId, showSubmitButton = true }: TabProps) => {
   const t = useTranslate(dictionary)
-  const [filter, setFilter] = useState('')
-
   const knowledgeSources = workspaceId
     ? workspaceAiClient.getKSFromWorkspaceAi.useQuery({ workspaceId })
     : aiClient.knowledgeSources.useQuery({
       visibility, order: 'a-to-z', types: ['snippet', 'api', 'event', 'custom'],
     })
-
+  const initialValue = useMemo(() => {
+    const currentlySelected = allKS.current?.map(ks => ks.id)
+    return knowledgeSources.filter(ks => currentlySelected?.includes(ks.slug))
+  }, [])
   const listFavorites = dataIntegrationClient.knowledgeSources.useQuery({ visibility: 'favorite' })
   const [addFavorite, pendingAddFav] = dataIntegrationClient.addFavoriteKnowledgeSource.useMutation()
   const [removeFavorite, pendingRemoveFav] = dataIntegrationClient.removeFavoriteKnowledgeSource.useMutation()
@@ -165,56 +160,19 @@ export const KnowledgeSourcesTab = ({ visibility, allKS, onSubmit, workspaceId,
     } 
   })
 
-  const [value, setValue] = useState<KnowledgeSourceItemResponse[]>((() => {
-    const currentlySelected = allKS.current?.map(ks => ks.id)
-    return knowledgeSources.filter(ks => currentlySelected?.includes(ks.slug))
-  })())
-  const filtered = useMemo(
-    () => filter 
-      // Recreate the list so that the favorites list is taken into account
-      ? knowledgeSources.filter(ks => value.includes(ks) || ks.name.toLocaleLowerCase().includes(filter.toLocaleLowerCase()))
-      : [...knowledgeSources]
-    , [knowledgeSources, filter, value, listFavorites],
-  )
-
-  const onChange = useCallback((newValue: KnowledgeSourceItemResponse[]) => {
-    setValue((current) => {
-      const added = difference(newValue, current)
-      const removed = difference(current, newValue)
-      allKS.current = allKS.current?.filter(ks => !removed.some(r => r.slug === ks.id)) ?? []
-      allKS.current = uniqBy([...allKS.current, ...added.map(ks => ({ id: ks.slug, label: ks.name }))], 'id')
-      return newValue
-    })
-  }, [])
-
   return (
     <>
       <div className="content">
-        <IconInput icon={<Search />} value={filter} onChange={setFilter} className="search" />
-        {!!filtered.length && <DescribedCheckboxGroup
-          options={filtered}
-          renderAfterElement={({ slug }) => (
-            <ButtonFavorite 
-              favorite={{ 
-                idOrSlug: slug, listFavorites, onAddFavorite, onRemoveFavorite }} />
-          )}
-          keygen={ks => ks.id}
-          value={value}
-          onChange={onChange}
-          renderLabel={ks => ks.name}
-          renderDescription={ks => ks.description}
-          optionClassName={ks => (filter && !ks.name.toLocaleLowerCase().includes(filter.toLocaleLowerCase()) && value.includes(ks))
-            ? 'filtered-out'
-            : ''
-          }
-          className="option-list"
-        />}
-        {!!filtered.length && !filtered.length && (
-          <Placeholder title={t.noSearchResults} description={t.noSearchResultsDescription} />
-        )}
-        {!filtered.length && <Placeholder title={t.noData} description={t.noDataDescription} className="no-data-placeholder" />}
+        <DescribedCheckboxGroup
+          options={knowledgeSources}
+          initialValue={initialValue}
+          globalSelection={allKS}
+          data={ks => ({ idOrSlug: ks.slug, description: ks.description, name: ks.name, listFavorites, onAddFavorite, onRemoveFavorite })}
+          emptyResults={<Placeholder title={t.noSearchResults} description={t.noSearchResultsDescription} />}
+          emptyDataset={<Placeholder title={t.noData} description={t.noDataDescription} className="no-data-placeholder" />}
+        />
       </div>
-      {!!filtered.length && showSubmitButton && <Button onClick={onSubmit}>{t.apply}</Button>}
+      {!!knowledgeSources.length && showSubmitButton && <Button onClick={onSubmit}>{t.apply}</Button>}
     </>
   )
 }

package/src/views/MessageInput/AgentSelector.tsx

@@ -1,5 +1,5 @@
-import { Flex, IconBox, Image } from '@citric/core'
-import { Agent } from '@citric/icons'
+import { Icon } from '@stack-spot/citric-icons'
+import { IconBox, Row, Text } from '@stack-spot/citric-react'
 import { AgentResponseWithBuiltIn, agentToolsClient, workspaceAiClient } from '@stack-spot/portal-network'
 import { useCallback } from 'react'
 import { Selector } from '../../components/Selector'
@@ -10,15 +10,17 @@ import { useAgentFavorites } from '../Agents/useAgentFavorites'
 type AgentWithSpaceName = AgentResponseWithBuiltIn
 
 const AgentItem = ({ avatar, name, spaceName }: AgentWithSpaceName) => {
-  const avatarComponent = avatar ? <Image width="32" height="32" radius="full" src={avatar} /> : <IconBox size="md"><Agent /></IconBox>
+  const avatarComponent = avatar
+    ? <img style={{ width: '32px', height: '32px', borderRadius: '50%' }} src={avatar} />
+    : <IconBox size="md" icon="Agent" style={{ backgroundColor: 'transparent' }} />
 
-  return <Flex flexWrap="nowrap" alignItems="center" sx={{ gap: '8px' }}>
+  return <Row gap="8px" style={{ flexWrap: 'nowrap' }}>
     {avatarComponent}
     <div>
-      <p className="selector-description">{spaceName}</p>
-      <p className="selector-title">{name}</p>
+      <Text className="selector-description">{spaceName}</Text>
+      <Text className="selector-title">{name}</Text>
     </div>
-  </Flex>
+  </Row>
 }
 
 export const AgentSelector = ({ inputRef, isTrial }: {
@@ -72,7 +74,7 @@ export const AgentSelector = ({ inputRef, isTrial }: {
     selectorConfig={{
       resourceName: 'Agent',
       shortcut: '@',
-      icon: <Agent />,
+      icon: <Icon icon="Agent" />,
       regex: agentRegex,
       urlBuilder: (agent) => `/agents/${agent?.id}`,
       searchProp: 'name',

package/src/views/MessageInput/ButtonAgent.tsx

@@ -1,6 +1,5 @@
-import { Flex, IconBox } from '@citric/core'
-import { Agent, TimesMini } from '@citric/icons'
-import { IconButton, Tooltip } from '@citric/ui'
+import { Icon } from '@stack-spot/citric-icons'
+import { Center, ImageButton, ImageWithFallback, Tooltip } from '@stack-spot/citric-react'
 import { MiniLogo } from '@stack-spot/portal-components/svg'
 import { agentToolsClient } from '@stack-spot/portal-network'
 import { useEffect } from 'react'
@@ -41,17 +40,30 @@ export const ButtonAgent = () => {
     <div className="button-group">
       {features.agent && (
         <div className="group-agent">
-          <IconButton aria-label={t.agent} title={t.agent} className="agent" onClick={() => widget.set('panel', 'agent')}>
+          <ImageButton
+            aria-label={t.agent}
+            title={t.agent}
+            className="agent"
+            onClick={() => widget.set('panel', 'agent')}
+          >
             {agentDefault?.avatar ? <img src={agentDefault.avatar} className="image" /> : <MiniLogo />}
-          </IconButton>
+          </ImageButton>
           {agent?.id && !isAgentDefault() &&
-            <Tooltip text={t.remove} >
-              <IconButton aria-label={t.remove} className="agent agent-selected" onClick={setAgentDefault}>
-                {agent?.image ? <img src={agent.image} className="image" /> : <IconBox className="image" size="xs"><Agent /></IconBox>}
-                <Flex className="icon-remove" alignContent="center" justifyContent="center">
-                  <TimesMini />
-                </Flex>
-              </IconButton>
+            <Tooltip content={t.remove} >
+              <ImageButton
+                aria-label={t.remove}
+                className="agent agent-selected"
+                onClick={setAgentDefault}
+              >
+                <ImageWithFallback
+                  src={agent.image ?? ''}
+                  className="image"
+                  fallback={<Icon className="image" size="xs" icon="Agent" />}
+                />
+                <Center className="icon-remove">
+                  <Icon icon="TimesMini" />
+                </Center>
+              </ImageButton>
             </Tooltip>
           }
         </div>

package/src/views/MessageInput/ButtonBar.tsx

@@ -1,6 +1,4 @@
-import { Flex } from '@citric/core'
-import { ArrowUp, Code, Stop } from '@citric/icons'
-import { IconButton } from '@citric/ui'
+import { IconButton, Row } from '@stack-spot/citric-react'
 import { useCurrentChat, useCurrentChatState, useWidget } from '../../context/hooks'
 import { useMessageInputDictionary } from './dictionary'
 import { SelectContent } from './SelectContent'
@@ -25,27 +23,33 @@ export const ButtonBar = ({ onSend, isLoading }: SelectionBarProps) => {
 
   return (
     <SelectionBarWrapper className="button-group">
-      <Flex sx={{ gap: '4px' }}>
+      <Row center={false} gap="4px">
         <SelectContent />
         {features.editor && (
-          <IconButton color="light" appearance="square" aria-label={t.code} title={t.code} onClick={() => widget.set('panel', 'editor')}>
-            <Code />
-          </IconButton>
+          <IconButton icon="Code" appearance="square" aria-label={t.code} title={t.code} onClick={() => widget.set('panel', 'editor')} />
         )}
-      </Flex>
-      <Flex>
-        {isLoading ? (
-          <IconButton appearance="square" color="inverse" aria-label={t.cancel} 
-            onClick={() => chat.abort()} className="send" title={t.cancel}>
-            <Stop />
-          </IconButton>
-        ) : (
-          <IconButton appearance="square" color="inverse" aria-label={t.send} 
-            onClick={onSend} className="send" title={t.send}>
-            <ArrowUp />
-          </IconButton>
-        )}
-      </Flex>
+      </Row>
+      {isLoading ? (
+        <IconButton
+          icon="Stop"
+          appearance="square"
+          colorScheme="inverse"
+          aria-label={t.cancel}
+          onClick={() => chat.abort()}
+          className="send"
+          title={t.cancel}
+        />
+      ) : (
+        <IconButton
+          icon="ArrowUp"
+          appearance="square"
+          colorScheme="inverse"
+          aria-label={t.send}
+          onClick={onSend}
+          className="send"
+          title={t.send}
+        />
+      )}
     </SelectionBarWrapper>
   )
 }

package/src/views/MessageInput/ContextBar.tsx

@@ -1,7 +1,5 @@
-import { Text } from '@citric/core'
-import { TimesMini } from '@citric/icons'
-import { Badge, IconButton } from '@citric/ui'
 import { loader } from '@monaco-editor/react'
+import { Badge, IconButton, Text } from '@stack-spot/citric-react'
 import { ColorPaletteName, listToClass } from '@stack-spot/portal-theme'
 import { useCallback, useMemo } from 'react'
 import { FadingOverflow } from '../../components/FadingOverflow'
@@ -16,14 +14,16 @@ interface ContextBadgeProps {
 }
 
 const ContextBadge = ({ label, color, dismiss, onDismiss }: ContextBadgeProps) => (
-  <Badge appearance="square" palette={color} className="context-badge" 
-    afterElement={
-      onDismiss && 
-      <IconButton appearance="square" colorIcon={`${color}.800`} onClick={onDismiss} title={dismiss} aria-label={dismiss}>
-        <TimesMini />
-      </IconButton>
-    }>
+  <Badge appearance="square" colorPalette={color} className="context-badge">
     <Text nowrapEllipsis>{label}</Text>
+    {onDismiss && <IconButton
+      icon="TimesMini"
+      appearance="none"
+      onClick={onDismiss}
+      title={dismiss}
+      arial-label={dismiss}
+      style={{ padding: '2px 0 2px 2px', color: 'inherit' }}
+    />}
   </Badge>
 )
 
@@ -74,10 +74,12 @@ export const ContextBar = () => {
       <div className="content">
         {shouldRenderRemoveAllButton && (
           <IconButton 
-            appearance="square" color="light" 
-            aria-label={t.removeConfig} title={t.removeConfig} onClick={removeAll}>
-            <TimesMini />
-          </IconButton>
+            icon="TimesMini"
+            aria-label={t.removeConfig}
+            title={t.removeConfig}
+            onClick={removeAll}
+            style={{ marginLeft: '8px' }}
+          />
         )}
         <FadingOverflow className="list-overflow" scroll="arrows" enableHorizontalScrollWithVerticalWheel>
           <ul>

package/src/views/MessageInput/QuickCommandSelector.tsx

@@ -1,4 +1,4 @@
-import { QuickCommand } from '@citric/icons'
+import { Icon } from '@stack-spot/citric-icons'
 import { aiClient, workspaceAiClient } from '@stack-spot/portal-network'
 import { QuickCommandResponse } from '@stack-spot/portal-network/api/ai'
 import { useCallback } from 'react'
@@ -113,7 +113,7 @@ export const QuickCommandSelector = ({ inputRef, isTrial }:
     selectorConfig={{
       resourceName: 'Quick Command',
       shortcut: '/',
-      icon: <QuickCommand />,
+      icon: <Icon icon="QuickCommand" />,
       searchProp: 'slug',
       urlBuilder: (qc) => `/quick-command/${qc?.slug}`,
       regex: quickCommandRegex,

package/src/views/MessageInput/SelectContent.tsx

@@ -1,29 +1,25 @@
-import { Agent, DocumentUpload, KnowledgeSource, Plus, QuickCommand, Spaces, Stack } from '@citric/icons'
-import { IconButton } from '@citric/ui'
-import { SelectionList } from '@stack-spot/portal-components/SelectionList'
-import { useMemo, useState } from 'react'
+import { Icon } from '@stack-spot/citric-icons'
+import { IconButton, ImageButton, MenuAction, MenuOverlay, MenuSection } from '@stack-spot/citric-react'
+import { useMemo } from 'react'
 import { useCurrentChat, useCurrentChatState, useWidget } from '../../context/hooks'
 import { useUploadManager } from '../../utils/upload/context'
 import { useMessageInputDictionary } from './dictionary'
 
-type ChatFeatures = 'workspace' | 'knowledgeSource' | 'stack'
-type ChatPanel = 'ks' | 'workspace' | 'stack'
-
 export const SelectContent = () => {
   const widget = useWidget()
-  const [visibleMenu, setVisibleMenu] = useState(false)
   const features = useCurrentChatState('features')
   const t = useMessageInputDictionary()
   const uploadManager = useUploadManager()
   const hasFeatureButtons = features.workspace || features.knowledgeSource || features.stack || features.upload
   const chat = useCurrentChat()
 
-  const itemConfigs = [
+  const itemConfigs = useMemo<(MenuAction & { key: keyof typeof features, section: 0 | 1 })[]>(() => ([
     {
       role: 'button',
       key: 'agent',
+      section: 0,
       label: t.chatAgent,
-      icon: <Agent />,
+      icon: <Icon icon="Agent" />,
       onClick: (event?: React.MouseEvent) => {
         event?.stopPropagation()
         chat.set('nextMessage', '@')
@@ -32,8 +28,9 @@ export const SelectContent = () => {
     {
       role: 'button',
       key: 'quickCommands',
+      section: 0,
       label: 'Quick Commands',
-      icon: <QuickCommand />,
+      icon: <Icon icon="QuickCommand" />,
       onClick: (event?: React.MouseEvent) => {
         event?.stopPropagation()
         chat.set('nextMessage', '/')
@@ -42,81 +39,61 @@ export const SelectContent = () => {
     {
       role: 'button',
       key: 'knowledgeSource',
+      section: 0,
       label: 'Knowledge Sources',
-      icon: <KnowledgeSource />,
-      panel: 'ks',
+      icon: <Icon icon="KnowledgeSource" />,
+      onClick: () => widget.set('panel', 'ks'),
     },
     {
       role: 'button',
       key: 'stack',
+      section: 0,
       label: 'Stacks AI',
-      icon: <Stack />,
-      panel: 'stack',
+      icon: <Icon icon="Stack" />,
+      onClick: () => widget.set('panel', 'stack'),
     },
     {
       role: 'button',
       key: 'workspace',
+      section: 0,
       label: 'Spots',
-      icon: <Spaces />,
-      panel: 'workspace',
+      icon: <Icon icon="Spaces" />,
+      onClick: () => widget.set('panel', 'workspace'),
     },
     {
       role: 'button',
       key: 'upload',
+      section: 1,
       label: t.upload,
       onClick: () => uploadManager.open(),
       className: 'upload-item',
-      icon: <DocumentUpload />,
+      icon: <Icon icon="DocumentUpload" />,
     },
-  ]
+  ] as const).filter(({ key }) => features[key]), [features, widget])
 
-  const listItems = useMemo(() =>
-    itemConfigs.filter(chatFeatures => features[chatFeatures.key as ChatFeatures])
-      .map(chatFeatures => ({
-        label: chatFeatures.label,
-        icon: chatFeatures.icon,
-        className: chatFeatures.className,
-        onClick: chatFeatures.panel ? (() => {
-          widget.set('panel', chatFeatures.panel as ChatPanel)
-          setVisibleMenu(false)
-        }) : chatFeatures.onClick,
-        role: chatFeatures.role,
-      })), [features, widget])
+  const listItems = useMemo(() => {
+    const sections: [MenuSection, MenuSection] = [
+      { type: 'section', children: [] },
+      { type: 'section', children: [] },
+    ]
+    itemConfigs.forEach(item => sections[item.section].children.push(item))
+    return sections.filter(s => !!s.children.length)
+  }, [itemConfigs])
 
   if (!hasFeatureButtons) return null
 
-  return listItems.length > 1 ? (
-    <>
-      <IconButton
-        id="button-content-select"
-        color="light"
-        appearance="square"
-        title={t.chatViewMenu}
-        data-test-hint="button-options"
-        aria-label={t.chatViewMenu}
-        aria-controls="chatMessageMenu"
-        aria-expanded={visibleMenu} 
-        onClick={() => setVisibleMenu(state => !state)}>
-        <Plus />
-      </IconButton>
-      <SelectionList
-        className="message-menu"
-        id="chatMessageMenu"
-        visible={visibleMenu}
-        onHide={() => setVisibleMenu(false)}
-        items={listItems}
-        aria-label={t.chatViewMenu}
-      />
-    </>
+  return itemConfigs.length > 1 ? (
+    <MenuOverlay items={listItems} position="top" bgLevel={500} spaced roundedItems>
+      <IconButton icon="Plus" />
+    </MenuOverlay>
   ) : (
-    <IconButton
-      color="light"
+    <ImageButton
       appearance="square"
-      role="button"
-      title={listItems[0].label}
-      aria-label={listItems[0].label}
-      onClick={listItems[0].onClick}>
-      {listItems[0].icon}
-    </IconButton>
+      title={`${itemConfigs[0].label}`}
+      aria-label={`${itemConfigs[0].label}`}
+      onClick={itemConfigs[0].onClick}
+    >
+      {itemConfigs[0].icon}
+    </ImageButton>
   )
 }

package/src/views/MessageInput/UploadDragNDrop.tsx

@@ -1,5 +1,4 @@
-import { IconBox } from '@citric/core'
-import { Document } from '@citric/icons'
+import { Icon } from '@stack-spot/citric-icons'
 import { DragEvent, useCallback, useEffect, useState } from 'react'
 import { useCurrentChat } from '../../context/hooks'
 import { useMessageInputDictionary } from './dictionary'
@@ -72,9 +71,7 @@ export const UploadDragNDrop = ({ isDragging, onDrop, onDragLeave }: UploadDragN
         e.stopPropagation()
       }}
     >
-      <IconBox size="lg" aria-hidden={true}>
-        <Document />
-      </IconBox>
+      <Icon icon="Document" size="lg" aria-hidden={true} />
       <h2 id="upload-drag-drop-title">{t.uploadDragDropTitle}</h2>
       <p id="upload-drag-drop-description">{t.uploadDragDropDescription}</p>
     </div>

package/src/views/MessageInput/index.tsx

@@ -1,8 +1,7 @@
-import { listToClass, theme } from '@stack-spot/portal-theme'
+import { ProgressBar } from '@stack-spot/citric-react'
 import { interpolate } from '@stack-spot/portal-translate'
 import { useCallback, useEffect, useRef, useState } from 'react'
 import { AdaptiveTextArea } from '../../components/AdaptiveTextArea'
-import { ProgressBar } from '../../components/ProgressBar'
 import { useCurrentChat, useCurrentChatState, useWidgetState } from '../../context/hooks'
 import { quickCommandRegex } from '../../regex'
 import { ChatEntry } from '../../state/ChatEntry'
@@ -25,7 +24,8 @@ import { UploadDragNDrop, useUploadDragDrop } from './UploadDragNDrop'
  * going to be used for the question and the buttons to send, cancel, set the workspace, among others. This also includes the Quick
  * Commands panel for auto completing.
  */
-export const MessageInput = ({ chatWindowRef }: { chatWindowRef?: React.RefObject<HTMLElement> }) => {
+export const MessageInput = ({ chatWindowRef, customInputMessage }: 
+  { chatWindowRef?: React.RefObject<HTMLElement>, customInputMessage?: string }) => {
   const t = useMessageInputDictionary()
   const [focused, setFocused] = useState(false)
   const [, setExpanded] = useState(true)
@@ -137,11 +137,11 @@ export const MessageInput = ({ chatWindowRef }: { chatWindowRef?: React.RefObjec
 
   return (
     <UploadProvider value={chat.uploadManager}>
-      <MessageInputBox aria-busy={isLoading} className="message-input" $inputFocused={focused}>
+      <MessageInputBox aria-busy={isLoading} className={`message-input ${focused ? 'focused' : ''}`}>
         <div className="wrapper-action">
           <QuickCommandSelector inputRef={textAreaRef} isTrial={isTrial} />
           <AgentSelector inputRef={textAreaRef} isTrial={isTrial} />
-          <div className={listToClass(['action-box', focused && 'focused', isLoading && 'disabled'])}>
+          <div className="action-box">
             <ButtonAgent />
             <AdaptiveTextArea
               maxHeight={() => isMinimized
@@ -151,10 +151,10 @@ export const MessageInput = ({ chatWindowRef }: { chatWindowRef?: React.RefObjec
               ref={textAreaRef}
               placeholder={
                 agentLabel
-                  ? interpolate(t.placeholder, agentLabel)
+                  ? customInputMessage ?? interpolate(t.placeholder, agentLabel)
                   : t.typing
               }
-              onChange={e => chat.set('nextMessage', e.target.value)}
+              onChange={v => chat.set('nextMessage', v)}
               value={value}
               onFocus={() => setFocused(true)}
               onBlur={() => setFocused(false)}
@@ -165,8 +165,7 @@ export const MessageInput = ({ chatWindowRef }: { chatWindowRef?: React.RefObjec
             />
           </div>
         </div>
-        <ProgressBar visible={true} animate={isLoading} 
-          backgroundColor={isLoading || !focused ? theme.color.light[500] : theme.color.primary[500]} />
+        <ProgressBar className="progress-bar" progress={isLoading ? undefined : (focused ? 100 : 0)} speed="fast" />
         <ContextBar />
         {chat.get('features').upload && <UploadBar />}
         <ButtonBar onSend={onSend} isLoading={isLoading} />