fix: update Jetson Dockerfile to use correct L4T r36.4.0 base image

- Use dustynv/l4t-pytorch:r36.4.0 for JetPack 6.x compatibility
- Simplify to single-stage build (no multi-stage needed)
- Add Jetson usage hint to run.sh help text

Dockerfile.jetson

@@ -1,8 +1,13 @@
 # Dockerfile for NVIDIA Jetson (ARM64) - Jetson Orin Nano, Xavier, etc.
 # Uses NVIDIA's L4T PyTorch container as base for proper GPU support
+#
+# Check your JetPack/L4T version: cat /etc/nv_tegra_release
+#
+# Available base images:
+#   JetPack 6.x (L4T R36.x): dustynv/l4t-pytorch:r36.4.0
+#   JetPack 5.1.x (L4T R35.x): nvcr.io/nvidia/l4t-pytorch:r35.2.1-pth2.0-py3
 
-# Stage 1: Builder
-FROM nvcr.io/nvidia/l4t-pytorch:r36.2.0-pth2.1-py3 AS builder
+FROM dustynv/l4t-pytorch:r36.4.0
 
 WORKDIR /app
 
@@ -14,63 +19,39 @@ RUN apt-get update && apt-get install -y \
     && apt-get install -y nodejs \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy frontend package files and install dependencies
+# Copy and build frontend
 COPY crossword-app/frontend/package*.json ./frontend/
 RUN cd frontend && npm ci
 
-# Install Python dependencies (PyTorch already included in base image)
+COPY crossword-app/frontend/ ./frontend/
+RUN cd frontend && npm run build
+
+# Install Python dependencies (PyTorch already in base image)
 COPY crossword-app/backend-py/requirements.txt ./backend-py/
-RUN pip install --no-cache-dir -r backend-py/requirements.txt
+RUN pip3 install --no-cache-dir -r backend-py/requirements.txt
 
-# Copy all source code
-COPY crossword-app/frontend/ ./frontend/
+# Copy backend code
 COPY crossword-app/backend-py/ ./backend-py/
 COPY crossword-app/words/ ./backend-py/words/
 
 # Copy cache directory with pre-built models and NLTK data
-COPY cache-dir/ ./cache-dir/
-RUN chmod -R 755 ./cache-dir/ || true
+COPY cache-dir/ ./backend-py/cache/
+RUN chmod -R 755 ./backend-py/cache/ || true
 
-# Build the React frontend
-RUN cd frontend && npm run build
-
-# Copy built frontend files to Python backend public directory
+# Copy built frontend to backend public directory
 RUN mkdir -p backend-py/public && cp -r frontend/dist/* backend-py/public/
 
-# Stage 2: Runtime
-FROM nvcr.io/nvidia/l4t-pytorch:r36.2.0-pth2.1-py3 AS runtime
-
-# Install minimal runtime dependencies
-RUN apt-get update && apt-get install -y \
-    curl \
-    && rm -rf /var/lib/apt/lists/*
-
-# Create non-root user
-RUN useradd -m -u 1000 appuser
-
 WORKDIR /app/backend-py
 
-# Copy Python packages from builder (sentence-transformers, etc.)
-COPY --from=builder /usr/local/lib/python3.10/dist-packages /usr/local/lib/python3.10/dist-packages
-
-# Copy built application files
-COPY --from=builder --chown=appuser:appuser /app/backend-py ./
-
-# Copy cache directory
-COPY --from=builder --chown=appuser:appuser /app/cache-dir /app/backend-py/cache
-
-USER appuser
-
 EXPOSE 7860
 
 ENV NODE_ENV=production
 ENV PORT=7860
 ENV PYTHONPATH=/app/backend-py
 ENV PYTHONUNBUFFERED=1
-ENV PIP_NO_CACHE_DIR=1
 ENV CACHE_DIR=/app/backend-py/cache
 ENV NLTK_DATA=/app/backend-py/cache/nltk_data
 ENV VOCAB_SOURCE=norvig
 ENV NORVIG_VOCAB_PATH=/app/backend-py/words/norvig/count_1w100k.txt
 
-CMD ["python", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+CMD ["python3", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]

run-jetson.sh

@@ -2,19 +2,21 @@
 set -e
 
 # Build and run script for NVIDIA Jetson devices (Orin Nano, Xavier, etc.)
+# Uses Dockerfile.jetson with L4T PyTorch base image
 
 show_usage() {
     echo "Usage: $0 [COMMAND]"
     echo ""
     echo "Commands:"
     echo "  build    - Build the Jetson Docker image"
-    echo "  run      - Run the container with GPU support"
-    echo "  both     - Build and run (default)"
+    echo "  run      - Run the container with GPU support (default)"
+    echo "  both     - Build and run"
     echo "  shell    - Run with bash shell for debugging"
     echo ""
 }
 
-IMAGE_NAME="crossword-py-ai:jetson"
+IMAGE_NAME="crossword-app:jetson"
+
 DOCKER_ARGS="--rm -p 7860:7860 --runtime nvidia \
     -e ENABLE_DEBUG_TAB=true \
     -e VOCAB_SOURCE=norvig \
@@ -36,7 +38,7 @@ run_shell() {
 }
 
 # Parse command
-COMMAND="${1:-both}"
+COMMAND="${1:-run}"
 
 case "$COMMAND" in
     build)
@@ -50,7 +52,6 @@ case "$COMMAND" in
         run_container
         ;;
     shell)
-        build_image
         run_shell
         ;;
     -h|--help|help)

run.sh

@@ -15,6 +15,7 @@ show_usage() {
     echo "  $0 gpu      # Force GPU mode"
     echo "  $0 cpu      # Force CPU-only mode"
     echo ""
+    echo "For Jetson devices, use: ./run-jetson.sh"
 }
 
 # Parse command line arguments
@@ -68,11 +69,11 @@ check_gpu_available() {
     if ! command -v nvidia-smi &> /dev/null; then
         return 1
     fi
-    
+
     if ! docker run --rm --gpus all nvidia/cuda:12.1.0-base-ubuntu22.04 nvidia-smi &> /dev/null; then
         return 1
     fi
-    
+
     return 0
 }
 
@@ -108,4 +109,4 @@ case "$MODE" in
             run_cpu
         fi
         ;;
-esac 
+esac