diff --git a/ui/src/components/chat/ChatInput.tsx b/ui/src/components/chat/ChatInput.tsx
index 5607e07..31ea32c 100644
--- a/ui/src/components/chat/ChatInput.tsx
+++ b/ui/src/components/chat/ChatInput.tsx
@@ -2473,6 +2473,11 @@ export const ChatInput: React.FC<ChatInputProps> = ({ onOpenSettings, scrollToBo
                     wasAborted={working.wasAborted}
                     abortActive={working.abortActive}
                     retryInfo={working.retryInfo}
+                    loopInfo={loopState ? { current: loopState.current, total: loopState.total, prompt: loopState.prompt } : null}
+                    onStopLoop={() => {
+                        setLoopState(null);
+                        abortCurrentOperation();
+                    }}
                     showAbortStatus={showAbortStatus}
                 />
             </div>