FROM ciwei123321/rose:latest

USER root
RUN apt-get update && apt-get install -y socat python3 python3-pip && \
    pip3 install huggingface-hub && \
    rm -rf /var/lib/apt/lists/*

ENV ROSE_ADMIN_PASSWORD=rose123 \
    TZ=Asia/Shanghai

RUN mkdir -p /app/data /app/strm

EXPOSE 7860

CMD ["/bin/sh", "-c", "\
    if [ -n \"$HF_DATASET\" ] && [ -n \"$HF_TOKEN\" ]; then \
        echo '=== Syncing data from Hugging Face Hub ===' && \
        huggingface-cli download $HF_DATASET --repo-type=dataset --local-dir=/app/data --resume-download || true; \
    else \
        echo '=== No HF credentials, skipping data sync ==='; \
    fi; \
    echo '=== Starting Rose service (python3 run.pyc) ===' && \
    cd /app && \
    python3 run.pyc --port 8787 & \
    sleep 5 && \
    echo '=== Starting port forwarding 7860 -> 8787 ===' && \
    exec socat TCP-LISTEN:7860,fork,reuseaddr TCP:127.0.0.1:8787 \
"]