IQ.Pilot Prebuilt Release @ ab07000

tools/iqmacvisiond/menubar.py

@@ -0,0 +1,80 @@
+#!/usr/bin/env python3
+"""
+Copyright © IQ.Lvbs, apart of Project Teal Lvbs, All Rights Reserved, licensed under https://konn3kt.com/tos/
+"""
+from __future__ import annotations
+
+import json
+import os
+import subprocess
+import sys
+import time
+import urllib.request
+from pathlib import Path
+
+import rumps
+
+HERE = Path(__file__).resolve().parent
+STATUS_URL = "http://127.0.0.1:51995/status.json"
+DASHBOARD_URL = "http://127.0.0.1:51995"
+
+
+class IQVisionApp(rumps.App):
+  def __init__(self):
+    super().__init__("IQ Vision", title="◎", quit_button=None)
+    self.item_state = rumps.MenuItem("Starting…")
+    self.item_device = rumps.MenuItem("Device: —")
+    self.item_frames = rumps.MenuItem("Frames: —")
+    self.item_exec = rumps.MenuItem("Inference: —")
+    self.item_awake = rumps.MenuItem("Keep awake: —")
+    self.menu = [
+      self.item_state, None,
+      self.item_device, self.item_frames, self.item_exec, self.item_awake, None,
+      rumps.MenuItem("Open Dashboard", callback=self.open_dashboard),
+      rumps.MenuItem("Quit IQ Vision", callback=self.quit_app),
+    ]
+    self.proc: subprocess.Popen | None = None
+    self._start_server()
+    self.timer = rumps.Timer(self.refresh, 1)
+    self.timer.start()
+
+  def _start_server(self) -> None:
+    env = dict(os.environ)
+    self.proc = subprocess.Popen([sys.executable, str(HERE / "server.py")], env=env)
+
+  def refresh(self, _) -> None:
+    if self.proc is not None and self.proc.poll() is not None:
+      self.title = "◎!"
+      self.item_state.title = "Server stopped — reopen the app"
+      return
+    try:
+      with urllib.request.urlopen(STATUS_URL, timeout=0.8) as r:
+        s = json.load(r)
+    except Exception:
+      self.title = "◎"
+      self.item_state.title = "Warming up…"
+      return
+    live = s.get("connected") and s.get("fresh")
+    self.title = "◉" if live else "◎"
+    self.item_state.title = "Connected" if live else "Waiting for device"
+    self.item_device.title = f"Device: {s.get('peer') or '—'}"
+    self.item_frames.title = f"Frames: {s.get('infer_count', 0):,}"
+    p50, p99 = s.get("exec_p50_ms", 0.0), s.get("exec_p99_ms", 0.0)
+    self.item_exec.title = f"Inference: {p50:.0f} / {p99:.0f} ms" if p50 else "Inference: —"
+    self.item_awake.title = f"Keep awake: {'on' if s.get('awake') else 'off'}"
+
+  def open_dashboard(self, _) -> None:
+    subprocess.Popen(["open", DASHBOARD_URL])
+
+  def quit_app(self, _) -> None:
+    if self.proc is not None:
+      self.proc.terminate()
+      try:
+        self.proc.wait(timeout=3)
+      except subprocess.TimeoutExpired:
+        self.proc.kill()
+    rumps.quit_application()
+
+
+if __name__ == "__main__":
+  IQVisionApp().run()