2 files changed, 13 insertions, 9 deletions
diff --git a/src/components/OllamaCommandPallete.tsx b/src/components/OllamaCommandPallete.tsx
index e8e9c42..dabdc9b 100644
--- a/src/components/OllamaCommandPallete.tsx
+++ b/src/components/OllamaCommandPallete.tsx
@@ -6,7 +6,7 @@ import {
   CommandItem,
   CommandList,
 } from "@/components/ui/command"
-import { convertToFlashCardCurrentBlock, DivideTaskIntoSubTasksCurrentBlock, summarize, summarizeBlock } from "@/ollama";
+import { convertToFlashCardCurrentBlock, DivideTaskIntoSubTasksCurrentBlock, summarizePage, summarizeBlock } from "@/ollama";
 import { PromptAI } from "./PromptAI";
 
 export function OllamaCommandPallete({ options, theme }: { options: string[], theme: string }) {
@@ -20,9 +20,9 @@ export function OllamaCommandPallete({ options, theme }: { options: string[], th
         logseq.hideMainUI()
         DivideTaskIntoSubTasksCurrentBlock()
         break;
-      case "summarize":
+      case "summarize page":
         logseq.hideMainUI()
-        summarize()
+        summarizePage()
         break;
       case "summarize block":
         logseq.hideMainUI()
@@ -51,12 +51,14 @@ export function OllamaCommandPallete({ options, theme }: { options: string[], th
     };
   }, []);
 
-  if (isEnterPressed && (selection !== 'ask ai' && selection !== 'define' && selection !== 'ask with context')) {
-    return null
+  const validSelections = ['ask with page context', 'ask with block context', 'ask ai', 'define', 'ask with context'];
+
+  if (isEnterPressed && !validSelections.includes(selection)) {
+    return null;
   }
 
   return (
-    selection === 'ask with context' || selection === 'ask ai' || selection === 'define' ? (<PromptAI theme={theme} type={selection} />) : (
+    validSelections.includes(selection) ? (<PromptAI theme={theme} type={selection} />) : (
       <Command className={(theme === 'dark' ? "dark dark:bg-gray-900" : "bg-gray-200") + " rounded-lg border shadow-md w-1/2"}>
         <CommandInput className="ai-input" placeholder="Type a command or search..." />
         <CommandList>
diff --git a/src/components/PromptAI.tsx b/src/components/PromptAI.tsx
index 5fb9942..6b4616d 100644
--- a/src/components/PromptAI.tsx
+++ b/src/components/PromptAI.tsx
@@ -4,7 +4,7 @@ import { Input } from '@/components/ui/input';
 
 export const PromptAI = ({ type, theme }: { type: string, theme: string }) => {
 
-  const placeholder = type === 'ask ai' ? "Prompt..." : "Define..."
+  const placeholder = type.startsWith('ask') ? "Prompt..." : "Define..."
   const [inputValue, setInputValue] = useState('');
   const [hitEnter, setHitEnter] = useState(false)
 
@@ -15,8 +15,10 @@ export const PromptAI = ({ type, theme }: { type: string, theme: string }) => {
         askAI(inputValue, "")
       } else if (type === 'define') {
         defineWord(inputValue)
-      } else if (type === 'ask with context') {
-        askWithContext(inputValue)
+      } else if (type === 'ask with page context') {
+        askWithContext(inputValue, 'page')
+      } else if (type === 'ask with block context') {
+        askWithContext(inputValue, 'block')
       }
     }
   }, [hitEnter])