diff --git a/client/src/components/Input/SubmitButton.jsx b/client/src/components/Input/SubmitButton.jsx
index b3309027f3..de31c47809 100644
--- a/client/src/components/Input/SubmitButton.jsx
+++ b/client/src/components/Input/SubmitButton.jsx
@@ -1,5 +1,5 @@
import React, { useState } from 'react';
-import { StopGeneratingIcon } from '~/components';
+import { ListeningIcon, StopGeneratingIcon } from '~/components';
import { Settings } from 'lucide-react';
import { SetTokenDialog } from './SetTokenDialog';
import store from '~/store';
@@ -10,6 +10,7 @@ export default function SubmitButton({
handleStopGenerating,
disabled,
isSubmitting,
+ isListening,
endpointsConfig,
}) {
const [setTokenDialogOpen, setSetTokenDialogOpen] = useState(false);
@@ -61,6 +62,14 @@ export default function SubmitButton({
/>
>
);
+ } else if (isListening) {
+ return (
+
+ );
} else {
return (