# -*- mode: python ; coding: utf-8 -*-
from PyInstaller.utils.hooks import collect_data_files
from PyInstaller.utils.hooks import collect_submodules
from PyInstaller.utils.hooks import copy_metadata

datas = [('web/index.html', 'web'), ('icon.png', '.'), ('scene_setup.py', '.')]
hiddenimports = [
    'image_to_3d',
    'generate_equirect',
    'diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion',
    'diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion_inpaint',
    'diffusers.pipelines.stable_diffusion.pipeline_output',
    'diffusers.models.autoencoders.autoencoder_kl',
    'diffusers.models.autoencoders.vae',
    'diffusers.models.unets.unet_2d_condition',
    'diffusers.models.unets.unet_2d_blocks',
    'diffusers.schedulers.scheduling_ddim',
    'diffusers.schedulers.scheduling_ddpm',
    'diffusers.schedulers.scheduling_dpmsolver_multistep',
    'diffusers.schedulers.scheduling_euler_discrete',
    'diffusers.schedulers.scheduling_lms_discrete',
    'diffusers.schedulers.scheduling_pndm',
    'transformers.models.auto.configuration_auto',
    'transformers.models.auto.feature_extraction_auto',
    'transformers.models.auto.image_processing_auto',
    'transformers.models.auto.modeling_auto',
    'transformers.models.auto.tokenization_auto',
    'transformers.models.clip.configuration_clip',
    'transformers.models.clip.image_processing_clip',
    'transformers.models.clip.modeling_clip',
    'transformers.models.clip.processing_clip',
    'transformers.models.clip.tokenization_clip',
    'transformers.models.clip.tokenization_clip_fast',
    'safetensors.torch',
]
datas += collect_data_files('diffusers')
datas += collect_data_files('transformers')
datas += collect_data_files('huggingface_hub')
datas += copy_metadata('replicate')
hiddenimports += collect_submodules(
    'diffusers.schedulers',
    filter=lambda name: '.scheduling_' in name and '_flax' not in name,
)


a = Analysis(
    ['main.py'],
    pathex=[],
    binaries=[],
    datas=datas,
    hiddenimports=hiddenimports,
    hookspath=[],
    hooksconfig={},
    runtime_hooks=[],
    excludes=[
        'gi',
        'PyQt5',
        'PyQt6',
        'PySide2',
        'PySide6',
        'tkinter',
        'altair',
        'cv2',
        'datasets',
        'flax',
        'jax',
        'jaxlib',
        'librosa',
        'llvmlite',
        'matplotlib',
        'nltk',
        'numba',
        'onnx',
        'onnxruntime',
        'pandas',
        'pygame',
        'sklearn',
        'soundfile',
        'spacy',
        'tensorflow',
        'torchaudio',
        'torchvision',
    ],
    noarchive=False,
    optimize=0,
)
pyz = PYZ(a.pure)

exe = EXE(
    pyz,
    a.scripts,
    [],
    exclude_binaries=True,
    name='3D Model Generator',
    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=['icon.icns'],
)
coll = COLLECT(
    exe,
    a.binaries,
    a.datas,
    strip=False,
    upx=True,
    upx_exclude=[],
    name='3D Model Generator',
)
app = BUNDLE(
    coll,
    name='3D Model Generator.app',
    icon='icon.icns',
    bundle_identifier=None,
)