Updated download button

starscape.py

@@ -226,25 +226,39 @@ playBtn.addEventListener('click', () => {
   playBtn.textContent = '✦ Speaking…';
   trigger.click();
 
-  // Reset the button when the audio finishes (poll briefly for the <audio> tag).
-  let tries = 0;
-  const watch = setInterval(() => {
+  // Reset the button when the freshly-loaded audio finishes. Gradio replaces
+  // the <audio> element when it sets a new src, so we wait for that src to
+  // appear, then listen on THAT element.
+  let lastSrc = '';
+  let finished = false;
+  let poll = null;
+  const startTime = Date.now();
+
+  function finish() {
+    if (finished) return;
+    finished = true;
+    if (poll) clearInterval(poll);
+    playBtn.classList.remove('playing');
+    playBtn.textContent = '▶ Play wisdom';
+  }
+
+  poll = setInterval(() => {
+    if (Date.now() - startTime > 90000) return finish();
     const audio = findGradioAudio('tts-audio');
-    tries++;
-    if (audio) {
-      const reset = () => {
-        playBtn.classList.remove('playing');
-        playBtn.textContent = '▶ Play wisdom';
-        audio.removeEventListener('ended', reset);
-      };
-      audio.addEventListener('ended', reset);
-      clearInterval(watch);
-    } else if (tries > 80) {
-      clearInterval(watch);
-      playBtn.classList.remove('playing');
-      playBtn.textContent = '▶ Play wisdom';
-    }
-  }, 250);
+    if (!audio || !audio.src) return;
+    if (audio.src === lastSrc) return;
+
+    lastSrc = audio.src;
+    audio.addEventListener('ended', finish, { once: true });
+
+    const scheduleByDuration = () => {
+      if (audio.duration > 0 && !Number.isNaN(audio.duration)) {
+        setTimeout(finish, (audio.duration + 1.5) * 1000);
+      }
+    };
+    if (audio.readyState >= 1) scheduleByDuration();
+    else audio.addEventListener('loadedmetadata', scheduleByDuration, { once: true });
+  }, 300);
 });
 
 const THREE = await import('https://esm.sh/three@0.160.0');