| """ |
| Windows Native Taskbar System Tray Companion for Pabot. |
| Places a sleek Pabot icon directly into the Windows Taskbar Notification Area |
| (System Tray) right next to your clock and other system icons. |
| |
| Clicking the icon immediately opens a native Windows 11 flyout popup window! |
| """ |
|
|
| import os |
| import subprocess |
| import sys |
| import threading |
| from pathlib import Path |
| from PIL import Image, ImageDraw |
|
|
| import pystray |
| from pystray import MenuItem as item |
|
|
| |
| REPO_ROOT = Path(__file__).resolve().parents[1] |
| BACKEND_URL = os.environ.get("PABOT_BACKEND_URL", "http://127.0.0.1:8000") |
| WIDGET_URL = f"{BACKEND_URL}/widget.html" |
| DASHBOARD_URL = f"{BACKEND_URL}/" |
|
|
| EDGE_PATH = r"C:\Program Files (x86)\Microsoft\Edge\Application\msedge.exe" |
| if not os.path.exists(EDGE_PATH): |
| EDGE_PATH = r"C:\Program Files\Microsoft\Edge\Application\msedge.exe" |
|
|
|
|
| def create_tray_image(width=64, height=64) -> Image.Image: |
| """Generate a high-res, crisp glowing Pabot icon for the Windows taskbar tray.""" |
| img = Image.new("RGBA", (width, height), (0, 0, 0, 0)) |
| draw = ImageDraw.Draw(img) |
|
|
| |
| draw.ellipse([4, 4, width - 4, height - 4], fill=(124, 58, 237, 255), outline=(167, 139, 250, 200), width=2) |
| |
| |
| |
| cx, cy = width // 2, height // 2 |
| points = [ |
| (cx, 16), |
| (width - 16, cy), |
| (cx, height - 16), |
| (16, cy) |
| ] |
| draw.polygon(points, fill=(245, 158, 11, 230)) |
| |
| |
| draw.ellipse([cx - 4, cy - 4, cx + 4, cy + 4], fill=(255, 255, 255, 255)) |
| return img |
|
|
|
|
| def open_flyout_popup(url=WIDGET_URL): |
| """ |
| Open the widget in a native, frameless Windows 11 flyout popup window. |
| Edge --app mode renders a pristine, native popup without URL bars or tabs. |
| """ |
| try: |
| |
| import ctypes |
| user32 = ctypes.windll.user32 |
| screen_w = user32.GetSystemMetrics(0) |
| screen_h = user32.GetSystemMetrics(1) |
|
|
| win_w = 400 |
| win_h = 640 |
| pos_x = max(0, screen_w - win_w - 18) |
| pos_y = max(0, screen_h - win_h - 68) |
|
|
| if os.path.exists(EDGE_PATH): |
| cmd = [ |
| EDGE_PATH, |
| f"--app={url}", |
| f"--window-size={win_w},{win_h}", |
| f"--window-position={pos_x},{pos_y}", |
| ] |
| subprocess.Popen(cmd) |
| else: |
| import webbrowser |
| webbrowser.open(url) |
| except Exception as exc: |
| print(f"Error opening flyout: {exc}") |
|
|
|
|
| def on_open_flyout(icon, item): |
| open_flyout_popup(WIDGET_URL) |
|
|
|
|
| def on_open_dashboard(icon, item): |
| open_flyout_popup(DASHBOARD_URL) |
|
|
|
|
| def on_quick_task(icon, item): |
| open_flyout_popup(f"{WIDGET_URL}#tasks") |
|
|
|
|
| def on_security_approvals(icon, item): |
| open_flyout_popup(f"{WIDGET_URL}#approvals") |
|
|
|
|
| def on_quit(icon, item): |
| icon.stop() |
|
|
|
|
| def run_tray(): |
| icon_image = create_tray_image() |
|
|
| menu = ( |
| item("⚡ Open Pabot Flyout", on_open_flyout, default=True), |
| item("📋 Quick Tasks", on_quick_task), |
| item("🛡️ Security Approvals", on_security_approvals), |
| pystray.Menu.SEPARATOR, |
| item("🌐 Open Full Dashboard", on_open_dashboard), |
| pystray.Menu.SEPARATOR, |
| item("✕ Exit Tray", on_quit), |
| ) |
|
|
| tray_icon = pystray.Icon( |
| "PabotAssistant", |
| icon_image, |
| "Pabot AI Assistant (Click to open flyout)", |
| menu |
| ) |
|
|
| print("[PABOT] Windows Taskbar System Tray icon started.") |
| print("[PABOT] Single-click or double-click the icon in your taskbar to open the flyout popup!") |
| tray_icon.run() |
|
|
|
|
| if __name__ == "__main__": |
| run_tray() |
|
|
