shutdown test

Kiln-AI · Aug 12, 2024 · 59e4e7d · 59e4e7d
1 parent 6aa2481
commit 59e4e7d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/desktop/desktop.py b/desktop/desktop.py
@@ -91,6 +91,7 @@ def run_taskbar():
     tray = pystray.Icon("fune", image, "fune", menu)
     # running detached since we use tk mainloop to get events from dock icon
     tray.run_detached()
+    return tray
 
 
 def close_splash():
@@ -122,7 +123,9 @@ def close_splash():
         root.withdraw()  # remove the window
         # Register callback for the dock icon to reopen the web app
         root.createcommand("tk::mac::ReopenApplication", show_studio)
-        run_taskbar()
+        tray = run_taskbar()
         root.after(10, show_studio)
         root.after(10, close_splash)
         root.mainloop()
+        tray.stop()
+        raise SystemExit(0)