added trame reqs

Dockerfile

@@ -14,6 +14,10 @@ ENV OMP_NUM_THREADS=4
 # Environment variables for off-screen rendering
 ENV PYVISTA_OFF_SCREEN=true
 ENV VTK_DEFAULT_RENDER_WINDOW_OFFSCREEN=1
+ENV DISPLAY=:99
+ENV LIBGL_ALWAYS_SOFTWARE=1
+ENV MESA_LOADER_DRIVER_OVERRIDE=llvmpipe
+ENV MESA_GL_VERSION_OVERRIDE=3.3
 
 # System deps + Deadsnakes PPA for Python 3.12
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -76,6 +80,10 @@ RUN chown -R 1000:1000 /app
 # Expose Gradio port
 EXPOSE 7860
 
+# Set HOST and PORT via environment variables
+ENV HOST=0.0.0.0
+ENV PORT=7860
+
 # Run app
 CMD ["python", "app.py"]
 

app.py

@@ -1,10 +1,21 @@
 
-
+# ========= Headless / Offscreen safety (before any VTK import) =========
 import os
+# Set headless rendering environment variables BEFORE any VTK/pyvista imports
+os.environ.setdefault("VTK_DEFAULT_RENDER_WINDOW_OFFSCREEN", "1")
+os.environ.setdefault("LIBGL_ALWAYS_SOFTWARE", "1")
+os.environ.setdefault("MESA_LOADER_DRIVER_OVERRIDE", "llvmpipe")
+os.environ.setdefault("MESA_GL_VERSION_OVERRIDE", "3.3")
+# Use dummy DISPLAY value instead of empty string to avoid X server connection attempts
+if "DISPLAY" not in os.environ or os.environ.get("DISPLAY") == "":
+    os.environ["DISPLAY"] = ":99"
+os.environ.setdefault("PYVISTA_OFF_SCREEN", "True")
+
 import sys
 import tempfile
 from pathlib import Path
 import traceback
+import signal
 
 
 # Writable base dir
@@ -81,41 +92,75 @@ def load_private_app(repo_path):
         raise RuntimeError(f"Failed to load app from private repo: {e}")
 
 def main():
-    hf_token = os.environ.get("HF_TOKEN")
-    repo_id = os.environ.get("REPO_ID")
+    # Setup signal handlers for graceful shutdown
+    def signal_handler(sig, frame):
+        print("\n🛑 Received shutdown signal, cleaning up...")
+        sys.exit(0)
+    
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+    
+    try:
+        hf_token = os.environ.get("HF_TOKEN")
+        repo_id = os.environ.get("REPO_ID")
 
-    if not hf_token or not repo_id:
-        raise ValueError("HF_TOKEN and REPO_ID must be set")
+        if not hf_token or not repo_id:
+            raise ValueError("HF_TOKEN and REPO_ID must be set")
 
-    print(f"🚀 Starting app with repo_id: {repo_id}")
-    print(f"📂 Current working directory: {os.getcwd()}")
-    
-    cache_dir = setup_environment()
-    repo_path = download_private_repo(cache_dir, repo_id, hf_token)    
-    demo = load_private_app(repo_path)
+        print(f"🚀 Starting app with repo_id: {repo_id}")
+        print(f"📂 Current working directory: {os.getcwd()}")
+        
+        cache_dir = setup_environment()
+        repo_path = download_private_repo(cache_dir, repo_id, hf_token)    
+        demo = load_private_app(repo_path)
 
-    # Check if it's a Trame app (has server.start method) or Gradio app (has launch method)
-    if hasattr(demo, 'server') and hasattr(demo.server, 'start'):
-        # Trame app
-        port = int(os.environ.get("PORT", "7860"))
-        demo.server.start(
-            port=port,
-            open_browser=False,
-            show_connection_info=True,
-            backend="aiohttp",
-            exec_mode="main",
-        )
-    elif hasattr(demo, 'launch'):
-        # Gradio app
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=7860,
-            share=False,
-            show_error=True,
-            allowed_paths=[str(repo_path), DATA_DIR]
-        )
-    else:
-        raise RuntimeError("App does not have launch() or server.start() method")
+        # Check if it's a Trame app (has server.start method) or Gradio app (has launch method)
+        if hasattr(demo, 'server') and hasattr(demo.server, 'start'):
+            # Trame app
+            port = int(os.environ.get("PORT", "7860"))
+            host = os.environ.get("HOST", "0.0.0.0")  # Allow host to be configured via env var
+            
+            print(f"🚀 Starting Trame server on {host}:{port}")
+            
+            # Configure server to bind to the specified host
+            try:
+                # Try with host parameter first (if supported)
+                demo.server.start(
+                    host=host,
+                    port=port,
+                    open_browser=False,
+                    show_connection_info=True,
+                    backend="aiohttp",
+                    exec_mode="main",
+                )
+            except TypeError:
+                # If host parameter is not supported, configure via server options
+                # Most Trame versions bind to 0.0.0.0 by default with aiohttp
+                demo.server.start(
+                    port=port,
+                    open_browser=False,
+                    show_connection_info=True,
+                    backend="aiohttp",
+                    exec_mode="main",
+                )
+        elif hasattr(demo, 'launch'):
+            # Gradio app
+            demo.launch(
+                server_name="0.0.0.0",
+                server_port=int(os.environ.get("PORT", "7860")),
+                share=False,
+                show_error=True,
+                allowed_paths=[str(repo_path), DATA_DIR]
+            )
+        else:
+            raise RuntimeError("App does not have launch() or server.start() method")
+    except KeyboardInterrupt:
+        print("\n🛑 Keyboard interrupt received, shutting down...")
+        sys.exit(0)
+    except Exception as e:
+        print(f"❌ Fatal error: {e}")
+        traceback.print_exc()
+        sys.exit(1)
 
 
 if __name__ == "__main__":