77 lines
2.2 KiB
Docker
77 lines
2.2 KiB
Docker
# ============================================
|
|
# BraceIQMed Brace Generator - FastAPI + PyTorch (CPU)
|
|
# Build context: repo root (braceiqmed/)
|
|
# ============================================
|
|
|
|
FROM python:3.10-slim
|
|
|
|
# Prevent interactive prompts
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
# Install system dependencies
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
build-essential \
|
|
libgl1 \
|
|
libglib2.0-0 \
|
|
libsm6 \
|
|
libxext6 \
|
|
libxrender-dev \
|
|
wget \
|
|
curl \
|
|
git \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
WORKDIR /app
|
|
|
|
# Install PyTorch CPU version (smaller, no CUDA)
|
|
RUN pip install --no-cache-dir --upgrade pip && \
|
|
pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu
|
|
|
|
# Copy and install requirements
|
|
COPY brace-generator/requirements.txt /app/requirements.txt
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
# Copy scoliovis-api requirements and install
|
|
COPY scoliovis-api/requirements.txt /app/requirements-scoliovis.txt
|
|
RUN pip install --no-cache-dir -r requirements-scoliovis.txt || true
|
|
|
|
# Create brace_generator package structure
|
|
RUN mkdir -p /app/brace_generator
|
|
|
|
# Copy brace-generator code as a package
|
|
COPY brace-generator/*.py /app/brace_generator/
|
|
COPY brace-generator/__init__.py /app/brace_generator/__init__.py
|
|
|
|
# Also keep server_DEV structure for compatibility
|
|
RUN mkdir -p /app/brace_generator/server_DEV
|
|
COPY brace-generator/*.py /app/brace_generator/server_DEV/
|
|
|
|
# Copy scoliovis-api
|
|
COPY scoliovis-api/ /app/scoliovis-api/
|
|
|
|
# Copy templates
|
|
COPY templates/ /app/templates/
|
|
|
|
# Set Python path - include both locations
|
|
ENV PYTHONPATH=/app:/app/scoliovis-api
|
|
|
|
# Environment variables
|
|
ENV HOST=0.0.0.0
|
|
ENV PORT=8002
|
|
ENV DEVICE=cpu
|
|
ENV MODEL=scoliovis
|
|
ENV TEMP_DIR=/tmp/brace_generator
|
|
ENV CORS_ORIGINS=*
|
|
|
|
# Create directories
|
|
RUN mkdir -p /tmp/brace_generator /app/data/uploads /app/data/outputs
|
|
|
|
EXPOSE 8002
|
|
|
|
# Health check
|
|
HEALTHCHECK --interval=30s --timeout=10s --start-period=60s --retries=3 \
|
|
CMD curl -f http://localhost:8002/health || exit 1
|
|
|
|
# Run the server from the brace_generator package
|
|
CMD ["python", "-m", "uvicorn", "brace_generator.server_DEV.app:app", "--host", "0.0.0.0", "--port", "8002"]
|