eigh 0.1.0__tar.gz

eigh-0.1.0/.github/workflows/publish.yml

@@ -0,0 +1,70 @@
+name: Build and publish wheels
+
+on:
+  workflow_dispatch:
+  push:
+    tags:
+      - 'v*'
+
+jobs:
+  build_wheels:
+    name: Build wheels on ${{ matrix.os }}
+    runs-on: ${{ matrix.os }}
+    strategy:
+      matrix:
+        os: [ubuntu-latest, macos-latest]
+        # os: [ubuntu-latest, macos-latest, windows-latest]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Build wheels
+        uses: pypa/cibuildwheel@v2.22.0
+        env:
+          CIBW_ARCHS_MACOS: x86_64 arm64
+          # Linux: only x86_64 for now (aarch64 requires QEMU and is slow)
+          CIBW_ARCHS_LINUX: x86_64
+          # Use manylinux_2_28 (AlmaLinux 8) - CentOS 7 repos are dead
+          CIBW_MANYLINUX_X86_64_IMAGE: manylinux_2_28
+          # Skip: PyPy (pp*), Python 3.8/3.9 (jaxlib compat), musllinux (Alpine)
+          CIBW_SKIP: "pp* cp38-* cp39-* *-musllinux_*"
+
+      - uses: actions/upload-artifact@v4
+        with:
+          name: cibw-wheels-${{ matrix.os }}-${{ strategy.job-index }}
+          path: ./wheelhouse/*.whl
+
+  build_sdist:
+    name: Build source distribution
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Build sdist
+        run: pipx run build --sdist
+
+      - uses: actions/upload-artifact@v4
+        with:
+          name: cibw-sdist
+          path: dist/*.tar.gz
+
+  upload_pypi:
+    needs: [build_wheels, build_sdist]
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      id-token: write
+    if: github.event_name == 'push' && startsWith(github.ref, 'refs/tags/v')
+    steps:
+      - uses: actions/download-artifact@v4
+        with:
+          pattern: cibw-*
+          path: dist
+          merge-multiple: true
+
+      - uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          # registry-url: https://test.pypi.org/legacy/
+          # To use trusted publishing, remove 'password' and setup trust on PyPI
+          # Or use secrets.PYPI_API_TOKEN
+           password: ${{ secrets.PYPI_API_TOKEN }}

eigh-0.1.0/.gitignore

@@ -0,0 +1,134 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+.DS_store
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+#lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# VSCode
+.vscode/*

eigh-0.1.0/CHANGES.md

@@ -0,0 +1,33 @@
+# Changes Made
+
+Renaming to prevent conflicts with JAX's standard `eigh`.
+
+| File | Old Name | New Name |
+| :--- | :--- | :--- |
+| `lapack.cc` | `lapack_ssygvd` | `eigh_lapack_ssygvd` |
+| | `lapack_dsygvd` | `eigh_lapack_dsygvd` |
+| | `lapack_chegvd` | `eigh_lapack_chegvd` |
+| | `lapack_zhegvd` | `eigh_lapack_zhegvd` |
+| | `lapack_*_ffi` variants | `eigh_lapack_*_ffi` |
+| `solver.cc` | `cusolver_sygvd_ffi` | `eigh_cusolver_sygvd_ffi` |
+| `_core.py` | Updated `prepare_lapack_call` and CUDA target name | - |
+
+The FFI target names are just string identifiers used to look up the registered function pointers.
+The flow is:
+
+1. C++ registers function pointer with name `"eigh_lapack_ssygvd_ffi"`
+2. Python calls `ffi.register_ffi_target("eigh_lapack_ssygvd_ffi", ...)`
+3. During JIT lowering, Python requests target `"eigh_lapack_ssygvd_ffi"`
+4. XLA looks up the function pointer by that name
+
+As long as both sides agree on the name (which they now do), everything works. The actual kernel code is unchanged.
+
+# To Apply
+
+You need to rebuild the package since the C++ files changed:
+
+```bash
+pip install -e .
+```
+
+After rebuilding, both JAX's standard `eigh` and this package's `eigh` can coexist without conflicts.

eigh-0.1.0/CMakeLists.txt

@@ -0,0 +1,150 @@
+cmake_minimum_required(VERSION 3.18)
+project(eigh_standalone LANGUAGES CXX)
+
+set(CMAKE_CXX_STANDARD 17)
+set(CMAKE_CXX_STANDARD_REQUIRED ON)
+set(CMAKE_POSITION_INDEPENDENT_CODE ON)
+
+# Fix for NVHPC compiler compatibility with nanobind
+if(CMAKE_CXX_COMPILER_ID MATCHES "NVHPC|PGI")
+    add_compile_definitions(_GLIBCXX_USE_CXX11_ABI=1)
+    add_compile_definitions(__GXX_ABI_VERSION=1016)
+endif()
+
+# Find required packages
+# Use Development.Module instead of Development - manylinux containers
+# don't have full Python development files (libpython), but extension
+# modules don't need them anyway
+find_package(Python REQUIRED COMPONENTS Interpreter Development.Module)
+
+# Find nanobind
+execute_process(
+    COMMAND "${Python_EXECUTABLE}" -m nanobind --cmake_dir
+    OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE NB_DIR
+)
+list(APPEND CMAKE_PREFIX_PATH "${NB_DIR}")
+find_package(nanobind CONFIG REQUIRED)
+
+# Find BLAS and LAPACK
+# If NVHPC SDK paths are in environment, set hints for CMake
+set(NVHPC_LIB_PATH "/softs/nvidia/hpc_sdk/Linux_x86_64/22.1/compilers/lib")
+if(EXISTS "${NVHPC_LIB_PATH}/libblas.so")
+    set(BLA_VENDOR "NVHPC")
+    set(BLAS_LIBRARIES "${NVHPC_LIB_PATH}/libblas.so")
+    set(LAPACK_LIBRARIES "${NVHPC_LIB_PATH}/liblapack.so")
+    set(BLAS_FOUND TRUE)
+    set(LAPACK_FOUND TRUE)
+    message(STATUS "Using NVHPC BLAS: ${BLAS_LIBRARIES}")
+    message(STATUS "Using NVHPC LAPACK: ${LAPACK_LIBRARIES}")
+else()
+    find_package(BLAS REQUIRED)
+    find_package(LAPACK REQUIRED)
+endif()
+
+# XLA headers (from jaxlib)
+execute_process(
+    COMMAND "${Python_EXECUTABLE}" -c "import jaxlib; import os; print(os.path.join(jaxlib.__path__[0], 'include'))"
+    OUTPUT_STRIP_TRAILING_WHITESPACE OUTPUT_VARIABLE XLA_INCLUDE_DIR
+    RESULT_VARIABLE XLA_RESULT
+)
+
+if(NOT XLA_RESULT EQUAL 0 OR NOT EXISTS "${XLA_INCLUDE_DIR}")
+    message(FATAL_ERROR "Could not find XLA headers. Make sure jaxlib is installed: pip install jaxlib")
+endif()
+
+message(STATUS "XLA headers found at: ${XLA_INCLUDE_DIR}")
+
+# Include directories
+include_directories(
+    ${CMAKE_SOURCE_DIR}/include
+    ${CMAKE_SOURCE_DIR}/src/cpu
+    ${CMAKE_SOURCE_DIR}/src/cuda
+    ${XLA_INCLUDE_DIR}
+)
+
+# CPU LAPACK module
+nanobind_add_module(
+    eigh_lapack
+    STABLE_ABI
+    src/cpu/lapack.cc
+    src/cpu/lapack_kernels.cc
+)
+target_link_libraries(eigh_lapack PRIVATE ${BLAS_LIBRARIES} ${LAPACK_LIBRARIES})
+
+# Add NVHPC Fortran runtime when linking against NVHPC BLAS/LAPACK
+if(EXISTS "${NVHPC_LIB_PATH}/libnvf.so")
+    find_library(NVF_LIB NAMES nvf PATHS "${NVHPC_LIB_PATH}" NO_DEFAULT_PATH)
+    if(NVF_LIB)
+        target_link_libraries(eigh_lapack PRIVATE ${NVF_LIB} rt)
+        message(STATUS "Found NVHPC Fortran runtime: ${NVF_LIB}")
+    endif()
+endif()
+
+set_target_properties(eigh_lapack PROPERTIES
+    LIBRARY_OUTPUT_DIRECTORY ${CMAKE_SOURCE_DIR}/src/python/eigh
+)
+
+# CUDA module (optional - only if CUDA is available)
+include(CheckLanguage)
+check_language(CUDA)
+
+if(CMAKE_CUDA_COMPILER)
+    enable_language(CUDA)
+    set(CMAKE_CUDA_STANDARD 17)
+    set(CMAKE_CUDA_STANDARD_REQUIRED ON)
+
+    find_package(CUDAToolkit REQUIRED)
+
+    nanobind_add_module(
+        eigh_cuda
+        STABLE_ABI
+        src/cuda/solver.cc
+        src/cuda/solver_kernels.cc
+    )
+
+    # Add CUDA include directories
+    target_include_directories(eigh_cuda PRIVATE
+        ${CUDAToolkit_INCLUDE_DIRS}
+    )
+
+    set_source_files_properties(
+        src/cuda/solver_kernels.cc
+        PROPERTIES LANGUAGE CUDA
+    )
+
+    # Link CUDA libraries - handle both imported targets and direct paths
+    if(TARGET CUDA::cudart AND TARGET CUDA::cusolver AND TARGET CUDA::cublas)
+        target_link_libraries(eigh_cuda PRIVATE
+            CUDA::cudart
+            CUDA::cusolver
+            CUDA::cublas
+        )
+    else()
+        # Fallback for NVHPC SDK where imported targets may not be available
+        target_link_libraries(eigh_cuda PRIVATE
+            ${CUDAToolkit_LIBRARY_DIR}/libcudart.so
+            ${CUDAToolkit_LIBRARY_DIR}/libcusolver.so
+            ${CUDAToolkit_LIBRARY_DIR}/libcublas.so
+        )
+    endif()
+
+    set_target_properties(eigh_cuda PROPERTIES
+        LIBRARY_OUTPUT_DIRECTORY ${CMAKE_SOURCE_DIR}/src/python/eigh
+        CUDA_SEPARABLE_COMPILATION ON
+    )
+
+    message(STATUS "CUDA support enabled")
+else()
+    message(STATUS "CUDA not found - GPU support will be disabled")
+endif()
+
+# Install target - scikit-build-core will handle the final install location
+install(TARGETS eigh_lapack
+    LIBRARY DESTINATION eigh
+)
+
+if(TARGET eigh_cuda)
+    install(TARGETS eigh_cuda
+        LIBRARY DESTINATION eigh
+    )
+endif()

eigh-0.1.0/MANIFEST.in

@@ -0,0 +1,13 @@
+# Include source files for building
+recursive-include include *.h
+recursive-include src/cpu *.cc *.h
+recursive-include src/cuda *.cc *.h
+include CMakeLists.txt
+include README.md
+include requirements.txt
+
+# Exclude build artifacts
+global-exclude *.so
+global-exclude *.dylib
+global-exclude __pycache__
+global-exclude *.pyc

eigh-0.1.0/PKG-INFO

@@ -0,0 +1,104 @@
+Metadata-Version: 2.2
+Name: eigh
+Version: 0.1.0
+Summary: Differentiable eigenvalue decomposition with JAX (CPU/GPU)
+Keywords: jax,eigenvalue,linear-algebra,gpu,cuda,autodiff
+Author-Email: Xing Zhang <fishjojo@gmail.com>
+License: Apache-2.0
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Science/Research
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Mathematics
+Project-URL: Homepage, https://github.com/fishjojo/pyscfad
+Project-URL: Repository, https://github.com/fishjojo/pyscfad
+Requires-Python: >=3.10
+Requires-Dist: jax>=0.4.0
+Requires-Dist: jaxlib>=0.4.0
+Requires-Dist: numpy>=1.20.0
+Provides-Extra: cuda
+Requires-Dist: jax[cuda12]>=0.4.0; extra == "cuda"
+Provides-Extra: cuda-local
+Requires-Dist: jax[cuda12-local]>=0.4.0; extra == "cuda-local"
+Provides-Extra: test
+Requires-Dist: pytest>=7.0.0; extra == "test"
+Requires-Dist: scipy>=1.7.0; extra == "test"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: scipy>=1.7.0; extra == "dev"
+Description-Content-Type: text/markdown
+
+# Differentiable Generalized Eigenvalue Decomposition
+
+<img src="img/eig.png" alt="Eigh Logo" width="400">
+
+Standalone implementation of differentiable eigenvalue decomposition with CPU (LAPACK) and GPU (cuSOLVER) backends. Extracted from [pyscfad](https://github.com/fishjojo/pyscfad).
+
+## Features
+- **Generalized Problems**: `A @ V = B @ V @ diag(W)`, etc.
+- **JAX Integrated**: Full support for `jit`, `vmap`, `grad`, and `jvp`.
+- **High Performance**: Optimized LAPACK (CPU) and cuSOLVER (GPU) kernels.
+- **Precision**: `float32/64` and `complex64/128`.
+- **Degeneracy Handling**: Configurable `deg_thresh` for stable gradients.
+
+## Installation & Quick Start
+
+```bash
+# Install from source
+pip install .
+
+# For GPU support in this environment
+pip install .[cuda-local]
+```
+
+### Usage Example
+```python
+import jax
+import jax.numpy as jnp
+from eigh import eigh
+
+jax.config.update("jax_enable_x64", True)
+A = jnp.array([[2., 1.], [1., 2.]])
+w, v = eigh(A) # Standard
+grad = jax.grad(lambda A: eigh(A)[0].sum())(A) # Differentiable
+```
+
+## API Reference
+- **`eigh(a, b=None, *, lower=True, eigvals_only=False, type=1, deg_thresh=1e-9)`**
+  Scipy-compatible interface. `type` supports 1: `A@v=B@v@λ`, 2: `A@B@v=v@λ`, 3: `B@A@v=v@λ`.
+- **`eigh_gen(a, b, *, lower=True, itype=1, deg_thresh=1e-9)`**
+  Lower-level generalized solver.
+
+## Degenerate Eigenvalues & Gradients
+Individual eigenvalue gradients are ill-defined for degenerate (repeated) eigenvalues. However, symmetric functions (like `sum`, `var`, `trace`) have stable gradients. The `deg_thresh` parameter (default `1e-9`) masks divisions by near-zero gaps to maintain stability.
+
+## Development & Testing
+- **Requirements**: CMake 3.18+, C++17, JAX, NumPy, LAPACK/CUDA.
+- **Tests**:
+  ```bash
+  pytest tests/test_eigh.py     # Core functionality
+  pytest tests/test_eigh_gen.py # Generalized itypes
+  pytest tests/test_eigh_jit.py # JIT & vmap
+  ```
+- **GPU Setup**:
+  ```bash
+  source setup_gpu_env_clean.sh
+  ./run_gpu.sh python example_simple.py
+  ```
+
+## License & Citation
+Apache License 2.0. If used in research, please cite:
+```bibtex
+@software{pyscfad,
+  author = {Zhang, Xing},
+  title = {PySCFad: Automatic Differentiation for PySCF},
+  url = {https://github.com/fishjojo/pyscfad},
+  year = {2021-2025}
+}
+```

eigh-0.1.0/README.md

@@ -0,0 +1,68 @@
+# Differentiable Generalized Eigenvalue Decomposition
+
+<img src="img/eig.png" alt="Eigh Logo" width="400">
+
+Standalone implementation of differentiable eigenvalue decomposition with CPU (LAPACK) and GPU (cuSOLVER) backends. Extracted from [pyscfad](https://github.com/fishjojo/pyscfad).
+
+## Features
+- **Generalized Problems**: `A @ V = B @ V @ diag(W)`, etc.
+- **JAX Integrated**: Full support for `jit`, `vmap`, `grad`, and `jvp`.
+- **High Performance**: Optimized LAPACK (CPU) and cuSOLVER (GPU) kernels.
+- **Precision**: `float32/64` and `complex64/128`.
+- **Degeneracy Handling**: Configurable `deg_thresh` for stable gradients.
+
+## Installation & Quick Start
+
+```bash
+# Install from source
+pip install .
+
+# For GPU support in this environment
+pip install .[cuda-local]
+```
+
+### Usage Example
+```python
+import jax
+import jax.numpy as jnp
+from eigh import eigh
+
+jax.config.update("jax_enable_x64", True)
+A = jnp.array([[2., 1.], [1., 2.]])
+w, v = eigh(A) # Standard
+grad = jax.grad(lambda A: eigh(A)[0].sum())(A) # Differentiable
+```
+
+## API Reference
+- **`eigh(a, b=None, *, lower=True, eigvals_only=False, type=1, deg_thresh=1e-9)`**
+  Scipy-compatible interface. `type` supports 1: `A@v=B@v@λ`, 2: `A@B@v=v@λ`, 3: `B@A@v=v@λ`.
+- **`eigh_gen(a, b, *, lower=True, itype=1, deg_thresh=1e-9)`**
+  Lower-level generalized solver.
+
+## Degenerate Eigenvalues & Gradients
+Individual eigenvalue gradients are ill-defined for degenerate (repeated) eigenvalues. However, symmetric functions (like `sum`, `var`, `trace`) have stable gradients. The `deg_thresh` parameter (default `1e-9`) masks divisions by near-zero gaps to maintain stability.
+
+## Development & Testing
+- **Requirements**: CMake 3.18+, C++17, JAX, NumPy, LAPACK/CUDA.
+- **Tests**:
+  ```bash
+  pytest tests/test_eigh.py     # Core functionality
+  pytest tests/test_eigh_gen.py # Generalized itypes
+  pytest tests/test_eigh_jit.py # JIT & vmap
+  ```
+- **GPU Setup**:
+  ```bash
+  source setup_gpu_env_clean.sh
+  ./run_gpu.sh python example_simple.py
+  ```
+
+## License & Citation
+Apache License 2.0. If used in research, please cite:
+```bibtex
+@software{pyscfad,
+  author = {Zhang, Xing},
+  title = {PySCFad: Automatic Differentiation for PySCF},
+  url = {https://github.com/fishjojo/pyscfad},
+  year = {2021-2025}
+}
+```

eigh-0.1.0/build.sh

@@ -0,0 +1,80 @@
+#!/bin/bash
+# Build script for eigh_standalone
+
+set -e
+
+echo "==================================="
+echo "Building eigh_standalone"
+echo "==================================="
+
+# Detect Python (prefer .venv if it exists)
+if [ -f "../.venv/bin/python" ]; then
+    PYTHON="../.venv/bin/python"
+    echo "Using virtual environment Python"
+elif [ -f ".venv/bin/python" ]; then
+    PYTHON=".venv/bin/python"
+    echo "Using virtual environment Python"
+elif command -v python &> /dev/null; then
+    PYTHON="python"
+else
+    echo "Error: Python not found"
+    exit 1
+fi
+
+echo "Python: $($PYTHON --version)"
+
+# Check CMake
+if ! command -v cmake &> /dev/null; then
+    echo "Error: CMake not found. Please install CMake 3.18+"
+    exit 1
+fi
+
+echo "CMake: $(cmake --version | head -n1)"
+
+# Check CUDA (optional)
+if command -v nvcc &> /dev/null; then
+    echo "CUDA: $(nvcc --version | grep release)"
+    CUDA_AVAILABLE=1
+else
+    echo "CUDA: Not found (GPU support disabled)"
+    CUDA_AVAILABLE=0
+fi
+
+# Install Python dependencies
+echo ""
+echo "Installing Python dependencies..."
+$PYTHON -m pip install -r requirements.txt
+
+# Create build directory
+echo ""
+echo "Creating build directory..."
+rm -rf build
+mkdir build
+cd build
+
+# Configure
+echo ""
+echo "Configuring CMake..."
+if [ $CUDA_AVAILABLE -eq 1 ]; then
+    cmake -DPython_EXECUTABLE=$PYTHON ..
+else
+    cmake -DPython_EXECUTABLE=$PYTHON -DCMAKE_DISABLE_FIND_PACKAGE_CUDA=TRUE ..
+fi
+
+# Build
+echo ""
+echo "Building..."
+make -j$(nproc 2>/dev/null || sysctl -n hw.ncpu 2>/dev/null || echo 4)
+
+# Check output
+echo ""
+echo "==================================="
+echo "Build complete!"
+echo "==================================="
+echo ""
+echo "Built modules:"
+ls -lh ../src/python/*.so 2>/dev/null || ls -lh ../src/python/*.dylib 2>/dev/null || echo "No modules found"
+
+echo ""
+echo "To test, run:"
+echo "  python tests/test_eigh.py"

eigh-0.1.0/debug/diag_jax.py

@@ -0,0 +1,29 @@
+import os
+import sys
+
+print(f"Python version: {sys.version}")
+print(f"LD_LIBRARY_PATH: {os.environ.get('LD_LIBRARY_PATH', 'Not set')}")
+
+try:
+    import jax
+    print(f"JAX version: {jax.__version__}")
+
+    # Try to get devices
+    try:
+        devices = jax.devices()
+        print(f"Devices: {devices}")
+    except Exception as dev_err:
+        print(f"Error getting devices: {dev_err}")
+
+    # Check backends
+    try:
+        import jax.extend as jex
+        backend = jex.backend.get_backend()
+        print(f"Default backend: {backend.platform}")
+    except Exception as back_err:
+        print(f"Error getting backend: {back_err}")
+
+except Exception as e:
+    print(f"Main Error: {e}")
+    import traceback
+    traceback.print_exc()