New Version 2 -- prolly doesn't work.

Dockerfile

@@ -1,12 +1,12 @@
-FROM python:3.11-slim
+# Stegasoo Docker Image
+# Multi-stage build for smaller image size
+
+FROM python:3.11-slim as base
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 
-# Set work directory
-WORKDIR /app
-
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     gcc \
@@ -14,25 +14,47 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libffi-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# Install Python dependencies
+# ============================================================================
+# Builder stage - install Python packages
+# ============================================================================
+FROM base as builder
 
-COPY requirements.txt .
-#COPY requirements-ml.txt .
+WORKDIR /build
 
-RUN pip install --upgrade pip
-RUN pip install --no-cache-dir -r requirements.txt --root-user-action=ignore
-#RUN pip install --no-cache-dir -r requirements-ml.txt
+# Copy package files (including README.md which pyproject.toml references)
+COPY pyproject.toml README.md ./
+COPY src/ src/
+COPY data/ data/
 
-# Copy application
-COPY . .
+# Install the package with web extras
+RUN pip install --no-cache-dir ".[web]"
+
+# ============================================================================
+# Production stage - Web UI
+# ============================================================================
+FROM base as web
+
+WORKDIR /app
+
+# Copy installed packages from builder
+COPY --from=builder /usr/local/lib/python3.11/site-packages /usr/local/lib/python3.11/site-packages
+COPY --from=builder /usr/local/bin /usr/local/bin
+
+# Copy application files
+COPY src/ src/
+COPY data/ data/
+COPY frontends/web/ frontends/web/
 
 # Create upload directory
 RUN mkdir -p /tmp/stego_uploads
 
-# Create non-root user for security
+# Create non-root user
 RUN useradd -m -u 1000 stego && chown -R stego:stego /app /tmp/stego_uploads
 USER stego
 
+# Set Python path
+ENV PYTHONPATH=/app/src
+
 # Expose port
 EXPOSE 5000
 
@@ -40,5 +62,68 @@ EXPOSE 5000
 HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
     CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:5000/')" || exit 1
 
-# Run with gunicorn in production
+# Run with gunicorn
+WORKDIR /app/frontends/web
 CMD ["gunicorn", "--bind", "0.0.0.0:5000", "--workers", "2", "--threads", "4", "--timeout", "60", "app:app"]
+
+# ============================================================================
+# API stage - REST API
+# ============================================================================
+FROM base as api
+
+WORKDIR /app
+
+# Install API extras
+COPY pyproject.toml README.md ./
+COPY src/ src/
+COPY data/ data/
+RUN pip install --no-cache-dir ".[api]"
+
+# Copy API files
+COPY frontends/api/ frontends/api/
+
+# Create non-root user
+RUN useradd -m -u 1000 stego && chown -R stego:stego /app
+USER stego
+
+# Set Python path
+ENV PYTHONPATH=/app/src
+
+# Expose port
+EXPOSE 8000
+
+# Health check
+HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
+    CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:8000/')" || exit 1
+
+# Run with uvicorn
+WORKDIR /app/frontends/api
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+
+# ============================================================================
+# CLI stage - Command line tool
+# ============================================================================
+FROM base as cli
+
+WORKDIR /app
+
+# Install CLI extras
+COPY pyproject.toml README.md ./
+COPY src/ src/
+COPY data/ data/
+RUN pip install --no-cache-dir ".[cli]"
+
+# Copy CLI files
+COPY frontends/cli/ frontends/cli/
+
+# Create non-root user
+RUN useradd -m -u 1000 stego && chown -R stego:stego /app
+USER stego
+
+# Set Python path
+ENV PYTHONPATH=/app/src
+
+# Default to help
+WORKDIR /app/frontends/cli
+ENTRYPOINT ["python", "main.py"]
+CMD ["--help"]