Handle Gradio completion status

index.html

@@ -590,7 +590,7 @@
 
         async function init() {
             try {
-                client = await Client.connect(window.location.origin);
+                client = await Client.connect(window.location.origin, { events: ["data", "status"] });
             } catch (err) { console.error("Gradio Connection Error", err); }
         }
         init();
@@ -823,6 +823,13 @@
                     if (msg.type === "data" && msg.data) {
                         const chunk = msg.data[0];
                         finalAnswer = updateBotMessage(botDiv, chunk);
+                    } else if (msg.type === "status") {
+                        if (msg.stage === "complete") {
+                            break;
+                        }
+                        if (msg.stage === "error") {
+                            throw new Error(msg.message || "Generation failed");
+                        }
                     }
                 }