Add missing props

Author: ashercoren

vim--ai-scribe--react/src/components/templates/DebugView.tsx renamed to vim--ai-scribe--react/src/components/organisms/notes-tab/DebugView.tsx

@@ -1,6 +1,7 @@
-import { TranscriptionPanel } from "../organisms/transcription-panel/TranscriptionPanel";
-import { NotesSections } from "../organisms/notes-tab/NotesSections";
-import type { TranscriptionSegment } from "../organisms/ai-scribe-demo/transcription.mock";
+import { TranscriptionPanel } from "../transcription-panel/TranscriptionPanel";
+import { NotesSections } from "./NotesSections";
+import type { TranscriptionSegment } from "../ai-scribe-demo/transcription.mock";
+import { useUpdateEncounter } from "./useSectionWriteAvailability";
 
 interface DebugViewProps {
   transcriptionSegments: TranscriptionSegment[];
@@ -21,6 +22,8 @@ export const DebugView = ({
   onHoverSegment,
   renderHighlightedText,
 }: DebugViewProps) => {
+  const updateEncounterState = useUpdateEncounter();
+
   return (
     <div className="grid grid-cols-2 gap-6">
       <TranscriptionPanel
@@ -32,6 +35,7 @@ export const DebugView = ({
         hoveredSegment={hoveredSegment}
         transcriptionSegments={transcriptionSegments}
         renderHighlightedText={renderHighlightedText}
+        {...updateEncounterState}
       />
     </div>
   );

vim--ai-scribe--react/src/components/organisms/notes-tab/NotesTab.tsx

@@ -1,16 +1,11 @@
-import { useState } from "react";
 import { useNoteFormContext } from "@/providers/NoteFormContext";
 import { useVimOsContext } from "@/providers/VimOSContext";
+import { useState } from "react";
 import { Button } from "../../atoms/Button";
-import { DebugView } from "../../templates/DebugView";
+import { DebugView } from "./DebugView";
 import { MOCK_TRANSCRIPTION } from "../ai-scribe-demo/transcription.mock";
 import { NotesSections } from "./NotesSections";
-import {
-  useUpdateSubjective,
-  useUpdateObjective,
-  useUpdateAssessment,
-  useUpdatePlan,
-} from "./useSectionWriteAvailability";
+import { useUpdateEncounter } from "./useSectionWriteAvailability";
 
 export const NotesTab = ({
   patientName,
@@ -27,20 +22,14 @@ export const NotesTab = ({
   const { watch } = useNoteFormContext();
   const currentNote = watch();
 
-  // Use the hooks to get write availability and update functions
-  const { canUpdateSubjectiveNote, updateSubjectiveNote } =
-    useUpdateSubjective();
-  const { canUpdateObjectiveNote, updateObjectiveNote } = useUpdateObjective();
-  const { canUpdateAssessmentNote, updateAssessmentNote } =
-    useUpdateAssessment();
-  const { canUpdatePlanNote, updatePlanNote } = useUpdatePlan();
+  const updateEncounterState = useUpdateEncounter();
 
   // Only enable the button if all are true
   const canPushAll =
-    canUpdateSubjectiveNote &&
-    canUpdateObjectiveNote &&
-    canUpdateAssessmentNote &&
-    canUpdatePlanNote;
+    updateEncounterState.canUpdateSubjectiveNote &&
+    updateEncounterState.canUpdateObjectiveNote &&
+    updateEncounterState.canUpdateAssessmentNote &&
+    updateEncounterState.canUpdatePlanNote;
 
   const toggleDebugMode = () => {
     setIsDebugMode(!isDebugMode);
@@ -82,14 +71,7 @@ export const NotesTab = ({
           hoveredSegment={hoveredSegment}
           transcriptionSegments={MOCK_TRANSCRIPTION}
           renderHighlightedText={renderHighlightedText}
-          canUpdateSubjectiveNote={canUpdateSubjectiveNote}
-          canUpdateObjectiveNote={canUpdateObjectiveNote}
-          canUpdateAssessmentNote={canUpdateAssessmentNote}
-          canUpdatePlanNote={canUpdatePlanNote}
-          updateSubjectiveNote={updateSubjectiveNote}
-          updateObjectiveNote={updateObjectiveNote}
-          updateAssessmentNote={updateAssessmentNote}
-          updatePlanNote={updatePlanNote}
+          {...updateEncounterState}
         />
       )}
     </>

vim--ai-scribe--react/src/components/organisms/notes-tab/useSectionWriteAvailability.ts

@@ -83,3 +83,22 @@ export const useUpdatePlan = () => {
     updatePlanNote,
   };
 };
+
+export const useUpdateEncounter = () => {
+  const { canUpdateSubjectiveNote, updateSubjectiveNote } =
+    useUpdateSubjective();
+  const { canUpdateObjectiveNote, updateObjectiveNote } = useUpdateObjective();
+  const { canUpdateAssessmentNote, updateAssessmentNote } =
+    useUpdateAssessment();
+  const { canUpdatePlanNote, updatePlanNote } = useUpdatePlan();
+  return {
+    canUpdateSubjectiveNote,
+    updateSubjectiveNote,
+    canUpdateObjectiveNote,
+    updateObjectiveNote,
+    canUpdateAssessmentNote,
+    updateAssessmentNote,
+    canUpdatePlanNote,
+    updatePlanNote,
+  };
+};