aboutsummaryrefslogtreecommitdiff
path: root/src/components/OllamaCommandPallete.tsx
blob: d7218920c1e4cbd10483728ec0fe6350cf91961d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import React, { useEffect, useState } from "react"
import {
  Command,
  CommandEmpty,
  CommandInput,
  CommandItem,
  CommandList,
} from "@/components/ui/command"
import { convertToFlashCardCurrentBlock, DivideTaskIntoSubTasksCurrentBlock, summarize, summarizeBlock, customPromptBlock } from "@/ollama";
import { PromptAI } from "./PromptAI";

export function OllamaCommandPallete({ options, theme }: { options: string[], theme: string }) {
  const [selection, setSelection] = useState('')
  const [isEnterPressed, setIsEnterPressed] = useState(false);
  const handleSelection = (selection: string) => {
    setSelection(selection)
    setIsEnterPressed(true);
    switch (selection) {
      case "divide into subtasks":
        logseq.hideMainUI()
        DivideTaskIntoSubTasksCurrentBlock()
        break;
      case "summarize":
        logseq.hideMainUI()
        summarize()
        break;
      case "summarize block":
        logseq.hideMainUI()
        summarizeBlock()
        break;
      case "custom prompt on block":
        logseq.hideMainUI()
        customPromptBlock()
        break;
      case "convert to flash card":
        logseq.hideMainUI()
        convertToFlashCardCurrentBlock()
        break;
      default:
        break;
    }
  }

  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
  }

  return (
    selection === 'ask with context' || selection === 'ask ai' || selection === 'define' ? (<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>
          <CommandEmpty>No results found.</CommandEmpty>
          {
            options.map((option) => (
              <CommandItem key={option} onSelect={handleSelection} className="text-lg">
                <span>{option}</span>
              </CommandItem>
            ))
          }
        </CommandList>
      </Command>
    )
  )
}