feat: add dis/-like option for recommendation (#96)

* feat: add updateSupervisor method to useChats hook * feat: add option to dis/-like a recommended supervisor * refactor: small improvements
Solvro · Jan 25, 2025 · 4dfa9c3 · 4dfa9c3
1 parent 3f1b13d
commit 4dfa9c3
Show file tree

Hide file tree

Showing 3 changed files with 106 additions and 2 deletions.
diff --git a/src/components/supervisor.tsx b/src/components/supervisor.tsx
@@ -1,8 +1,15 @@
 "use client";
 
-import { FileText, GraduationCap, Star } from "lucide-react";
+import {
+  FileText,
+  GraduationCap,
+  Star,
+  ThumbsDown,
+  ThumbsUp,
+} from "lucide-react";
 
 import { Button } from "@/components/ui/button";
+import { useChats } from "@/hooks/use-chats";
 import { useSupervisors } from "@/hooks/use-supervisors";
 import { faculties } from "@/lib/faculties";
 import type { Supervisor as SupervisorType } from "@/types/supervisor";
@@ -12,16 +19,24 @@ import {
   AccordionItem,
   AccordionTrigger,
 } from "./ui/accordion";
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipProvider,
+  TooltipTrigger,
+} from "./ui/tooltip";
 
 export function Supervisor({
-  supervisor: { uuid, faculty, name, papers, theses },
+  supervisor,
   chatUuid,
   prompt,
 }: {
   supervisor: SupervisorType;
   chatUuid: string;
   prompt: string;
 }) {
+  const { uuid, faculty, name, papers, theses, isAdequate } = { ...supervisor };
+  const { updateSupervisor } = useChats();
   const { addSupervisor, getSupervisor, removeSupervisor } = useSupervisors();
   const isSaved = getSupervisor(uuid) !== null;
 
@@ -88,6 +103,74 @@ export function Supervisor({
               <p>{description}</p>
             </div>
           ))}
+          <div className="flex items-center justify-end pt-4">
+            <TooltipProvider delayDuration={0}>
+              <Tooltip>
+                <TooltipTrigger>
+                  <span className="mr-2 pt-1">
+                    Czy ta rekomendacja była sensowna?
+                  </span>
+                </TooltipTrigger>
+                <TooltipContent className="text-center">
+                  <span>
+                    Odpowiadając na pytanie pomagasz nam dostrajać model,
+                    <br />
+                    co przełoży się na bardziej jakościowe rekomendacje 😉
+                  </span>
+                </TooltipContent>
+              </Tooltip>
+            </TooltipProvider>
+            <Button
+              variant="transparent"
+              size="icon"
+              className="hover:bg-chat-user"
+              onClick={() => {
+                const newIsAdequate =
+                  isAdequate === undefined || !isAdequate ? true : undefined;
+
+                updateSupervisor(chatUuid, {
+                  ...supervisor,
+                  isAdequate: newIsAdequate,
+                });
+              }}
+            >
+              <ThumbsUp
+                size={20}
+                className={
+                  isAdequate === undefined
+                    ? "fill-transparent"
+                    : isAdequate
+                      ? "fill-white/70"
+                      : "fill-transparent"
+                }
+              />
+            </Button>
+            <Button
+              size="icon"
+              variant="transparent"
+              className="hover:bg-chat-user"
+              onClick={() => {
+                const newIsAdequate =
+                  isAdequate === undefined || isAdequate ? false : undefined;
+
+                updateSupervisor(chatUuid, {
+                  ...supervisor,
+                  isAdequate: newIsAdequate,
+                });
+              }}
+            >
+              <ThumbsDown
+                size={20}
+                className={
+                  isAdequate === undefined
+                    ? "fill-transparent"
+                    : isAdequate
+                      ? "fill-transparent"
+                      : "fill-white/70"
+                }
+              />
+            </Button>
+          </div>
         </AccordionContent>
       </div>
     </AccordionItem>

diff --git a/src/hooks/use-chats.ts b/src/hooks/use-chats.ts
@@ -3,6 +3,7 @@ import { useCallback } from "react";
 
 import { chatsAtom } from "@/atoms/chats";
 import type { Chat } from "@/types/chat";
+import type { Supervisor } from "@/types/supervisor";
 
 export function useChats() {
   const [chats, setChats] = useAtom(chatsAtom);
@@ -41,11 +42,30 @@ export function useChats() {
     [setChats],
   );
 
+  const updateSupervisor = useCallback(
+    (chatUuid: string, updatedSupervisor: Supervisor) => {
+      setChats((previousChats) =>
+        previousChats.map((chat) => {
+          if (chat.uuid === chatUuid && chat.supervisors !== undefined) {
+            const updatedSupervisors: Supervisor[] = chat.supervisors.map(
+              (s) =>
+                s.uuid === updatedSupervisor.uuid ? updatedSupervisor : s,
+            );
+            return { ...chat, supervisors: updatedSupervisors };
+          }
+          return chat;
+        }),
+      );
+    },
+    [setChats],
+  );
+
   return {
     chats,
     getChat,
     addChat,
     removeChat,
     updateChat,
+    updateSupervisor,
   };
 }
diff --git a/src/types/supervisor.ts b/src/types/supervisor.ts
@@ -12,6 +12,7 @@ interface SupervisorResponse {
 
 interface Supervisor extends SupervisorResponse {
   uuid: string;
+  isAdequate?: boolean;
 }
 
 interface SavedSupervisor extends Supervisor {