FROM python:3.9-slim

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

# Install pip packages
RUN pip install gradio==3.50.2

# Set the working directory
WORKDIR /src

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

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