generate button only on User sticky + not empty

jackschedel · Jan 2, 2024 · 127d130 · 127d130
1 parent b602914
commit 127d130
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/components/Chat/ChatContent/Message/MessageContent.tsx b/src/components/Chat/ChatContent/Message/MessageContent.tsx
@@ -26,6 +26,7 @@ const MessageContent = ({
           setIsEdit={setIsEdit}
           messageIndex={messageIndex}
           sticky={sticky}
+          role={role}
         />
       ) : (
         <ContentView

diff --git a/src/components/Chat/ChatContent/Message/View/EditView.tsx b/src/components/Chat/ChatContent/Message/View/EditView.tsx
@@ -17,11 +17,13 @@ const EditView = ({
   setIsEdit,
   messageIndex,
   sticky,
+  role,
 }: {
   content: string;
   setIsEdit: React.Dispatch<React.SetStateAction<boolean>>;
   messageIndex: number;
   sticky?: boolean;
+  role: string;
 }) => {
   const inputRole = useStore((state) => state.inputRole);
   const setChats = useStore((state) => state.setChats);
@@ -145,6 +147,8 @@ const EditView = ({
         cursorPosition={cursorPosition}
         _setContent={_setContent}
         messageIndex={messageIndex}
+        role={role}
+        content={content}
       />
       {isModalOpen && (
         <PopupModal
@@ -168,6 +172,8 @@ const EditViewButtons = memo(
     cursorPosition,
     _setContent,
     messageIndex,
+    role,
+    content,
   }: {
     sticky?: boolean;
     handleGenerate: () => void;
@@ -177,6 +183,8 @@ const EditViewButtons = memo(
     cursorPosition: number;
     _setContent: React.Dispatch<React.SetStateAction<string>>;
     messageIndex: number;
+    role: string;
+    content: string;
   }) => {
     const { t } = useTranslation();
     const generating = useStore.getState().generating;
@@ -214,7 +222,7 @@ const EditViewButtons = memo(
             </button>
           )}
 
-          {sticky || (
+          {!sticky && role === 'user' && content != '' && (
             <button
               className='btn relative mr-2 btn-primary'
               onClick={() => {