FROM python:3.10-slim

# Install system deps (if needed for torch, etc.)
RUN apt-get update && apt-get install -y \
    gcc \
    && rm -rf /var/lib/apt/lists/*

# Upgrade pip and install Python deps
RUN pip install --no-cache-dir --upgrade pip
RUN pip install --no-cache-dir \
    torch \
    transformers \
    gradio \
    accelerate  # Helps with model loading on GPU/CPU

# Create non-root user
RUN useradd -m -u 1000 user
USER user
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

WORKDIR $HOME/app

# Copy app files
COPY --chown=user . $HOME/app

# Expose port (default for Gradio)
EXPOSE 7860

# Run the app
CMD ["python", "app.py"]