FROM python:3.9-slim

WORKDIR /app

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

COPY requirements.txt requirements.txt
RUN pip install -r requirements.txt

COPY . .

# Set environment variables
ENV TRANSFORMERS_CACHE=/app/cache
ENV MPLCONFIGDIR=/app/matplotlib

# Create necessary directories
RUN mkdir -p /app/cache /app/matplotlib

EXPOSE 5000

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