UI: add Broadcast — run a command on the given session across all online clients

app.py

@@ -274,6 +274,31 @@ def ui_send(client_label: str, session: str, command: str, timeout: int):
     return f"➡️ sent to `{cid}` [{session}]", ""
 
 
+def ui_broadcast(session: str, command: str, timeout: int):
+    """Queue the same command on the given session for every online container."""
+    if not command.strip():
+        return "⚠️ empty command", command
+    session = (session or "main").strip()
+    now = _now()
+    sent = []
+    with LOCK:
+        online = [cid for cid, c in CLIENTS.items()
+                  if now - c.get("last_seen", 0) < CLIENT_STALE_AFTER]
+        for cid in online:
+            cmd_id = uuid.uuid4().hex
+            COMMANDS[cmd_id] = {
+                "id": cmd_id, "client_id": cid, "session": session,
+                "command": command, "timeout": int(timeout),
+                "status": "pending", "created": now,
+            }
+            _append_and_trim(cid, cmd_id)
+            sent.append(cid)
+    if not sent:
+        return "⚠️ no online containers to broadcast to", command
+    return (f"📡 broadcast to {len(sent)} container(s) [{session}]: "
+            + ", ".join(f"`{c}`" for c in sent), "")
+
+
 def ui_put_file(client_label: str, fileobj, dest: str):
     """Queue a push: send an uploaded file to the selected container."""
     cid = _strip_dot(client_label)
@@ -416,6 +441,7 @@ with gr.Blocks(title="NII Relay", analytics_enabled=False) as demo:
                             placeholder="nvidia-smi    |    cd /mnt/data && ls -la    |    tail -n 50 train.log")
     with gr.Row():
         send_btn = gr.Button("Run ▶", variant="primary")
+        broadcast_btn = gr.Button("📡 Broadcast to all online", variant="secondary")
         send_status = gr.Markdown("")
 
     history_md = gr.Markdown("_Select a container to see its command history._")
@@ -450,6 +476,8 @@ with gr.Blocks(title="NII Relay", analytics_enabled=False) as demo:
                    [send_status, command_tb])
     command_tb.submit(ui_send, [client_dd, session_tb, command_tb, timeout_nb],
                       [send_status, command_tb])
+    broadcast_btn.click(ui_broadcast, [session_tb, command_tb, timeout_nb],
+                        [send_status, command_tb])
     refresh_btn.click(ui_refresh, [client_dd], [client_dd, history_md])
     client_dd.change(ui_history, [client_dd], history_md)
     client_dd.change(_fetched_files, [client_dd], downloads)