From 64f629d236cde37f3bf0942169173848a5db1051 Mon Sep 17 00:00:00 2001
From: Micah Woodard <micah.woodard@alleninstitute.org>
Date: Tue, 15 Sep 2026 16:28:55 -0700
Subject: [PATCH] logs notes during a session

---
 src/Extensions/IntraSessionVisualizer.bonsai  | 25 ++++++++++++++++++-
 .../data_contract/_dataset.py                 |  7 ++++++
 2 files changed, 31 insertions(+), 1 deletion(-)

diff --git a/src/Extensions/IntraSessionVisualizer.bonsai b/src/Extensions/IntraSessionVisualizer.bonsai
index f21e3c7..025ff5e 100644
--- a/src/Extensions/IntraSessionVisualizer.bonsai
+++ b/src/Extensions/IntraSessionVisualizer.bonsai
@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="utf-8"?>
-<WorkflowBuilder Version="2.9.0"
+<WorkflowBuilder Version="2.9.1"
                  xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
                  xmlns:rx="clr-namespace:Bonsai.Reactive;assembly=Bonsai.Core"
                  xmlns:imgui="clr-namespace:Bonsai.ImGui;assembly=Bonsai.ImGui"
@@ -306,12 +306,29 @@
             <Expression xsi:type="imgui:TableNextColumnBuilder">
               <imgui:Visible>true</imgui:Visible>
             </Expression>
+            <Expression xsi:type="imgui:ButtonBuilder">
+              <imgui:Name>NotesButton</imgui:Name>
+              <imgui:Visible>true</imgui:Visible>
+              <imgui:Text>Save Notes</imgui:Text>
+            </Expression>
             <Expression xsi:type="p1:MultiLineInputTextBuilder">
               <p1:Name>Notes</p1:Name>
               <p1:Visible>true</p1:Visible>
               <p1:Text>Notes</p1:Text>
               <p1:Capacity>16384</p1:Capacity>
             </Expression>
+            <Expression xsi:type="Combinator">
+              <Combinator xsi:type="rx:WithLatestFrom" />
+            </Expression>
+            <Expression xsi:type="Combinator">
+              <Combinator xsi:type="rx:DistinctUntilChanged" />
+            </Expression>
+            <Expression xsi:type="MemberSelector">
+              <Selector>Item2</Selector>
+            </Expression>
+            <Expression xsi:type="IncludeWorkflow" Path="Extensions\InlineSoftwareEvent.bonsai">
+              <EventName>Notes</EventName>
+            </Expression>
           </Nodes>
           <Edges>
             <Edge From="0" To="1" Label="Source1" />
@@ -385,6 +402,12 @@
             <Edge From="63" To="64" Label="Source1" />
             <Edge From="64" To="65" Label="Source2" />
             <Edge From="66" To="67" Label="Source1" />
+            <Edge From="66" To="68" Label="Source1" />
+            <Edge From="67" To="69" Label="Source1" />
+            <Edge From="68" To="69" Label="Source2" />
+            <Edge From="69" To="70" Label="Source1" />
+            <Edge From="70" To="71" Label="Source1" />
+            <Edge From="71" To="72" Label="Source1" />
           </Edges>
         </Workflow>
       </Expression>
diff --git a/src/aind_behavior_dynamic_foraging/data_contract/_dataset.py b/src/aind_behavior_dynamic_foraging/data_contract/_dataset.py
index 746501e..11a41b7 100644
--- a/src/aind_behavior_dynamic_foraging/data_contract/_dataset.py
+++ b/src/aind_behavior_dynamic_foraging/data_contract/_dataset.py
@@ -216,6 +216,13 @@ def make_dataset(
                                     root_path / "behavior/SoftwareEvents/LeftManualAutoReward.json"
                                 ),
                             ),
+                            SoftwareEvents(
+                                name="Notes",
+                                description="An event emitted when notes are saved in a session.",
+                                reader_params=SoftwareEvents.make_params(
+                                    root_path / "behavior/SoftwareEvents/Notes.json"
+                                ),
+                            ),
                             SoftwareEvents(
                                 name="RightManualAutoReward",
                                 description="An event emitted when the manual right auto reward is triggered. This manual reward is aligned with go cue.",