FROM python:3.9-slim

# Install system packages
RUN apt-get update && apt-get install -y \
    ffmpeg \
    imagemagick \
    fontconfig \
    && apt-get clean \
    && rm -rf /var/lib/apt/lists/*

# Install pip packages
RUN pip install gradio==3.50.2

# Set environment variables for matplotlib and fontconfig
ENV MPLCONFIGDIR=/tmp/matplotlib
ENV FONTCONFIG_PATH=/etc/fonts

# Create needed directories with correct permissions
RUN mkdir /tmp/matplotlib

# Set the working directory
WORKDIR /src

# Copy the rest of the application code
COPY . /src

# Ensure fonts directory exists and create the cache
RUN mkdir -p /src/fonts && \
    fc-cache -fv /src/fonts

# Set the entry point to run the prediction script
ENTRYPOINT ["python", "predict.py"]