diff --git a/apk-untrusted-toggle.sh b/apk-untrusted-toggle.sh
index dd6439a..40ce7e3 100644
--- a/apk-untrusted-toggle.sh
+++ b/apk-untrusted-toggle.sh
@@ -1,16 +1,34 @@
 #!/bin/sh
 
-PKG_SCRIPT="/usr/libexec/package-manager-call"
+TARGET_SCRIPT="/usr/libexec/package-manager-call"
 
 # 检查文件是否存在
-if [ ! -f "$PKG_SCRIPT" ]; then
-    echo "错误：目标脚本 $PKG_SCRIPT 不存在！"
-    echo "请修改脚本里的 PKG_SCRIPT 变量为真实路径"
+if [ ! -f "$TARGET_SCRIPT" ]; then
+    echo "❌ 错误：目标文件不存在，请修改脚本内的路径！"
     exit 1
 fi
 
-# 自动替换
-sed -i 's/^[[:space:]]*action="add"$/                action="add --allow-untrusted"/' "$PKG_SCRIPT"
+clear
+echo "======================================"
+echo "     APK 安装参数 --allow-untrusted 开关"
+echo "======================================"
+echo " 1 - 开启 （自动添加 --allow-untrusted）"
+echo " 2 - 关闭 （恢复默认，删除参数）"
+echo "======================================"
+read -p "请输入选项 [1/2]：" num
 
-echo "✅ 修改完成！"
-grep -n "add --allow-untrusted" "$PKG_SCRIPT"
\ No newline at end of file
+case $num in
+    1)
+        sed -i 's/^[[:space:]]*action="add"$/                action="add --allow-untrusted"/' "$TARGET_SCRIPT"
+        echo -e "\n✅ 已成功开启：apk add --allow-untrusted"
+        ;;
+    2)
+        sed -i 's/^[[:space:]]*action="add --allow-untrusted"$/                action="add"/' "$TARGET_SCRIPT"
+        echo -e "\n✅ 已成功关闭：恢复为 apk add"
+        ;;
+    *)
+        echo -e "\n❌ 输入错误，请输入 1 或 2"
+        ;;
+esac
+
+grep -n "add" "$TARGET_SCRIPT" | tail -2
\ No newline at end of file