feat(frontend): disabled searchBar if no remaining credits or no brai…

…n selected (#2788) # Description Please include a summary of the changes and the related issue. Please also include relevant motivation and context. ## Checklist before requesting a review Please delete options that are not relevant. - [ ] My code follows the style guidelines of this project - [ ] I have performed a self-review of my code - [ ] I have commented hard-to-understand areas - [ ] I have ideally added tests that prove my fix is effective or that my feature works - [ ] New and existing unit tests pass locally with my changes - [ ] Any dependent changes have been merged ## Screenshots (if appropriate):
QuivrHQ · Jul 1, 2024 · bfdc5c8 · bfdc5c8
1 parent 2e4b801
commit bfdc5c8
Show file tree

Hide file tree

Showing 9 changed files with 92 additions and 58 deletions.
diff --git a/...ctionsBar/components/ChatInput/components/ChatEditor/Editor/hooks/useCreateEditorState.ts b/...ctionsBar/components/ChatInput/components/ChatEditor/Editor/hooks/useCreateEditorState.ts
@@ -6,12 +6,15 @@ import { Text } from "@tiptap/extension-text";
 import { Extension, useEditor } from "@tiptap/react";
 import { useTranslation } from "react-i18next";
 
+import { useUserSettingsContext } from "@/lib/context/UserSettingsProvider/hooks/useUserSettingsContext";
+
 import { useBrainMention } from "./useBrainMention";
 
 // eslint-disable-next-line @typescript-eslint/explicit-module-boundary-types
 export const useCreateEditorState = (placeholder?: string) => {
   const { t } = useTranslation(["chat"]);
   const { BrainMention, items } = useBrainMention();
+  const { remainingCredits } = useUserSettingsContext();
 
   const PreventEnter = Extension.create({
     addKeyboardShortcuts: () => {
@@ -24,7 +27,7 @@ export const useCreateEditorState = (placeholder?: string) => {
 
   const editor = useEditor(
     {
-      autofocus: true,
+      autofocus: !!remainingCredits,
       onFocus: () => {
         editor?.commands.focus("end");
       },

diff --git a/frontend/app/chat/[chatId]/components/ActionsBar/components/ChatInput/index.module.scss b/frontend/app/chat/[chatId]/components/ActionsBar/components/ChatInput/index.module.scss
@@ -13,9 +13,11 @@
   .chat_wrapper {
     display: flex;
     padding: Spacings.$spacing05;
+    padding-top: 0;
 
-    &.with_brain {
-      padding-top: 0;
+    &.disabled {
+      pointer-events: none;
+      opacity: 0.4;
     }
   }
-}
+}
diff --git a/frontend/app/chat/[chatId]/components/ActionsBar/components/ChatInput/index.tsx b/frontend/app/chat/[chatId]/components/ActionsBar/components/ChatInput/index.tsx
@@ -4,6 +4,7 @@ import { CurrentBrain } from "@/lib/components/CurrentBrain/CurrentBrain";
 import Icon from "@/lib/components/ui/Icon/Icon";
 import { LoaderIcon } from "@/lib/components/ui/LoaderIcon/LoaderIcon";
 import { useBrainContext } from "@/lib/context/BrainProvider/hooks/useBrainContext";
+import { useUserSettingsContext } from "@/lib/context/UserSettingsProvider/hooks/useUserSettingsContext";
 
 import { ChatEditor } from "./components/ChatEditor/ChatEditor";
 import { useChatInput } from "./hooks/useChatInput";
@@ -12,10 +13,11 @@ import styles from "./index.module.scss";
 export const ChatInput = (): JSX.Element => {
   const { setMessage, submitQuestion, generatingAnswer, message } =
     useChatInput();
+  const { remainingCredits } = useUserSettingsContext();
   const { currentBrain } = useBrainContext();
 
   const handleSubmitQuestion = () => {
-    if (message.trim() !== "") {
+    if (message.trim() !== "" && remainingCredits && currentBrain) {
       submitQuestion();
     }
   };
@@ -30,12 +32,14 @@ export const ChatInput = (): JSX.Element => {
         }}
       >
         <div className={styles.chat_container}>
-          <CurrentBrain allowingRemoveBrain={false} />
+          <CurrentBrain
+            allowingRemoveBrain={false}
+            remainingCredits={remainingCredits}
+          />
           <div
-            className={`
-            ${styles.chat_wrapper}
-            ${currentBrain ? styles.with_brain : ""}
-          `}
+            className={`${styles.chat_wrapper} ${
+              !remainingCredits ? styles.disabled : ""
+            }`}
           >
             <ChatEditor
               message={message}
@@ -49,7 +53,7 @@ export const ChatInput = (): JSX.Element => {
                 name="followUp"
                 size="large"
                 color="accent"
-                disabled={!message}
+                disabled={!message || !remainingCredits || !currentBrain}
                 handleHover={true}
                 onClick={handleSubmitQuestion}
               />

diff --git a/frontend/app/search/page.module.scss b/frontend/app/search/page.module.scss
@@ -53,23 +53,6 @@
         }
       }
     }
-
-    .shortcuts_card_wrapper {
-      background-color: var(--background-2);
-      padding: Spacings.$spacing05;
-      gap: Spacings.$spacing03;
-      border-radius: Radius.$big;
-
-      .shortcut_wrapper {
-        display: flex;
-        align-items: center;
-        gap: Spacings.$spacing02;
-
-        .shortcut {
-          color: var(--primary-0);
-        }
-      }
-    }
   }
 }
 
@@ -105,4 +88,4 @@
     display: flex;
     gap: Spacings.$spacing05;
   }
-}
+}
diff --git a/frontend/app/search/page.tsx b/frontend/app/search/page.tsx
@@ -72,13 +72,6 @@ const Search = (): JSX.Element => {
               <SearchBar />
             </div>
           </div>
-          <div className={styles.shortcuts_card_wrapper}>
-            <div className={styles.shortcut_wrapper}>
-              <span>Press</span>
-              <span className={styles.shortcut}>@</span>
-              <span>to select a brain</span>
-            </div>
-          </div>
         </div>
         <UploadDocumentModal />
         <AddBrainModal />

diff --git a/frontend/lib/components/CurrentBrain/CurrentBrain.module.scss b/frontend/lib/components/CurrentBrain/CurrentBrain.module.scss
@@ -1,12 +1,20 @@
 @use "styles/Spacings.module.scss";
 @use "styles/Typography.module.scss";
 
-.current_brain_wrapper {
+%header_style {
   background-color: var(--background-2);
   padding-inline: Spacings.$spacing05;
   padding-block: Spacings.$spacing01;
   font-size: Typography.$small;
-  color: var(--text-1);
+}
+
+@mixin textColor($color) {
+  color: var(--#{$color});
+}
+
+.current_brain_wrapper {
+  @extend %header_style;
+  @include textColor(text-1);
 
   .brain_infos {
     display: flex;
@@ -20,7 +28,8 @@
       align-items: center;
       @include Typography.EllipsisOverflow;
 
-      .title {
+      .title,
+      .brain_name {
         white-space: nowrap;
       }
 
@@ -35,10 +44,27 @@
         }
 
         .brain_name {
-          color: var(--text-3);
+          @include textColor(text-3);
           @include Typography.EllipsisOverflow;
         }
       }
     }
   }
-}
+}
+
+.no_brain_selected,
+.no_credits_left {
+  @extend %header_style;
+}
+
+.no_brain_selected {
+  @include textColor(warning);
+
+  .strong {
+    font-weight: 800;
+  }
+}
+
+.no_credits_left {
+  @include textColor(dangerous);
+}
diff --git a/frontend/lib/components/CurrentBrain/CurrentBrain.tsx b/frontend/lib/components/CurrentBrain/CurrentBrain.tsx
@@ -9,19 +9,37 @@ import { Icon } from "../ui/Icon/Icon";
 
 interface CurrentBrainProps {
   allowingRemoveBrain: boolean;
+  remainingCredits: number | null;
 }
 
 export const CurrentBrain = ({
   allowingRemoveBrain,
+  remainingCredits,
 }: CurrentBrainProps): JSX.Element => {
   const { currentBrain, setCurrentBrainId } = useBrainContext();
   const { isDarkMode } = useUserSettingsContext();
   const removeCurrentBrain = (): void => {
     setCurrentBrainId(null);
   };
 
+  if (!remainingCredits) {
+    return (
+      <div className={styles.no_credits_left}>
+        <span>
+          You’ve run out of credits! Upgrade your plan now to continue chatting.
+        </span>
+      </div>
+    );
+  }
+
   if (!currentBrain) {
-    return <></>;
+    return (
+      <div className={styles.no_brain_selected}>
+        <span>
+          Press <strong className={styles.strong}>@</strong> to select a Brain
+        </span>
+      </div>
+    );
   }
 
   return (

diff --git a/frontend/lib/components/ui/SearchBar/SearchBar.module.scss b/frontend/lib/components/ui/SearchBar/SearchBar.module.scss
@@ -18,9 +18,11 @@
     align-items: center;
     justify-content: space-between;
     padding: Spacings.$spacing05;
+    padding-top: 0;
 
-    &.with_brain {
-      padding-top: 0;
+    &.disabled {
+      pointer-events: none;
+      opacity: 0.3;
     }
 
     .search_icon {
@@ -36,4 +38,4 @@
       }
     }
   }
-}
+}
diff --git a/frontend/lib/components/ui/SearchBar/SearchBar.tsx b/frontend/lib/components/ui/SearchBar/SearchBar.tsx
@@ -6,6 +6,7 @@ import { useChatInput } from "@/app/chat/[chatId]/components/ActionsBar/componen
 import { useChat } from "@/app/chat/[chatId]/hooks/useChat";
 import { useChatContext } from "@/lib/context";
 import { useBrainContext } from "@/lib/context/BrainProvider/hooks/useBrainContext";
+import { useUserSettingsContext } from "@/lib/context/UserSettingsProvider/hooks/useUserSettingsContext";
 
 import styles from "./SearchBar.module.scss";
 
@@ -23,6 +24,7 @@ export const SearchBar = ({
   const { setMessages } = useChatContext();
   const { addQuestion } = useChat();
   const { currentBrain, setCurrentBrainId } = useBrainContext();
+  const { remainingCredits } = useUserSettingsContext();
 
   useEffect(() => {
     setCurrentBrainId(null);
@@ -33,7 +35,7 @@ export const SearchBar = ({
   }, [message]);
 
   const submit = async (): Promise<void> => {
-    if (!searching) {
+    if (!!remainingCredits && !!currentBrain && !searching) {
       setSearching(true);
       setMessages([]);
       try {
@@ -50,18 +52,15 @@ export const SearchBar = ({
   };
 
   return (
-    <div
-      className={`
-      ${styles.search_bar_wrapper}
-      ${currentBrain ? styles.with_brain : ""}
-      `}
-    >
-      <CurrentBrain allowingRemoveBrain={true} />
+    <div className={styles.search_bar_wrapper}>
+      <CurrentBrain
+        allowingRemoveBrain={true}
+        remainingCredits={remainingCredits}
+      />
       <div
-        className={`
-      ${styles.editor_wrapper}
-      ${currentBrain ? styles.with_brain : ""}
-      `}
+        className={`${styles.editor_wrapper} ${
+          !remainingCredits ? styles.disabled : ""
+        }`}
       >
         <Editor
           message={message}
@@ -75,7 +74,11 @@ export const SearchBar = ({
           <LuSearch
             className={`
           ${styles.search_icon} 
-          ${isDisabled ? styles.disabled : ""}
+          ${
+            isDisabled || !remainingCredits || !currentBrain
+              ? styles.disabled
+              : ""
+          }
           `}
             onClick={() => void submit()}
           />