snapdoc_ai.spec

# -*- mode: python ; coding: utf-8 -*-

import sys
import os
from PyInstaller.utils.hooks import collect_all, collect_submodules

block_cipher = None

# Get the directory for voice models (needed for voice recognition)
voice_model_dir = os.path.join(os.path.expanduser('~'), '.snapdoc-ai', 'voice-model')
voice_model_exists = os.path.exists(voice_model_dir)

# Collect all necessary packages
packages = ['snapdoc_ai', 'customtkinter', 'onnx', 'torch', 'transformers', 'PIL', 
            'pdfplumber', 'docx', 'tkinter', 'structlog', 'dynaconf']

if voice_model_exists:
    packages.extend(['pyttsx3', 'vosk', 'pyaudio', 'numpy'])

# Collect data from all packages
datas = []
binaries = []
hiddenimports = []

for package in packages:
    try:
        pkg_data, pkg_binaries, pkg_hiddenimports = collect_all(package)
        datas.extend(pkg_data)
        binaries.extend(pkg_binaries)
        hiddenimports.extend(pkg_hiddenimports)
    except Exception as e:
        print(f"Error collecting package {package}: {e}")

# Add any additional hidden imports that might be missed
hiddenimports.extend([
    'customtkinter.windows.widgets',
    'customtkinter.windows.ctk_tk',
    'customtkinter.windows.ctk_toplevel',
    'PIL._tkinter_finder',
    'dynaconf.loaders.toml_loader',
])

# Add config files
datas.append(('src/snapdoc_ai/default_settings.toml', 'snapdoc_ai'))

# Add voice models if they exist
if voice_model_exists:
    datas.append((voice_model_dir, 'voice-model'))

a = Analysis(
    ['src/snapdoc_ai/__main__.py'],
    pathex=[],
    binaries=binaries,
    datas=datas,
    hiddenimports=hiddenimports,
    hookspath=[],
    hooksconfig={},
    runtime_hooks=[],
    excludes=[],
    win_no_prefer_redirects=False,
    win_private_assemblies=False,
    cipher=block_cipher,
    noarchive=False,
)

pyz = PYZ(a.pure, a.zipped_data, cipher=block_cipher)

exe = EXE(
    pyz,
    a.scripts,
    [],
    exclude_binaries=True,
    name='SnapDoc AI',
    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,
    icon='resources/app_icon.ico',
    uac_admin=False,
)

coll = COLLECT(
    exe,
    a.binaries,
    a.zipfiles,
    a.datas,
    strip=False,
    upx=True,
    upx_exclude=[],
    name='SnapDoc AI',
)