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

# Ensure cog is in the PATH
RUN ln -s /usr/local/bin/cog /usr/bin/cog

# Set the working directory
WORKDIR /src

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

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