Disable submit edit button when user input should be disabled

brave · Jul 16, 2024 · 0670c33 · 0670c33
1 parent 0b64a41
commit 0670c33
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/components/ai_chat/resources/page/components/conversation_list/index.tsx b/components/ai_chat/resources/page/components/conversation_list/index.tsx
@@ -203,6 +203,7 @@ function ConversationList(props: ConversationListProps) {
                       text={latestTurnText}
                       onSubmit={(text) => handleEditSubmit(id, text)}
                       onCancel={() => setEditInputId(null)}
+                      isSubmitDisabled={shouldDisableUserInput}
                     />
                   )}
                   {turn.selectedText && (

diff --git a/components/ai_chat/resources/page/components/edit_input/index.tsx b/components/ai_chat/resources/page/components/edit_input/index.tsx
@@ -14,6 +14,7 @@ interface Props {
   text: string
   onSubmit: (text: string) => void
   onCancel: () => void
+  isSubmitDisabled: boolean
 }
 
 function EditInput(props: Props) {
@@ -54,6 +55,7 @@ function EditInput(props: Props) {
         <Button
           size='small'
           kind='filled'
+          isDisabled={props.isSubmitDisabled}
           onClick={() => props.onSubmit(text)}
         >
         {getLocale('saveButtonLabel')}

diff --git a/components/ai_chat/resources/page/state/data-context-provider.tsx b/components/ai_chat/resources/page/state/data-context-provider.tsx
@@ -71,7 +71,7 @@ function DataContextProvider(props: DataContextProviderProps) {
   const [allModels, setAllModels] = React.useState<mojom.Model[]>([])
   const [conversationHistory, setConversationHistory] = React.useState<mojom.ConversationTurn[]>([])
   const [suggestedQuestions, setSuggestedQuestions] = React.useState<string[]>([])
-  const [isGenerating, setIsGenerating] = React.useState(false)
+  const [isGenerating, setIsGenerating] = React.useState(props.store?.isGenerating || false)
   const [suggestionStatus, setSuggestionStatus] = React.useState<mojom.SuggestionGenerationStatus>(mojom.SuggestionGenerationStatus.None)
   const [siteInfo, setSiteInfo] = React.useState<mojom.SiteInfo>({
     title: undefined,