FROM python:3.9-slim

# Install system packages
RUN apt-get update && apt-get install -y ffmpeg imagemagick

# Install pip packages
RUN pip install cog gradio==3.50.2

# Set the working directory
WORKDIR /src

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

# Set the entry point for Cog
ENTRYPOINT ["cog", "predict"]