Update Dockerfile

Dockerfile

@@ -1,13 +1,34 @@
+# Hugging Face Docker Space for Draw Things gRPC server (CUDA)
+# Uses the official prebuilt image; adds minimal glue for HF ($PORT) and model preloading.
+
 FROM drawthingsai/draw-things-grpc-server-cli:latest
 
-# adresář pro modely (můžeš je později nahrát do /grpc-models nebo stáhnout při startu)
-RUN mkdir -p /grpc-models
+# System deps for port forwarding + downloads
+USER root
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+      ca-certificates curl wget git jq unzip xz-utils tar socat && \
+    rm -rf /var/lib/apt/lists/*
+
+# Workdir & models dir
+WORKDIR /srv
+ENV MODELS_DIR=/grpc-models
+RUN mkdir -p ${MODELS_DIR}
+
+# Copy entrypoint scripts
+COPY start.sh /srv/start.sh
+COPY model_download.sh /srv/model_download.sh
+RUN chmod +x /srv/*.sh
+
+# --- Runtime env knobs ---
+# Comma or newline separated list of model URLs to fetch on startup (optional)
+ENV MODEL_URLS=""
+# Location to listen internally (gRPCServerCLI default is 7859)
+ENV GRPC_PORT=7859
 
-# nainstalujeme socat na forward $PORT -> 7859 (pokud nejde změnit port gRPC serveru)
-RUN apt-get update && apt-get install -y socat && rm -rf /var/lib/apt/lists/*
+# Hugging Face will inject $PORT for the public ingress. We forward $PORT -> 127.0.0.1:${GRPC_PORT}
+# Expose is informational for Docker; HF maps automatically
+EXPOSE 7860 7859
 
-# start: spustíme gRPC server + forwarder
-# HF předá cílový port v proměnné $PORT
-CMD bash -lc '\
-  gRPCServerCLI /grpc-models & \
-  socat TCP-LISTEN:${PORT},fork TCP:127.0.0.1:7859'
+# Start everything
+CMD ["/srv/start.sh"]