diff --git a/src/main.ts b/src/main.ts
index 4070232..876fbf9 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -55,7 +55,7 @@ import { notifyNtfy } from './api';
 import t from './i18n';
 import './main.scss';
 import { NotifyUtil } from './utils/notify';
-// import { editorUtil } from './utils/editor';
+import { editorUtil } from './utils/editor';
 
 export const OpenUrl = ref('https://baidu.com');
 const media = window.matchMedia('(prefers-color-scheme: dark)');
@@ -318,7 +318,7 @@ export default class AwesomeBrainManagerPlugin extends Plugin {
 
     override async onload(): Promise<void> {
         await this.pluginDataIO.load();
-        // editorUtil.init(this);
+        editorUtil.init(this);
         NotifyUtil.init(this);
         this.setupUI();
         this.setupCommands();
@@ -508,14 +508,6 @@ export default class AwesomeBrainManagerPlugin extends Plugin {
             },
         });
 
-		this.addCommand({
-            id: 'query-baidu',
-            name: t.command['query-baidu'] || '百度一下',
-            editorCallback: (editor: Editor, view: MarkdownView) => {
-                this.openBrowser(`https://baidu.com/s?wd=${editor.getSelection()}`)
-            },
-        });
-
         this.addCommand({
             id: 'query-openai',
             name: t.command['query-openai'],
@@ -591,7 +583,7 @@ export default class AwesomeBrainManagerPlugin extends Plugin {
         this.registerView(POMODORO_HISTORY_VIEW, leaf => new PomodoroHistoryView(leaf, this));
         this.registerView(BROWSER_VIEW, leaf => new BrowserView(leaf, this, OpenUrl));
 
-        // editorUtil.addTags(JSON.parse(SETTINGS.customTag.value));
+        editorUtil.addTags(JSON.parse(SETTINGS.customTag.value));
 
         // 左侧菜单，使用自定义图标
         this.addRibbonIcon('settings-2', 'Awesome Brain Manager', event => {
@@ -629,7 +621,7 @@ export default class AwesomeBrainManagerPlugin extends Plugin {
         // Remove listener when we unload
         this.register(() => media.removeEventListener('change', callback));
         this.registerDomEvent(activeDocument, 'mouseup', async (e: MouseEvent) => {
-            // editorUtil.changeToolbarPopover(e, SETTINGS.toolbar);
+            editorUtil.changeToolbarPopover(e, SETTINGS.toolbar);
         });
         this.registerDomEvent(activeDocument, 'click', async (e: MouseEvent) => {
             toggleMouseClickEffects(e, SETTINGS.clickString);
@@ -653,7 +645,7 @@ export default class AwesomeBrainManagerPlugin extends Plugin {
     }
 
     override async onunload(): Promise<void> {
-        // editorUtil.unloadCustomViewContainer();
+        editorUtil.unloadCustomViewContainer();
         toggleBlast('0');
         this.app.workspace.detachLeavesOfType(POMODORO_HISTORY_VIEW);
         this.style.detach();
diff --git a/src/settings.ts b/src/settings.ts
index e68ffb0..0452724 100644
--- a/src/settings.ts
+++ b/src/settings.ts
@@ -6,7 +6,7 @@ import type { PluginDataIO } from './data';
 import { toggleCursorEffects } from './render/CursorEffects';
 import { toggleBlast } from './render/Blast';
 import t from './i18n';
-// import { editorUtil } from './utils/editor';
+import { editorUtil } from './utils/editor';
 
 class Settings {
     settings: SettingTabModel = new SettingTabModel();
@@ -105,7 +105,7 @@ class Settings {
             .desc('Customized label configuration')
             .text(JSON.stringify(defaultTag))
             .onAnyValueChanged(context => {
-                // editorUtil.addTags(JSON.parse(SETTINGS.customTag.value));
+                editorUtil.addTags(JSON.parse(SETTINGS.customTag.value));
             })
             .build(new RawSerde());
 
diff --git a/src/stores/editor.ts b/src/stores/editor.ts
index 9446218..543950c 100644
--- a/src/stores/editor.ts
+++ b/src/stores/editor.ts
@@ -1,8 +1,9 @@
 import { defineStore } from 'pinia';
+
 import { reactive, ref } from 'vue';
 import type { EditorState } from '../types/types';
 
-export const useEditorStore = defineStore('pomodoro', () => {
+export const useEditorStore = defineStore('editor', () => {
     const totalTask = ref(0);
 
     const editorState: EditorState = reactive({
diff --git a/src/stores/index.ts b/src/stores/index.ts
index cf8b326..518ab97 100644
--- a/src/stores/index.ts
+++ b/src/stores/index.ts
@@ -1,11 +1,4 @@
-import { createApp } from 'vue';
 import { createPinia } from 'pinia';
-import { usePomodoroStore } from '@/stores/pomodoro';
-import { useEditorStore } from '@/stores/editor';
-
-import CustomViewContainer from '../ui/CustomViewContainer.vue';
-
-const customViewVueApp = createApp(CustomViewContainer);
-
-customViewVueApp.use(createPinia())
-export { usePomodoroStore, useEditorStore, customViewVueApp };
+import { useEditorStore } from './editor';
+export default createPinia();
+export { useEditorStore };
diff --git a/src/ui/Toolbar.vue b/src/ui/Toolbar.vue
index f22400f..91b7263 100644
--- a/src/ui/Toolbar.vue
+++ b/src/ui/Toolbar.vue
@@ -59,7 +59,7 @@
 </template>
 
 <script setup lang="ts">
-import { onMounted, onUnmounted, ref, onUpdated, toRefs, watchEffect } from 'vue';
+import { onMounted, onUnmounted, ref, onUpdated, watchEffect } from 'vue';
 import { NTooltip, NIcon, useNotification } from 'naive-ui';
 import { ServiceNames, chatWith } from '../api';
 import OpenAI from './components/icon/OpenAI.vue';
@@ -72,12 +72,10 @@ import { useEditorStore } from '../stores';
 import { storeToRefs } from 'pinia';
 
 import { customTitle, customContent, customAvatar, customDescription } from './CustomContent';
-import type AwesomeBrainManagerPlugin from '../main';
 import { eventTypes } from '../types/types';
 import Logger from '../utils/logger';
 
-const store = useEditorStore();
-const { editorState: currentState } = storeToRefs(store);
+const { editorState: currentState } = storeToRefs(useEditorStore());
 const isShow = ref(false);
 const notification = useNotification();
 let oldSelection = '';
@@ -125,22 +123,20 @@ const clickHandle = async (type: string, keyword: string) => {
             conversation(type, await chatWith(type, keyword));
             break;
         case ServiceNames.Baidu:
-			const baiduSearchEvent = new CustomEvent(eventTypes.openBrowser, {
-				detail: {
-					url: `https://baidu.com/s?wd=${keyword}`
-				}
-			});
-			console.log('ererer');
-			
-			window.dispatchEvent(baiduSearchEvent);
+            const baiduSearchEvent = new CustomEvent(eventTypes.openBrowser, {
+                detail: {
+                    url: `https://baidu.com/s?wd=${keyword}`,
+                },
+            });
+            window.dispatchEvent(baiduSearchEvent);
             break;
         case ServiceNames.Google:
-			const googleSearchEvent = new CustomEvent(eventTypes.openBrowser, {
-				detail: {
-					url: `https://www.google.com/search?q=${keyword}`
-				}
-			});
-			window.dispatchEvent(googleSearchEvent);
+            const googleSearchEvent = new CustomEvent(eventTypes.openBrowser, {
+                detail: {
+                    url: `https://www.google.com/search?q=${keyword}`,
+                },
+            });
+            window.dispatchEvent(googleSearchEvent);
             break;
         default:
             return;
diff --git a/src/utils/editor.ts b/src/utils/editor.ts
index 11f7a31..6620f03 100644
--- a/src/utils/editor.ts
+++ b/src/utils/editor.ts
@@ -1,17 +1,22 @@
 import type { Editor } from 'obsidian';
+import { createApp, type App } from 'vue';
 import type AwesomeBrainManagerPlugin from '../main';
 import type { SettingModel } from 'model/settings';
-import { customViewVueApp, useEditorStore } from '@/stores';
+import CustomViewContainer from '../ui/CustomViewContainer.vue';
 import { buildTagRules } from '../render/Tag';
 import { Tag, type ExtApp } from '@/types/types';
+import pinia, { useEditorStore } from '@/stores'
 
 export const elId = 'custom-view-container';
 
+
+
 export class EditorUtils {
     plugin: AwesomeBrainManagerPlugin;
     app: ExtApp;
     ele: HTMLDivElement;
     loaded: boolean = false;
+	customViewVueApp: App;
 
     constructor() {}
 
@@ -26,7 +31,9 @@ export class EditorUtils {
                 id: elId,
             },
         });
-        customViewVueApp.mount(`#${elId}`);
+		this.customViewVueApp = createApp(CustomViewContainer);
+		this.customViewVueApp.use(pinia)
+        this.customViewVueApp.mount(`#${elId}`);
         this.loaded = true;
     }
 
diff --git a/vite.config.ts b/vite.config.ts
index 98d3664..81d85d1 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -75,6 +75,9 @@ export default defineConfig({
             ],
         },
     },
+    define: {
+        'process.env.NODE_ENV': '"production"'
+    },
     plugins: [
         viteStaticCopy({
             targets: [