test splash screen builds for pymatting compilation

Author: pricklygorse

.github/workflows/build.yaml

@@ -50,19 +50,15 @@ jobs:
         if: runner.os != 'macOS'
         shell: bash
         run: |
-          pyinstaller --clean --onedir --windowed --name "Interactive-BG-Remover" \
-            --exclude-module torch \
-            backgroundremoval.py
+          pyinstaller --clean build_win_linux.spec
           mkdir -p dist/Interactive-BG-Remover/Models
           mv Models/place_models_here.txt dist/Interactive-BG-Remover/Models/
 
       - name: Build executable (macOS)
         if: runner.os == 'macOS'
         run: |
-          # mac has issues with --onedir. cpu only build so may as well use onefile
-          pyinstaller --clean --onefile --windowed --name "Interactive-BG-Remover" \
-            --exclude-module torch \
-            backgroundremoval.py
+          # mac has issues with --onedir. this is a cpu only build so we can use onefile as it produces smaller binaries than CUDA etc.
+          pyinstaller --clean build_mac.spec
           mkdir -p dist/mac-bundle/Models
           mv dist/Interactive-BG-Remover dist/mac-bundle/
           cp Models/place_models_here.txt dist/mac-bundle/Models/

backgroundremoval.py

@@ -1,4 +1,10 @@
 #!/usr/bin/env python3
+
+try:
+    import pyi_splash # noqa
+    pyi_splash.update_text("Loading Packages")
+except: pass
+
 import sys
 import os
 import math
@@ -22,9 +28,13 @@
 from PIL import Image, ImageOps, ImageDraw, ImageEnhance, ImageGrab, ImageFilter, ImageChops
 import onnxruntime as ort
 
+try: pyi_splash.update_text("Loading pymatting (Compiles on first run, approx 1-2 minutes)")
+except: pass
 
 from pymatting import estimate_alpha_cf, estimate_foreground_ml
 
+try: pyi_splash.update_text("Loading App Scripts")
+except: pass
 
 import src.download_manager as download_manager
 from src.ui_widgets import CollapsibleFrame, SynchronisedGraphicsView
@@ -103,12 +113,15 @@ def get_available_ep_options():
 AVAILABLE_EPS = get_available_ep_options()
 
 
-
 # --- Main Application ---
 
 class BackgroundRemoverGUI(QMainWindow):
     def __init__(self, image_paths):
         super().__init__()
+        
+        try: pyi_splash.update_text("App Initialisation")
+        except: pass
+        
         self.image_paths = image_paths if image_paths else []
         self.current_image_index = 0
         self.setWindowTitle("Interactive Image Background Remover")
@@ -137,7 +150,8 @@ def __init__(self, image_paths):
         self.loaded_sam_models = {}
         self.loaded_matting_models = {} 
         self.model_output_mask = None
-
+        try: pyi_splash.update_text("Loading UI")
+        except: pass
         self.init_ui()
         self.setup_keybindings()
 
@@ -169,6 +183,11 @@ def __init__(self, image_paths):
         self.set_theme(saved_theme)
 
 
+        try: pyi_splash.close()
+        except: pass
+
+
+
     def init_ui(self):
         self.setAcceptDrops(True)
         main = QWidget(); self.setCentralWidget(main)

build_mac.spec

@@ -0,0 +1,50 @@
+# -*- mode: python ; coding: utf-8 -*-
+
+
+a = Analysis(
+    ['backgroundremoval.py'],
+    pathex=[],
+    binaries=[],
+    datas=[],
+    hiddenimports=[],
+    hookspath=[],
+    hooksconfig={},
+    runtime_hooks=[],
+    excludes=['torch'],
+    noarchive=False,
+    optimize=0,
+)
+pyz = PYZ(a.pure)
+splash = Splash(
+    'src/assets/splash.png',
+    binaries=a.binaries,
+    datas=a.datas,
+    text_pos=(10, 30),
+    text_size=12,
+    text_color='white',
+    minify_script=True,
+    always_on_top=False,
+)
+
+exe = EXE(
+    pyz,
+    a.scripts,
+    a.binaries,
+    a.datas,
+    splash,
+    splash.binaries,
+    [],
+    name='Interactive-BG-Remover',
+    debug=False,
+    bootloader_ignore_signals=False,
+    strip=False,
+    upx=True,
+    upx_exclude=[],
+    runtime_tmpdir=None,
+    console=False,
+    disable_windowed_traceback=False,
+    argv_emulation=False,
+    target_arch=None,
+    codesign_identity=None,
+    entitlements_file=None,
+)

build_win_linux.spec

@@ -0,0 +1,56 @@
+# -*- mode: python ; coding: utf-8 -*-
+
+
+a = Analysis(
+    ['backgroundremoval.py'],
+    pathex=[],
+    binaries=[],
+    datas=[],
+    hiddenimports=[],
+    hookspath=[],
+    hooksconfig={},
+    runtime_hooks=[],
+    excludes=['torch'],
+    noarchive=False,
+    optimize=0,
+)
+pyz = PYZ(a.pure)
+splash = Splash(
+    'src/assets/splash.png',
+    binaries=a.binaries,
+    datas=a.datas,
+    text_pos=(10, 30),
+    text_size=12,
+    text_color='white',
+    minify_script=True,
+    always_on_top=False,
+)
+
+exe = EXE(
+    pyz,
+    a.scripts,
+    splash,
+    [],
+    exclude_binaries=True,
+    name='Interactive-BG-Remover-onedir',
+    debug=False,
+    bootloader_ignore_signals=False,
+    strip=False,
+    upx=True,
+    console=False,
+    disable_windowed_traceback=False,
+    argv_emulation=False,
+    target_arch=None,
+    codesign_identity=None,
+    entitlements_file=None,
+)
+coll = COLLECT(
+    exe,
+    a.binaries,
+    a.datas,
+    splash.binaries,
+    strip=False,
+    upx=True,
+    upx_exclude=[],
+    name='Interactive-BG-Remover',
+)