diff --git a/setup.sh b/setup.sh
index a982e9f..ce743e7 100644
--- a/setup.sh
+++ b/setup.sh
@@ -68,6 +68,20 @@ else
 fi
 chmod +x "${APP_DIR}/main.py" "${WALLARM_DIR}/deploy.sh"
 
+# ── Install Docker (for container-based deployment) ──────────────────
+if ! command -v docker >/dev/null 2>&1; then
+    echo -e "${YELLOW}Installing Docker...${NC}"
+    if command -v apt-get >/dev/null 2>&1; then
+        apt-get install -y -qq docker.io 2>/dev/null
+    elif command -v yum >/dev/null 2>&1; then
+        yum install -y -q docker 2>/dev/null
+    elif command -v dnf >/dev/null 2>&1; then
+        dnf install -y -q docker 2>/dev/null
+    fi
+    systemctl enable docker 2>/dev/null || true
+    systemctl start docker 2>/dev/null || true
+fi
+
 echo
 echo -e "${GREEN}${BOLD}Ready!${NC}"
 echo -e "  ${GREEN}sudo /opt/fw/deploy.sh${NC}"