# -*- mode: python ; coding: utf-8 -*-
from os.path import join
from platform import system
from PyInstaller.utils.hooks import copy_metadata

datas = [('src/interface/web', 'src/interface/web')]
datas += copy_metadata('tqdm')
datas += copy_metadata('regex')
datas += copy_metadata('requests')
datas += copy_metadata('packaging')
datas += copy_metadata('filelock')
datas += copy_metadata('numpy')
datas += copy_metadata('tokenizers')


block_cipher = None


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

# Filter out unused, duplicate shared libs
extension = {'Windows': '.dll', 'Darwin': '.dylib', 'Linux': '.so'}[system()]
torch_lib_paths = {
    join('torch', 'lib', 'libtorch_cuda' + extension),
    join('torch', 'lib', 'libtorch_cpu' + extension),
    join('torch', 'lib', 'libtorch_python' + extension)
}
a.datas = [entry for entry in a.datas if not entry[0] in torch_lib_paths]

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

exe = EXE(
    pyz,
    a.scripts,
    a.binaries,
    a.zipfiles,
    a.datas,
    [],
    name='Khoj',
    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='x86_64',
    codesign_identity=None,
    entitlements_file=None,
    icon='src/interface/web/assets/icons/favicon.icns',
)

if system() == 'Darwin':
    app = BUNDLE(
        exe,
        name='Khoj.app',
        icon='src/interface/web/assets/icons/favicon.icns',
        bundle_identifier=None,
    )