[ViPPET] Add model-download service to Docker Compose stack (open-edge-platform#2092)

Author: p-zak

tools/visual-pipeline-and-platform-evaluation-tool/Makefile

@@ -6,7 +6,7 @@ SHELL := bash -eu -o pipefail
 # default goal to show help
 .DEFAULT_GOAL := help
 .PHONY: all lint build mdlint ruff fix-linter format pyright run build-dev run-dev test \
-	shell shell-vippet shell-models shell-collector shell-videogenerator stop clean clean-output-videos help \
+	shell shell-vippet shell-models shell-model-download shell-collector shell-videogenerator stop clean clean-output-videos help \
 	build-videogenerator build-models build-onvif-discovery docker-build env-setup \
 	install-models-once install-models-force install-models-all \
 	test-smoke test-full
@@ -33,7 +33,7 @@ all: lint build run test ## Run lint, build, run and test
 
 lint: $(VENV_DIR) mdlint ruff pyright ## Run all linters
 
-MD_FILES := $(shell find . -type f \( -name '*.md' \) -not -path './.*' -print )
+MD_FILES := $(shell find . -type f \( -name '*.md' \) -not -path './.*' -not -path './shared/*' -print )
 mdlint: ## Lint MD files
 # download tool from https://github.com/igorshubovych/markdownlint-cli
 	markdownlint --version
@@ -42,19 +42,19 @@ mdlint: ## Lint MD files
 ruff: $(VENV_DIR) ## Run ruff linter
 	set +u; . ./$</bin/activate; set -u ;\
 	ruff --version ;\
-	# Exclude shared/scripts/ to avoid formatting user created scripts ;\
-	ruff check --exclude shared/scripts/ ;\
-	ruff format --check --exclude shared/scripts/ 
+	# Exclude shared/ — runtime-mounted volume, not source code ;\
+	ruff check --exclude shared/ ;\
+	ruff format --check --exclude shared/
 
 fix-linter: $(VENV_DIR) ## Fix linter issues using ruff
 	set +u; . ./$</bin/activate; set -u ;\
-	# Exclude shared/scripts/ to avoid formatting user created scripts ;\
-	ruff check --fix --exclude shared/scripts/ 
+	# Exclude shared/ — runtime-mounted volume, not source code ;\
+	ruff check --fix --exclude shared/
 
 format: $(VENV_DIR) ## Format code using ruff
 	set +u; . ./$</bin/activate; set -u ;\
-	# Exclude shared/scripts/ to avoid formatting user created scripts ;\
-	ruff format --exclude shared/scripts/ 
+	# Exclude shared/ — runtime-mounted volume, not source code ;\
+	ruff format --exclude shared/
 
 pyright: $(VENV_DIR) ## Run pyright type checker
 	set +u; . ./$</bin/activate; set -u ;\
@@ -69,6 +69,8 @@ generate_openapi: $(VENV_DIR) ## Generate OpenAPI schema file
 env-setup: ## Environment setup target: always run before build/run targets that need .env and shared dirs
 	mkdir -p shared/collector-signals && chmod o+w shared/collector-signals
 	mkdir -p shared/models/output && chmod o+w shared/models/output
+	mkdir -p shared/model-download/venv && chmod o+w shared/model-download/venv
+	mkdir -p shared/model-download/cache && chmod o+w shared/model-download/cache
 	mkdir -p shared/videos/input && chmod o+w shared/videos/input
 	mkdir -p shared/videos/output && chmod o+w shared/videos/output
 	mkdir -p shared/videos/video-generator && chmod o+w shared/videos/video-generator
@@ -138,17 +140,20 @@ shell-vippet: ## Open shell in vippet container
 shell-models: ## Open shell in models container
 	$(MAKE) shell SERVICE=models
 
+shell-model-download: ## Open shell in model-download container
+	$(MAKE) shell SERVICE=model-download
+
 shell-collector: ## Open shell in collector container
 	$(MAKE) shell SERVICE=collector
 
 shell-videogenerator: ## Open shell in videogenerator container
 	$(MAKE) shell SERVICE=videogenerator
 
 stop: ## Stop the docker compose services
-	docker compose down models collector videogenerator mediamtx vippet-ui vippet onvif-discovery
+	docker compose down models model-download collector videogenerator mediamtx vippet-ui vippet onvif-discovery
 
 clean: ## Clean all build artifacts
-	rm -rf shared/collector-signals shared/models/output/ shared/onvif/onvif_cameras.json shared/videos/input shared/videos/output shared/videos/video-generator
+	rm -rf shared/collector-signals shared/models/output/ shared/model-download/ shared/onvif/onvif_cameras.json shared/videos/input shared/videos/output shared/videos/video-generator
 
 clean-output-videos: ## Clean only output videos
 	rm -rf shared/videos/output/*

tools/visual-pipeline-and-platform-evaluation-tool/compose.yml

@@ -101,6 +101,36 @@ services:
       - https_proxy=${https_proxy}
       - no_proxy=${no_proxy}
     command: './model_manager.sh'
+  model-download:
+    image: docker.io/intel/model-download:2026.1.0-20260331-weekly
+    container_name: model-download
+    command: --plugins ${MODEL_DOWNLOAD_PLUGINS:-openvino,huggingface,ultralytics,geti}
+    healthcheck:
+      test: ["CMD", "curl", "--fail", "--silent", "--show-error", "--output", "/dev/null", "http://localhost:8000/api/v1/health"]
+      interval: 30s
+      timeout: 10s
+      retries: 5
+      start_period: 60s
+      start_interval: 5s
+    restart: on-failure:5
+    environment:
+      MODELS_DIR: /models/output
+      CORS_ALLOW_ORIGINS: "*"
+      HF_HUB_ENABLE_HF_TRANSFER: "1"
+      HF_TOKEN: ${HF_TOKEN:-}
+      UV_CACHE_DIR: /cache/uv
+      GETI_HOST: ${GETI_HOST:-}
+      GETI_TOKEN: ${GETI_TOKEN:-}
+      GETI_WORKSPACE_ID: ${GETI_WORKSPACE_ID:-}
+      GETI_SERVER_API_VERSION: ${GETI_SERVER_API_VERSION:-v1}
+      GETI_SERVER_SSL_VERIFY: ${GETI_SERVER_SSL_VERIFY:-False}
+      http_proxy: ${http_proxy}
+      https_proxy: ${https_proxy}
+      no_proxy: ${no_proxy}
+    volumes:
+      - ./shared/models/:/models
+      - ./shared/model-download/venv:/opt/.venv
+      - ./shared/model-download/cache:/cache
   vippet:  # Always 'vippet', hardware config from override files
     <<: *vippet
     container_name: vippet

tools/visual-pipeline-and-platform-evaluation-tool/pyrightconfig.json

@@ -16,6 +16,7 @@
     "**/node_modules",
     "**/__pycache__",
     ".git",
-    "shared/scripts"
-  ],
+    "shared/scripts",
+    "shared/model-download"
+  ]
 }

tools/visual-pipeline-and-platform-evaluation-tool/shared/model-download/.gitkeep

@@ -24,6 +24,24 @@ server {
         proxy_read_timeout 30;
     }
 
+    # Reverse proxy for model-download service API
+    location /model-download/ {
+        proxy_pass http://model-download:8000/api/v1/;
+        proxy_set_header Host $host;
+        proxy_set_header X-Real-IP $remote_addr;
+        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
+        proxy_set_header X-Forwarded-Proto $scheme;
+        proxy_set_header X-Forwarded-Host $host:$server_port;
+
+        # Timeout settings (model downloads can take a long time)
+        proxy_connect_timeout 30;
+        proxy_send_timeout 300;
+        proxy_read_timeout 300;
+
+        # Allow large model uploads
+        client_max_body_size 5G;
+    }
+
     # WebSocket proxy for metrics
     location /metrics/ws/ {
         proxy_pass http://vippet:7860;

tools/visual-pipeline-and-platform-evaluation-tool/ui/nginx.conf

@@ -28,6 +28,13 @@ export default defineConfig(({ mode }) => {
           secure: false,
           ws: true,
         },
+        "/model-download": {
+          target: env.VITE_MODEL_DOWNLOAD_URL || "http://localhost:8000",
+          changeOrigin: true,
+          secure: false,
+          ws: false,
+          rewrite: (path: string) => path.replace(/^\/model-download/, "/api/v1"),
+        },
         "/api": {
           target: env.VITE_API_URL || "http://localhost:7860",
           changeOrigin: true,