FROM python:3.9-slim

WORKDIR /code

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

# Copy only requirements to leverage Docker caching
COPY ./requirements.txt /code/requirements.txt

# Install Python dependencies
RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

# Copy all code and data
COPY . /code/

# Create necessary directories
RUN mkdir -p /code/models
RUN mkdir -p /code/stimuli

# Make sure stimuli and models are writable
RUN chmod -R 777 /code/models
RUN chmod -R 777 /code/stimuli

# Set up the command to run the app
CMD ["python", "app.py"]