From e29079ddc9932ad7d76a16e6b003e28cd1b41836 Mon Sep 17 00:00:00 2001
From: mic <m@micz.it>
Date: Wed, 25 Mar 2026 00:06:37 +0100
Subject: [PATCH] correctly loading the working icon. see #707

---
 mzta-background.js | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mzta-background.js b/mzta-background.js
index 57487ef2..02f9bbdf 100644
--- a/mzta-background.js
+++ b/mzta-background.js
@@ -479,12 +479,14 @@ async function _generateSummaryForMessage(headerMessageId, tabId) {
         }
 
         await summaryStore.setProcessing(headerMessageId);
+        taWorkingStatus.startWorking();
         browser.tabs.sendMessage(tabId, { command: "showSummaryGenerating" });
 
         const messageResult = await browser.messages.query({ headerMessageId: headerMessageId });
         if (!messageResult || messageResult.messages.length === 0) {
             await summaryStore.saveError(headerMessageId, "Message not found");
             browser.tabs.sendMessage(tabId, { command: "showSummary", data: { error: true, message: "Message not found" } });
+            taWorkingStatus.stopWorking();
             return;
         }
 
@@ -496,6 +498,7 @@ async function _generateSummaryForMessage(headerMessageId, tabId) {
             const errorMsg = browser.i18n.getMessage('summarize_chatgpt_web_not_supported');
             await summaryStore.saveError(headerMessageId, errorMsg);
             browser.tabs.sendMessage(tabId, { command: "showSummary", data: { error: true, message: errorMsg } });
+            taWorkingStatus.stopWorking();
             return;
         }
 
@@ -522,11 +525,13 @@ async function _generateSummaryForMessage(headerMessageId, tabId) {
         };
         await summaryStore.saveSummary(summaryData, headerMessageId);
         browser.tabs.sendMessage(tabId, { command: "showSummary", data: summaryData });
+        taWorkingStatus.stopWorking();
 
     } catch (error) {
         console.error("[ThunderAI] Error generating summary:", error);
         await summaryStore.saveError(headerMessageId, error.message || String(error));
         browser.tabs.sendMessage(tabId, { command: "showSummary", data: { error: true, message: error.message || "Failed to generate summary" } });
+        taWorkingStatus.stopWorking();
     }
 }