FROM runpod/pytorch:2.1.0-py3.10-cuda11.8.0

WORKDIR /workspace

# Install system dependencies
RUN apt-get update && apt-get install -y \
    libgl1-mesa-glx \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

# Copy application code
COPY . .

# Install the package in editable mode
RUN pip install --no-cache-dir -e .

# Create directories for models and temporary files
RUN mkdir -p /model-weights /workspace/temp

# Download models (if needed)
# RUN python download_models.py

EXPOSE 8000

CMD ["python", "handler.py"]