FROM ubuntu:22.04
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && \
    apt-get install -y python3.11 python3-pip python3.11-venv curl && \
    apt-get clean && rm -rf /var/lib/apt/lists/*

WORKDIR /opt/bit_transformer
COPY . .

ARG TORCH_CUDA=cpu
RUN pip3 install --no-cache-dir --upgrade pip && \
    if [ "$TORCH_CUDA" = "cu118" ]; then \
        pip3 install torch==2.7.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118; \
    else \
        pip3 install torch==2.7.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu; \
    fi && \
    pip3 install -r requirements.txt

ENV MCP_SERVER_ADDR=http://127.0.0.1:7000

EXPOSE 5000 7000

RUN chmod +x start.sh

HEALTHCHECK CMD curl -f http://localhost:7000/health || exit 1

CMD ["/opt/bit_transformer/start.sh"]