1 files changed, 11 insertions, 2 deletions
diff --git a/src/render/stream.rs b/src/render/stream.rs
index a6233a5..d6cadde 100644
--- a/src/render/stream.rs
+++ b/src/render/stream.rs
@@ -36,11 +36,17 @@ pub async fn raw_stream(
     mut rx: UnboundedReceiver<SseEvent>,
     abort_signal: &AbortSignal,
 ) -> Result<()> {
+    let mut spinner = Some(spawn_spinner("Generating"));
+
     loop {
         if abort_signal.aborted() {
-            return Ok(());
+            break;
         }
         if let Some(evt) = rx.recv().await {
+            if let Some(spinner) = spinner.take() {
+                spinner.stop();
+            }
+
             match evt {
                 SseEvent::Text(text) => {
                     print!("{text}");
@@ -52,6 +58,9 @@ pub async fn raw_stream(
             }
         }
     }
+    if let Some(spinner) = spinner.take() {
+        spinner.stop();
+    }
     Ok(())
 }
 
@@ -70,7 +79,7 @@ async fn markdown_stream_inner(
 
     'outer: loop {
         if abort_signal.aborted() {
-            return Ok(());
+            break;
         }
         for reply_event in gather_events(&mut rx).await {
             if let Some(spinner) = spinner.take() {