Update Dockerfile

Dockerfile

@@ -1,70 +1,50 @@
-# Stage 1: Build stage
 FROM python:3.10-slim AS builder
 
-# Install build dependencies
+ENV PYTHONUNBUFFERED=1 \
+    PIP_NO_CACHE_DIR=off \
+    PIP_DISABLE_PIP_VERSION_CHECK=on \
+    PIP_DEFAULT_TIMEOUT=100
+
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     libsndfile1 \
     git \
+    build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# Create a non-root user
-RUN useradd -m -u 1000 user
-
-# Switch to non-root user
+RUN useradd -m -s /bin/bash -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
-
-# Set working directory
 WORKDIR /app
 
-# Copy requirements and setup scripts
-COPY --chown=user requirements.txt setup_imagebind.py README.md main.py ./
-
-# Install dependencies into a virtual environment
+COPY --chown=user requirements.txt ./
 RUN python -m venv /app/venv
 ENV PATH="/app/venv/bin:$PATH"
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
+RUN pip install --upgrade pip
+RUN pip install -r requirements.txt
 
-# Run setup script to download ImageBind
+COPY --chown=user setup_imagebind.py README.md ./
 RUN python setup_imagebind.py
+RUN pip install .
 
-# Install ImageBind
-RUN pip install --no-cache-dir .
+FROM python:3.10-slim AS runtime
 
-# Stage 2: Runtime stage
-FROM python:3.10-slim
+ENV PYTHONUNBUFFERED=1 \
+    PATH="/app/venv/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" \
+    APP_USER="user"
 
-# Install runtime dependencies only
 RUN apt-get update && apt-get install -y \
     ffmpeg \
     libsndfile1 \
     && rm -rf /var/lib/apt/lists/*
 
-# Create a non-root user
-RUN useradd -m -u 1000 user
-
-# Switch to non-root user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
-
-# Set working directory
+RUN useradd -m -s /bin/bash -u 1000 ${APP_USER}
+USER ${APP_USER}
 WORKDIR /app
 
-# Copy virtual environment from builder
-COPY --from=builder --chown=user /app/venv /app/venv
-ENV PATH="/app/venv/bin:$PATH"
-
-# Copy ImageBind from builder
-COPY --from=builder --chown=user /app/imagebind* .
-COPY --from=builder --chown=user /app/setup.py .
-COPY --from=builder --chown=user /app/build .
-
-# Copy application code
-COPY --chown=user main.py .
+COPY --from=builder --chown=${APP_USER} /app/venv /app/venv
+COPY --chown=${APP_USER} main.py .
+COPY --chown=${APP_USER} .env .
 
-# Expose the port
 EXPOSE 7860
-
-# Command to run the application
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]