enhanced-notes-multiple-per-session

apps/desktop/src/components/chat/session.tsx

@@ -4,8 +4,8 @@ import { type ReactNode, useEffect, useMemo, useRef } from "react";
 
 import { CustomChatTransport } from "../../chat/transport";
 import type { HyprUIMessage } from "../../chat/types";
-import { useToolRegistry } from "../../contexts/tool";
 import { useLanguageModel } from "../../hooks/useLLMConnection";
+import { useToolRegistry } from "../../contexts/tool";
 import * as main from "../../store/tinybase/main";
 import { id } from "../../utils";
 

apps/desktop/src/components/chat/view.tsx

@@ -1,8 +1,8 @@
 import { useCallback, useRef } from "react";
 
 import type { HyprUIMessage } from "../../chat/types";
-import { useShell } from "../../contexts/shell";
 import { useLanguageModel } from "../../hooks/useLLMConnection";
+import { useShell } from "../../contexts/shell";
 import * as main from "../../store/tinybase/main";
 import { id } from "../../utils";
 import { ChatBody } from "./body";

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

@@ -12,7 +12,7 @@ export function FloatingActionButton({
   const currentTab = useCurrentNoteTab(tab);
   const hasTranscript = useHasTranscript(tab.id);
 
-  if (!(currentTab === "raw" && !hasTranscript)) {
+  if (!(currentTab.type === "raw" && !hasTranscript)) {
     return null;
   }
 

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

@@ -72,7 +72,7 @@ export function TabContentNote({
   });
 
   const showTimeline =
-    tab.state.editor === "transcript" &&
+    tab.state.editor?.type === "transcript" &&
     Boolean(audioUrl) &&
     listenerStatus === "inactive";
 

apps/desktop/src/components/main/body/sessions/note-input/enhanced/editor.tsx

@@ -7,8 +7,8 @@ import * as main from "../../../../../../store/tinybase/main";
 
 export const EnhancedEditor = forwardRef<
   { editor: TiptapEditor | null },
-  { sessionId: string }
->(({ sessionId }, ref) => {
+  { sessionId: string; enhancedNoteId: string }
+>(({ enhancedNoteId }, ref) => {
   const store = main.UI.useStore(main.STORE_ID);
   const [initialContent, setInitialContent] = useState<string>("");
 
@@ -41,7 +41,7 @@ export const EnhancedEditor = forwardRef<
     <div className="h-full">
       <NoteEditor
         ref={ref}
-        key={`session-${sessionId}-enhanced`}
+        key={`enhanced-note-${enhancedNoteId}`}
         initialContent={initialContent}
         handleChange={handleChange}
         mentionConfig={mentionConfig}

apps/desktop/src/components/main/body/sessions/note-input/enhanced/index.tsx

@@ -9,9 +9,9 @@ import { StreamingView } from "./streaming";
 
 export const Enhanced = forwardRef<
   { editor: TiptapEditor | null },
-  { sessionId: string }
->(({ sessionId }, ref) => {
-  const taskId = createTaskId(sessionId, "enhance");
+  { sessionId: string; enhancedNoteId: string }
+>(({ sessionId, enhancedNoteId }, ref) => {
+  const taskId = createTaskId(enhancedNoteId, "enhance");
 
   const { status } = useAITaskTask(taskId, "enhance");
 
@@ -20,8 +20,14 @@ export const Enhanced = forwardRef<
   }
 
   if (status === "generating") {
-    return <StreamingView sessionId={sessionId} />;
+    return <StreamingView enhancedNoteId={enhancedNoteId} />;
   }
 
-  return <EnhancedEditor ref={ref} sessionId={sessionId} />;
+  return (
+    <EnhancedEditor
+      ref={ref}
+      sessionId={sessionId}
+      enhancedNoteId={enhancedNoteId}
+    />
+  );
 });

apps/desktop/src/components/main/body/sessions/note-input/enhanced/streaming.tsx

@@ -12,8 +12,8 @@ import {
 } from "../../../../../../store/zustand/ai-task/task-configs";
 import { type TaskStepInfo } from "../../../../../../store/zustand/ai-task/tasks";
 
-export function StreamingView({ sessionId }: { sessionId: string }) {
-  const taskId = createTaskId(sessionId, "enhance");
+export function StreamingView({ enhancedNoteId }: { enhancedNoteId: string }) {
+  const taskId = createTaskId(enhancedNoteId, "enhance");
   const { streamedText, isGenerating } = useAITaskTask(taskId, "enhance");
 
   const containerRef = useAutoScrollToBottom(streamedText);