added trame reqs

Dockerfile

@@ -1,27 +1,28 @@
-
 # Use NVIDIA CUDA base image (Ubuntu 22.04 jammy)
 FROM nvidia/cuda:12.3.2-devel-ubuntu22.04
 
-# Non-interactive apt
+# ---------------- Base env ----------------
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PYTHONUNBUFFERED=1
 ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=${CUDA_HOME}/bin:/usr/local/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
 
-# Proxy (auto-configurable: overridden at runtime if needed)
-# ENV http_proxy=""
-# ENV https_proxy=""
-# ENV HTTP_PROXY=""
-# ENV HTTPS_PROXY=""
+# Fix OpenMP threads to a valid integer (libgomp warning)
+ENV OMP_NUM_THREADS=4
+
+# Force headless: never try X11 DISPLAY
+# ENV DISPLAY=""
 
-# System deps + Deadsnakes PPA for Python 3.12
+# ---------------- System deps + Python 3.12 + headless GL ----------------
 RUN apt-get update && apt-get install -y --no-install-recommends \
     software-properties-common \
     ca-certificates \
     curl wget git build-essential \
     libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender1 \
     libgomp1 libgcc-s1 \
+    libosmesa6 libosmesa6-dev \
+    libegl1-mesa libegl1-mesa-dev \
  && add-apt-repository ppa:deadsnakes/ppa -y \
  && apt-get update && apt-get install -y --no-install-recommends \
     python3.12 python3.12-dev python3.12-venv \
@@ -32,7 +33,7 @@ RUN curl -sS https://bootstrap.pypa.io/get-pip.py -o /tmp/get-pip.py \
  && python3.12 /tmp/get-pip.py \
  && rm /tmp/get-pip.py
 
-# Make 'python' and 'pip' point to 3.12
+# Make 'python' and 'pip' point to Python 3.12
 RUN ln -sf /usr/bin/python3.12 /usr/bin/python && \
     ln -sf /usr/local/bin/pip3.12 /usr/local/bin/pip
 
@@ -41,7 +42,7 @@ ENV APP_DATA_DIR=/data
 ENV HF_HOME=/data/hf_home
 ENV HUGGINGFACE_HUB_CACHE=/data/hf_home
 ENV TRANSFORMERS_CACHE=/data/hf_home
-ENV MPLCONFIGDIR=/data/matplotlib   
+ENV MPLCONFIGDIR=/data/matplotlib
 
 RUN mkdir -p /data/geometry /data/solution /data/weights /data/hf_home /data/matplotlib \
  && chmod -R 777 /data
@@ -61,19 +62,17 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel \
  && pip install --no-cache-dir -r requirements.txt
 
-# Copy the rest of the source
+# Copy the rest of the source code
 COPY . .
 
-# Create cache directory for downloaded repos
-RUN mkdir -p /app/cache && chmod 777 /app/cache
-
 # Permissions
 RUN chmod +x app.py
 RUN chown -R 1000:1000 /app
 
-# Expose Gradio port
+# Expose Trame/Gradio port
 EXPOSE 7860
 
-# Run app
-CMD ["python", "app.py"]
-
+# Run app (Trame or Gradio, same entrypoint)
+# CMD ["python", "app.py"]
+# Run app on 0.0.0.0 so HF proxy can reach it
+CMD python app.py --host 0.0.0.0 --port ${PORT:-7860} --server

app.py

@@ -6,12 +6,6 @@ import tempfile
 from pathlib import Path
 import traceback
 
-# Proxy support (office networks often require this)
-# proxy = os.environ.get("HTTP_PROXY") or os.environ.get("http_proxy")
-# if proxy:
-#     print(f"🌐 Using proxy: {proxy}")
-# else:
-#     print("🚀 No proxy detected, connecting directly")
 
 # Writable base dir
 DATA_DIR = os.environ.get("APP_DATA_DIR", os.path.join(tempfile.gettempdir(), "appdata"))
@@ -52,15 +46,15 @@ def download_private_repo(cache_dir, repo_id, hf_token):
             continue
             
         if os.path.isdir(src_path):
-            print(f"📁 Copying directory: {item}")
+            # print(f"📁 Copying directory: {item}")
             if os.path.exists(dst_path):
                 shutil.rmtree(dst_path)
             shutil.copytree(src_path, dst_path)
-            print(f"✅ Directory {item} copied successfully")
+            # print(f"✅ Directory {item} copied successfully")
         elif os.path.isfile(src_path):
-            print(f"📄 Copying file: {item}")
+            # print(f"📄 Copying file: {item}")
             shutil.copy2(src_path, dst_path)
-            print(f"✅ File {item} copied successfully")
+            # print(f"✅ File {item} copied successfully")
     
     print(f"🎉 All repository contents copied to current working directory")
     

requirements.txt

@@ -1,42 +1,75 @@
-
-huggingface_hub>=0.16.0
 --index-url https://download.pytorch.org/whl/cu121
 --extra-index-url https://pypi.org/simple
 
-torch==2.5.0
-torchvision
-torchaudio
+##############################
+# Core ML stack (CUDA build) #
+##############################
 
-torch_geometric
+torch==2.5.0
+torchvision==0.20.0
+torchaudio==2.5.0
 
 -f https://data.pyg.org/whl/torch-2.5.0+cu121.html
+torch_geometric
 pyg_lib
 torch_scatter
 torch_sparse
 torch_cluster
 torch_spline_conv
 
+#########################
+# Numerics / science    #
+#########################
+
+numpy
+scipy
+pandas
+scikit-learn
+
+#########################
+# Plotting & misc       #
+#########################
+
 matplotlib
-vtk
-timm
-einops
 seaborn
-pyvista
 imageio
-pandas
 meshio
-accelerate
-omegaconf
+h5py
 tensorboard
+
+#########################
+# Model / config helpers#
+#########################
+
+timm
+einops
+omegaconf
+accelerate
+huggingface-hub
 lion-pytorch
-scikit-learn
 
-gradio==5.44.1
+#########################
+# 3D / VTK / Trame stack#
+#########################
+
+vtk
+pyvista
+
+trame==3.12.0
+trame-vtk==2.10.0
+trame-vuetify==3.1.0
+trame-server==3.6.2
+trame-client==3.10.4
+wslink
+
+#########################
+# Web / misc            #
+#########################
+
 pathlib2>=2.3.0
 typing-extensions>=4.0.0
 transformers>=4.20.0
 trimesh
-gradio_client==1.12.1
 pytest
-h5py
-
+requests
+PyYAML