From 481ae0c24785a1798a0445a2a982acce6db7e4f2 Mon Sep 17 00:00:00 2001
From: omagdy7 <omar.professional8777@gmail.com>
Date: Thu, 22 Feb 2024 19:12:22 +0200
Subject: Added the ability to Ask in context of the block only instead of the
 whole page

---
 src/components/OllamaCommandPallete.tsx | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

(limited to 'src/components/OllamaCommandPallete.tsx')

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>
-- 
cgit v1.2.3

