# Use Python base image
FROM python:3.10-slim

# Set working directory
WORKDIR /app

# Install system dependencies
RUN apt-get update && apt-get install -y \
    git \
    build-essential \
    libgl1-mesa-glx \
    libglib2.0-0 \
    && rm -rf /var/lib/apt/lists/*

# Install Python dependencies
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt

# Clone and install Detectron2
RUN pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
RUN git clone https://github.com/facebookresearch/detectron2.git && \
    pip install -e detectron2

# Copy your app code
COPY . .

# Expose Streamlit default port
EXPOSE 7860

# Streamlit Environment Variables
ENV STREAMLIT_SERVER_PORT=7860
ENV STREAMLIT_SERVER_HEADLESS=true
ENV STREAMLIT_SERVER_ENABLECORS=false

# Command to run the app
CMD ["streamlit", "run", "app.py"]