feat(sessions): add spinner forancing and finalizing states

Author: ComputelessComputer

apps/desktop/src/components/main/body/sessions/index.tsx

@@ -6,6 +6,7 @@ import { commands as miscCommands } from "@hypr/plugin-misc";
 
 import AudioPlayer from "../../../../contexts/audio-player";
 import { useListener } from "../../../../contexts/listener";
+import { useIsSessionEnhancing } from "../../../../hooks/useEnhancedNotes";
 import * as main from "../../../../store/tinybase/main";
 import { rowIdfromTab, type Tab } from "../../../../store/zustand/tabs";
 import { StandardTabWrapper } from "../index";
@@ -35,15 +36,19 @@ export const TabItemNote: TabItem<Extract<Tab, { type: "sessions" }>> = ({
     main.STORE_ID,
   );
   const sessionMode = useListener((state) => state.getSessionMode(tab.id));
+  const isEnhancing = useIsSessionEnhancing(tab.id);
   const isActive =
     sessionMode === "running_active" || sessionMode === "finalizing";
+  const isFinalizing = sessionMode === "finalizing";
+  const showSpinner = isFinalizing || isEnhancing;
 
   return (
     <TabItemBase
       icon={<StickyNoteIcon className="w-4 h-4" />}
       title={title || "Untitled"}
       selected={tab.active}
       active={isActive}
+      finalizing={showSpinner}
       tabIndex={tabIndex}
       handleCloseThis={() => handleCloseThis(tab)}
       handleSelectThis={() => handleSelectThis(tab)}

apps/desktop/src/components/main/sidebar/timeline/item.tsx

@@ -2,8 +2,11 @@ import { memo, useCallback, useMemo } from "react";
 
 import { commands as analyticsCommands } from "@hypr/plugin-analytics";
 import { ContextMenuItem } from "@hypr/ui/components/ui/context-menu";
+import { Spinner } from "@hypr/ui/components/ui/spinner";
 import { cn } from "@hypr/utils";
 
+import { useListener } from "../../../../contexts/listener";
+import { useIsSessionEnhancing } from "../../../../hooks/useEnhancedNotes";
 import * as main from "../../../../store/tinybase/main";
 import { type TabInput, useTabs } from "../../../../store/zustand/tabs";
 import { id } from "../../../../utils";
@@ -36,6 +39,14 @@ export const TimelineItemComponent = memo(
     const timestamp =
       item.type === "event" ? item.data.started_at : item.data.created_at;
 
+    const sessionId = item.type === "session" ? item.id : null;
+    const sessionMode = useListener((state) =>
+      sessionId ? state.getSessionMode(sessionId) : "inactive",
+    );
+    const isEnhancing = useIsSessionEnhancing(sessionId ?? "");
+    const isFinalizing = sessionMode === "finalizing";
+    const showSpinner = isFinalizing || isEnhancing;
+
     const handleClick = () => {
       if (item.type === "event") {
         handleEventClick(false);
@@ -171,11 +182,18 @@ export const TimelineItemComponent = memo(
           !selected && "hover:bg-neutral-100",
         ])}
       >
-        <div className="flex flex-col gap-0.5">
-          <div className="text-sm font-normal truncate">{title}</div>
-          {displayTime && (
-            <div className="text-xs text-neutral-500">{displayTime}</div>
+        <div className="flex items-start gap-2">
+          {showSpinner && (
+            <div className="flex-shrink-0 mt-0.5">
+              <Spinner size={14} />
+            </div>
           )}
+          <div className="flex flex-col gap-0.5 min-w-0">
+            <div className="text-sm font-normal truncate">{title}</div>
+            {displayTime && (
+              <div className="text-xs text-neutral-500">{displayTime}</div>
+            )}
+          </div>
         </div>
       </InteractiveButton>
     );

apps/desktop/src/hooks/useEnhancedNotes.ts

@@ -1,8 +1,10 @@
-import { useCallback, useEffect } from "react";
+import { useCallback, useEffect, useMemo } from "react";
 
 import { useHasTranscript } from "../components/main/body/sessions/shared";
+import { useAITask } from "../contexts/ai-task";
 import { useListener } from "../contexts/listener";
 import * as main from "../store/tinybase/main";
+import { createTaskId } from "../store/zustand/ai-task/task-configs";
 
 export function useCreateEnhancedNote() {
   const store = main.UI.useStore(main.STORE_ID) as main.Store | undefined;
@@ -170,3 +172,24 @@ export function useEnsureDefaultSummary(sessionId: string) {
     createEnhancedNote,
   ]);
 }
+
+export function useIsSessionEnhancing(sessionId: string): boolean {
+  const enhancedNoteIds = main.UI.useSliceRowIds(
+    main.INDEXES.enhancedNotesBySession,
+    sessionId,
+    main.STORE_ID,
+  );
+
+  const taskIds = useMemo(
+    () => (enhancedNoteIds || []).map((id) => createTaskId(id, "enhance")),
+    [enhancedNoteIds],
+  );
+
+  const isEnhancing = useAITask((state) => {
+    return taskIds.some(
+      (taskId) => state.tasks[taskId]?.status === "generating",
+    );
+  });
+
+  return isEnhancing;
+}