diff --git a/src/views/badge/preview.vue b/src/views/badge/preview.vue
index a5b6074..c715b6d 100644
--- a/src/views/badge/preview.vue
+++ b/src/views/badge/preview.vue
@@ -76,7 +76,7 @@
     <div style="height: 110px;"></div>
     <div class="confirm-box">
       <div class="action-section">
-        <button v-if="trialCode == false" @click="reload" :disabled="flag < 2" class="action-btn"><img class="action-img" src="@/assets/badge/reload.png" alt=""> 再次生成</button>
+        <button v-if="trialCode == true" @click="reload" :disabled="flag < 2" class="action-btn"><img class="action-img" src="@/assets/badge/reload.png" alt=""> 再次生成</button>
         <button v-if="trialCode == false" @click="compare" :disabled="flag < 2" class="action-btn"><img class="action-img" src="@/assets/badge/duibi.png" alt=""> 前后对比</button>
         <button v-if="trialCode == false" @click="save" :disabled="flag < 2" class="action-btn"><img class="action-img" src="@/assets/badge/down.png" alt=""> 保存图片</button>
       </div>
@@ -169,9 +169,11 @@
   </div>
 </template>
 <script setup lang="ts">
-import { onMounted, ref } from 'vue';
-import { useRoutimporter } from 'vue-router';
+import { onMounted, ref, onUnmounted } from 'vue';
+import { useRouter, useRoute } from 'vue-router';
 import { showSuccessToast, showToast, showNotify, showConfirmDialog  } from 'vant';
+import { Encrypt } from '@/utils/encrypt'
+import { getLocalStorage, setLocalStorage } from '@/utils/storage'
 import * as badgeApi from '@/api/badge'
 const router = useRouter();
 const value = ref(0);
@@ -426,7 +428,7 @@ const getImageList = () => {
       imageUrl.value = data.list[0].origin_url
       imageList.value = data.list
       key.value = data.list[0].key
-      localStorage.remove('userId')
+      removeLocalStorage('userId')
     }
   }).catch((err) => {
     clearInterval(progressTimer.value)
@@ -441,15 +443,17 @@ const getImageList = () => {
 }
 const trialCode = ref(false)
 const getTrialCode = () => {
-  const value = localStorage.get('trialCode')
+  const value = getLocalStorage('trialCode')
   if (value) {
     trialCode.value = true
+  } else {
+    getSizeList()
   }
 }
 function handleBeforeUnload(_event: BeforeUnloadEvent) {
-  localStorage.remove('code')
-  localStorage.remove('trialCode')
-  localStorage.remove('userId')
+  removeLocalStorage('code')
+  removeLocalStorage('trialCode')
+  removeLocalStorage('userId')
 }
 
 const pid = ref(0)
@@ -461,13 +465,12 @@ onMounted(() => {
   // group.value = 1;
   getImageList()
   progressList()
-  getSizeList()
   getCompareImage()
   getTrialCode()
-  window.addEventListener('beforeunload', handleBeforeUnload);
   timer.value = setInterval(() => {
     getImageList()
   }, 10000)
+  window.addEventListener('beforeunload', handleBeforeUnload);
 })
 
 onUnmounted(() => {