(null);
75 | const genAI = new GoogleGenerativeAI(geminiApiKey as string);
76 | const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
77 |
78 | const editor = useEditor({
79 | extensions,
80 | content: initialResponse,
81 | onUpdate: ({ editor }) => {
82 | editor.commands.setContent(initialResponse);
83 | },
84 | });
85 |
86 | const handleSelectNode = () => {
87 | const { state } = editor!;
88 | const { from, to } = state.selection;
89 | const selectedNode = state.doc.textBetween(from, to, " ");
90 | setSelectedNode(selectedNode);
91 | };
92 |
93 | const handlePrompt = async (
94 | promptType: keyof typeof PROMPT_TYPES | "Custom"
95 | ) => {
96 | let prompt;
97 | if (promptType === "Custom") {
98 | prompt = `This is the whole response: ${initialResponse}. ${inputRef.current?.value} Specifically focus on this part: "${selectedNode}". Ensure the modified part aligns seamlessly with the rest of the response. Provide the entire modified response back, preserving the essential introductory and concluding phrases without adding any new non-contextual information.`;
99 | } else {
100 | const promptInstructions = {
101 | Longer: "Lengthen",
102 | Shorter: "Shorten",
103 | Regenerate: "Regenerate",
104 | Remove: "Remove",
105 | Simplify: "Simplify the language of",
106 | Elaborate: "Elaborate on",
107 | Formalize: "Rewrite in a more formal tone",
108 | Casual: "Rewrite in a more casual tone",
109 | Persuasive: "Rewrite to be more persuasive",
110 | Technical: "Add more technical details to",
111 | Metaphor: "Incorporate a relevant metaphor into",
112 | Examples: "Add relevant examples to",
113 | Counterargument: "Present a counterargument to",
114 | Summary: "Provide a concise summary of",
115 | };
116 | prompt = `This is the whole response: ${initialResponse}. ${promptInstructions[promptType]} a specific part of the response, specifically "${selectedNode}". Ensure it aligns seamlessly with the rest of the response. Provide the entire modified response back, preserving the essential introductory and concluding phrases without adding any new non-contextual information.`;
117 | }
118 |
119 | try {
120 | setUpdateLoader(true);
121 | const result = await model.generateContent(prompt);
122 | const response = await result.response;
123 | const text = response.text();
124 | if (!text) throw new Error("Error while generating prompt");
125 | const updatedContent = await updateResponse({
126 | chatUniqueId,
127 | updatedResponse: text,
128 | });
129 | setInitialResponse(updatedContent.message.message.llmResponse as string);
130 | editor?.commands.setContent(
131 | updatedContent.message.message.llmResponse as string
132 | );
133 | setDropdown(false);
134 | } catch (error) {
135 | console.error("Error generating response:", error);
136 | } finally {
137 | setUpdateLoader(false);
138 | }
139 | };
140 |
141 | useEffect(() => {
142 | const handleClickOutside = (event: MouseEvent) => {
143 | if (
144 | dropdownRef.current &&
145 | !dropdownRef.current.contains(event.target as Node)
146 | ) {
147 | setDropdown(false);
148 | }
149 | };
150 |
151 | document.addEventListener("mousedown", handleClickOutside);
152 | return () => document.removeEventListener("mousedown", handleClickOutside);
153 | }, []);
154 |
155 |
156 |
157 | const handleButtonClick = () => {
158 | handleSelectNode();
159 | if (buttonRef.current) {
160 | const rect = buttonRef.current.getBoundingClientRect();
161 | setDropdownPosition({
162 | top: rect.bottom + window.scrollY,
163 | left: rect.left + window.scrollX,
164 | });
165 | setDropdown(true);
166 | }
167 | };
168 | const DropdownContent = () => (
169 |
178 |
179 |
{
182 | e.key === "Enter" && handlePrompt("Custom");
183 | }}
184 | type="text"
185 | className="p-2 rounded-lg w-full outline-none "
186 | placeholder="Modify with a prompt"
187 | />
188 |
189 | {Object.keys(PROMPT_TYPES).map((type) => (
190 |
197 | ))}
198 |
199 |
200 |
216 |
223 |