push

Dockerfile

@@ -26,6 +26,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libxext6 \
     libxrender1 \
     libgl1-mesa-dri \
+    mesa-common-dev \
+    libglu1-mesa-dev \
     && rm -rf /var/lib/apt/lists/*
 
 # Copy requirements and install Python dependencies

renderer.py

@@ -1,11 +1,24 @@
 import os
-os.environ['PYOPENGL_PLATFORM'] = 'osmesa'
 
 import numpy as np
-import pyrender
 import trimesh
 from typing import Tuple, Optional
 
+os.environ['PYOPENGL_PLATFORM'] = 'osmesa'
+
+try:
+    import pyrender
+except Exception as e:
+    print(f"Warning: pyrender import failed with osmesa: {e}")
+    try:
+        os.environ['PYOPENGL_PLATFORM'] = 'egl'
+        import pyrender
+        print("Using EGL platform")
+    except Exception as e2:
+        print(f"Warning: pyrender import failed with egl: {e2}")
+        os.environ['PYOPENGL_PLATFORM'] = 'osmesa'
+        import pyrender
+
 
 class AvatarRenderer:
     def __init__(

requirements.txt

@@ -10,4 +10,6 @@ Pillow
 scikit-learn
 git+https://github.com/mattloper/chumpy.git
 scipy
+PyOpenGL
+PyOpenGL-accelerate