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

ENV DEBIAN_FRONTEND=noninteractive
ENV PYTHONUNBUFFERED=1

RUN apt-get update && apt-get install -y \
    python3.11 \
    python3.11-venv \
    python3-pip \
    git \
    wget \
    curl \
    ffmpeg \
    libsm6 \
    libxext6 \
    libxrender-dev \
    libgomp1 \
    build-essential \
    && rm -rf /var/lib/apt/lists/*

RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.11 1 && \
    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1

RUN git clone --branch main https://github.com/ack00gar/FunGen-AI-Powered-Funscript-Generator.git FunGen && \
    cd FunGen && \
    git config user.email "bot@example.com" && \
    git config user.name "Bot"

WORKDIR /FunGen

RUN pip install -r cuda.requirements.txt

RUN pip install tensorrt torch-tensorrt --extra-index-url https://download.pytorch.org/whl/nightly/cu128 || true

RUN pip install gradio==4.16.0 huggingface-hub==0.19.3 pydantic>=2.0

RUN mkdir -p /tmp/uploads /tmp/outputs /app/models

WORKDIR /app

COPY app.py /app/app.py

ENV FUNGEN_SKIP_CHECK=1

CMD ["python", "/app/app.py"]