1 files changed, 17 insertions, 0 deletions
diff --git a/src/components/OllamaCommandPallete.tsx b/src/components/OllamaCommandPallete.tsx
index d48d392..0ee8097 100644
--- a/src/components/OllamaCommandPallete.tsx
+++ b/src/components/OllamaCommandPallete.tsx
@@ -17,12 +17,15 @@ export function OllamaCommandPallete({ options, theme }: { options: string[], th
     setIsEnterPressed(true);
     switch (selection) {
       case "divide into subtasks":
+        logseq.hideMainUI()
         DivideTaskIntoSubTasks()
         break;
       case "summarize":
+        logseq.hideMainUI()
         summarize()
         break;
       case "convert to flash card":
+        logseq.hideMainUI()
         convertToFlashCard()
         break;
       default:
@@ -30,6 +33,20 @@ export function OllamaCommandPallete({ options, theme }: { options: string[], th
     }
   }
 
+  useEffect(() => {
+    const handleEsc = (e: any) => {
+      if (e.key === 'Escape') {
+        logseq.hideMainUI()
+      }
+    };
+
+    window.addEventListener('keydown', handleEsc);
+
+    return () => {
+      window.removeEventListener('keydown', handleEsc);
+    };
+  }, []);
+
   if (isEnterPressed && (selection !== 'ask ai' && selection !== 'define' && selection !== 'ask with context')) {
     return null
   }