FROM python:3.10-slim

# Install system dependencies for plotting, code execution, etc.
RUN apt-get update && apt-get install -y \
    git \
    build-essential \
    curl \
    ffmpeg \
    libsm6 \
    libxext6 \
    vim \
    less \
    wget \
    python3-pip \
    && rm -rf /var/lib/apt/lists/*

# Upgrade pip and install AutoGen with code execution support
RUN pip install --upgrade pip && \
    pip install \
    pyautogen[codeexec] \
    openai \
    jupyter \
    ipython \
    matplotlib

# Create a working directory (optional)
WORKDIR /workspace

# Set bash as the default shell for interactive use
CMD ["bash"]