FROM nvidia/cuda:12.1.1-runtime-ubuntu22.04

ENV SHELL=/bin/bash

# System deps
RUN apt-get update && apt-get install -y --no-install-recommends \
    python3 python3-pip python3-venv git curl wget build-essential pkg-config ca-certificates \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /app

# --- Install uv (Python package/dependency manager) ---
RUN curl -LsSf https://astral.sh/uv/install.sh | sh \
    && ln -s $HOME/.cargo/bin/uv /usr/local/bin/uv

# --- Install Rust + Cargo ---
ENV RUSTUP_HOME=/usr/local/rustup \
    CARGO_HOME=/usr/local/cargo \
    PATH=/usr/local/cargo/bin:$PATH
RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y \
    && rustup default stable

# --- Install pnpm (Node package manager) ---
ENV PNPM_HOME=/usr/local/pnpm
ENV PATH=$PNPM_HOME:$PATH
RUN curl -fsSL https://get.pnpm.io/install.sh | sh - \
    && ln -s $PNPM_HOME/pnpm /usr/local/bin/pnpm

# --- Python setup ---
RUN python3 -m pip install --upgrade pip

# Example: clone your external repo early
RUN git clone https://github.com/CarolinePascal/unmute.git /app/unmute

# Copy rest of your app
COPY . /app

# Ports
EXPOSE 7860
ENV PORT=7860

CMD ["/app/app.sh"]