Update Dockerfile

Dockerfile

@@ -5,7 +5,8 @@ FROM python:3.10-slim
 WORKDIR /app
 
 # 1. Install System Dependencies
-# ADDED: 'cmake' is strictly required for llama-cpp-python if building from source.
+# ADDED: 'cmake' is strictly required. 
+# We also keep 'build-essential' for other potential compilation needs.
 RUN apt-get update && apt-get install -y \
     git \
     curl \
@@ -17,8 +18,9 @@ RUN apt-get update && apt-get install -y \
 RUN curl -L -o /app/VT323.ttf https://github.com/google/fonts/raw/main/ofl/vt323/VT323-Regular.ttf
 
 # 3. Install Python Dependencies
-# CHANGED: Split llama-cpp-python to use pre-built wheels.
-# This avoids the "Building wheel..." hang by downloading a compiled binary instead.
+# CHANGED: Split installation into two steps to prevent hangs.
+
+# Step A: Install standard libraries
 RUN pip install --no-cache-dir \
     torch \
     torchvision \
@@ -32,8 +34,11 @@ RUN pip install --no-cache-dir \
     pillow \
     diskcache \
     safetensors \
-    scipy \
-    && pip install --no-cache-dir llama-cpp-python \
+    scipy
+
+# Step B: Install llama-cpp-python using PRE-BUILT WHEELS.
+# This avoids the "Building wheel..." infinite loop.
+RUN pip install llama-cpp-python \
     --extra-index-url https://abetlen.github.io/llama-cpp-python/whl/cpu
 
 # 4. Create a non-root user
@@ -57,7 +62,7 @@ from PIL import Image, ImageDraw
 from llama_cpp import Llama
 
 # ============================================================================
-# 1. FRONTEND ASSET (index.html embedded)
+# 1. FRONTEND ASSET
 # ============================================================================
 
 HTML_TEMPLATE = r"""