Update app.py

app.py

@@ -955,13 +955,15 @@ async def _run_agent_turn(prompt_text: str, cxr_path: str | None) -> None:
                 answer.content = (
                     "I've re-opened the most recent overlay in the panel on the "
                     "right — it already shows the localized findings on this image.")
-                answer.elements = []  # stop the live timer
+                answer.elements = [cl.CustomElement(
+                    name="TimerStop", props={"id": _timer_id}, display="inline")]
                 await answer.update()
                 return
             except Exception:  # noqa: BLE001 — fall through to the error message
                 pass
         answer.content = f"❌ Run failed: {e}"
-        answer.elements = []  # stop the live timer
+        answer.elements = [cl.CustomElement(
+            name="TimerStop", props={"id": _timer_id}, display="inline")]
         await answer.update()
         return
 
@@ -1036,6 +1038,10 @@ async def _run_agent_turn(prompt_text: str, cxr_path: str | None) -> None:
         content += "\n\n---\n" + "\n\n".join(footer_bits)
 
     answer.content = content
+    # Mount an invisible stopper so the live timer halts even if this Chainlit
+    # version doesn't unmount the removed TurnTimer element on update.
+    files.append(cl.CustomElement(
+        name="TimerStop", props={"id": _timer_id}, display="inline"))
     answer.elements = files          # replaces the live timer element (turn done)
     answer.actions = actions
     await answer.update()