diff --git a/docker-upgrade.sh b/docker-upgrade.sh
index 590118a3..9996bf96 100644
--- a/docker-upgrade.sh
+++ b/docker-upgrade.sh
@@ -1,16 +1,51 @@
 #!/bin/bash
 
-# 停止并移除现有容器
-docker compose down --remove-orphans
+# 初始化标志变量
+PULL_IMAGES=false
+DEMO_MODE=false
 
-if [ "$1" == "demo" ]; then
-    # 设置 DEMO_MODE 环境变量为 true
+# 解析命令行参数
+for arg in "$@"
+do
+    case $arg in
+        --pull)
+        PULL_IMAGES=true
+        shift
+        ;;
+        --demo)
+        DEMO_MODE=true
+        shift
+        ;;
+        *)
+        echo "Unknown argument: $arg"
+        exit 1
+        ;;
+    esac
+done
+
+# 停止并移除现有容器
+echo "Stopping all services..."
+docker compose down --remove-orphans
+echo "Stopped all services..."
+
+# 拉取镜像
+if [ "$PULL_IMAGES" = true ]; then
+    echo "Pulling latest images..."
+    docker compose pull
+    echo "Pulled latest images..."
+fi
+
+if [ "$DEMO_MODE" = true ]; then
+    # 启用 demo 模式
     export DEMO_MODE=true
     echo "Starting services for demo mode..."
+
     # 启动指定的服务
-    docker compose up -d --pull --remove-orphans mysql redis ui service guacd adminer
+    docker compose up -d --remove-orphans mysql redis ui service guacd adminer
+    echo "Started services for demo mode..."
 else
+    # 启动所有服务
     echo "Starting all services..."
-    # 正常启动所有服务
-    docker compose up -d --pull --remove-orphans
+    docker compose up -d --remove-orphans
+    echo "Started all services..."
 fi