np-workflows 1.6.89__py3-none-any.whl → 1.6.91__py3-none-any.whl

np_workflows/shared/widgets.py

@@ -21,17 +21,20 @@ import np_workflows.shared.npxc as npxc
 
 logger = np_logging.getLogger(__name__)
 
-np_logging.getLogger('Comm').propagate = False
-np_logging.getLogger('PIL').propagate = False
+np_logging.getLogger("Comm").propagate = False
+np_logging.getLogger("PIL").propagate = False
 
 global_state = {}
 """Global variable for persisting widget states."""
 
+
 def elapsed_time_widget() -> IPython.display.DisplayHandle | None:
     """Displays a clock showing the elapsed time since the cell was first run."""
 
     clock_widget = ipw.Label("")
-    reminder_widget = ipw.Label("Remember to restart JupyterLab and run update.bat before every experiment!")
+    reminder_widget = ipw.Label(
+        "Remember to restart JupyterLab and run update.bat before every experiment!"
+    )
     global start_time
     if "start_time" not in globals():
         start_time = time.time()
@@ -44,11 +47,13 @@ def elapsed_time_widget() -> IPython.display.DisplayHandle | None:
             elapsed_sec = time.time() - start_time
             hours, remainder = divmod(elapsed_sec, 3600)
             minutes, seconds = divmod(remainder, 60)
-            clock_widget.value = "Elapsed time: {:02}h {:02}m {:02}s".format(
-                int(hours), int(minutes), int(seconds)
+            clock_widget.value = (
+                f"Elapsed time: {int(hours):02}h {int(minutes):02}m {int(seconds):02}s"
             )
             if hours > 4:  # ipywidgets >= 8.0
-                clock_widget.style = dict(text_color="red",)
+                clock_widget.style = dict(
+                    text_color="red",
+                )
             time.sleep(0.2)
 
     thread = threading.Thread(target=update_timer, args=())
@@ -61,12 +66,14 @@ def user_and_mouse_widget() -> tuple[np_session.User, np_session.Mouse]:
     user_description = "User:"
     mouse_description = "Mouse:"
     user_widget = ipw.Select(options=npxc.lims_user_ids, description=user_description)
-    mouse_widget = ipw.Text(value=str(npxc.default_mouse_id), description=mouse_description)
-    for widget, string in zip((user_widget, mouse_widget), ('user', 'mouse')):
-        if (selected := global_state.get(f'selected_{string}')):
+    mouse_widget = ipw.Text(
+        value=str(npxc.default_mouse_id), description=mouse_description
+    )
+    for widget, string in zip((user_widget, mouse_widget), ("user", "mouse")):
+        if selected := global_state.get(f"selected_{string}"):
             widget.value = selected
             with console:
-                print(f'Current {string}: {selected}')
+                print(f"Current {string}: {selected}")
     user = np_session.User(str(user_widget.value))
     mouse = np_session.Mouse(str(mouse_widget.value))
 
@@ -74,7 +81,7 @@ def user_and_mouse_widget() -> tuple[np_session.User, np_session.Mouse]:
         if str(user) == (new := str(new_user).strip()):
             return
         user.__init__(new)
-        global_state['selected_user'] = new
+        global_state["selected_user"] = new
         with console:
             print(f"User updated: {user}")
 
@@ -83,26 +90,28 @@ def user_and_mouse_widget() -> tuple[np_session.User, np_session.Mouse]:
             return
         if len(new) < 6:
             return
-        global_state['selected_mouse'] = new
+        global_state["selected_mouse"] = new
         mouse.__init__(new)
         with console:
             print(f"Mouse updated: {mouse}")
-        
+
     def new_value(change) -> None:
-        if change['name'] != 'value':
+        if change["name"] != "value":
             return
-        if (options := getattr(change['owner'], 'options', None)) and change['new'] not in options:
+        if (options := getattr(change["owner"], "options", None)) and change[
+            "new"
+        ] not in options:
             return
-        if change['new'] == change['old']:
+        if change["new"] == change["old"]:
             return
-        if (desc := getattr(change['owner'], 'description')) == user_description:
-                update_user(change['new'])
+        if (desc := change["owner"].description) == user_description:
+            update_user(change["new"])
         elif desc == mouse_description:
-                update_mouse(change['new'])
-            
+            update_mouse(change["new"])
+
     user_widget.observe(new_value)
     mouse_widget.observe(new_value)
-    
+
     IPython.display.display(ipw.VBox([user_widget, mouse_widget, console]))
     return user, mouse
 
@@ -110,8 +119,7 @@ def user_and_mouse_widget() -> tuple[np_session.User, np_session.Mouse]:
 def mtrain_widget(
     labtracks_mouse_id: str | int | np_session.Mouse,
 ) -> IPython.display.DisplayHandle | None:
-    """Displays a widget to view and edit MTrain regimen/stage for a mouse.
-    """
+    """Displays a widget to view and edit MTrain regimen/stage for a mouse."""
     if not isinstance(labtracks_mouse_id, np_session.Mouse):
         mtrain = np_session.MTrain(labtracks_mouse_id)
     else:
@@ -162,9 +170,9 @@ def mtrain_widget(
         reset_update_button()
         if change["new"] is None:
             return
-        if change["new"] != stage_label.value or str(
-            regimen_dropdown.value
-        ) != str(regimen_label.value):
+        if change["new"] != stage_label.value or str(regimen_dropdown.value) != str(
+            regimen_label.value
+        ):
             # enable button if stage name changed, or regimen name changed (some
             # regimens have the same stage names as other regimens)
             update_button.disabled = False
@@ -218,20 +226,24 @@ def mtrain_widget(
 
     return IPython.display.display(display)
 
+
 def check_widget(check: str, *checks: str) -> ipw.Widget:
     layout = ipw.Layout(min_width="600px")
-    widget = ipw.VBox([
-        ipw.Label(check, layout=layout), 
-        *(ipw.Checkbox(description=_, layout=layout) for _ in checks),
-        # ipw.Button(description="Continue", disabled=True)
-        ])
+    widget = ipw.VBox(
+        [
+            ipw.Label(check, layout=layout),
+            *(ipw.Checkbox(description=_, layout=layout) for _ in checks),
+            # ipw.Button(description="Continue", disabled=True)
+        ]
+    )
     return widget
 
+
 def await_all_checkboxes(widget: ipw.Box) -> None:
     while any(_.value is False for _ in widget.children if isinstance(_, ipw.Checkbox)):
         time.sleep(0.1)
-    
-    
+
+
 def check_openephys_widget() -> None:
     check = "OpenEphys checks:"
     checks = (
@@ -242,7 +254,8 @@ def check_openephys_widget() -> None:
     )
     IPython.display.display(widget := check_widget(check, *checks))
 
-def check_hardware_widget() ->  None:
+
+def check_hardware_widget() -> None:
     check = "Stage checks:"
     checks = (
         "Cartridge raised (fully retract probes before raising!)",
@@ -252,181 +265,273 @@ def check_hardware_widget() ->  None:
     )
     IPython.display.display(widget := check_widget(check, *checks))
 
+
 def check_mouse_widget() -> None:
     check = "Mouse checks before lowering cartridge:"
     checks = (
         "Stabilization screw",
-        ("Silicon oil applied" if npxc.RIG.idx == 0 else "Quickcast removed, agarose applied"),
+        (
+            "Silicon oil applied"
+            if npxc.RIG.idx == 0
+            else "Quickcast removed, agarose applied"
+        ),
         "Tail cone down",
         "Continuity/Resistance check",
     )
     IPython.display.display(widget := check_widget(check, *checks))
 
+
 def pre_stim_check_widget() -> None:
     check = "Before running stim:"
     checks = (
         "Behavior cameras are in focus",
-        "Eye-tracking mirror in place", 
+        "Eye-tracking mirror in place",
         "Windows minimized on Stim computer (Win+D)",
         "Monitor closed",
         "Photodoc light off",
-        "Curtain down",  
+        "Curtain down",
     )
     IPython.display.display(widget := check_widget(check, *checks))
-    
+
+
 def finishing_checks_widget() -> None:
     check = "Finishing checks:"
     checks = (
         "Add quickcast etc.",
-        "Remove and water mouse", 
-        "Dip probes",     
+        "Remove and water mouse",
+        "Dip probes",
     )
     IPython.display.display(widget := check_widget(check, *checks))
-    
-    
-def wheel_height_widget(session: np_session.PipelineSession) -> IPython.display.DisplayHandle | None:
+
+
+def wheel_height_widget(
+    session: np_session.PipelineSession,
+) -> IPython.display.DisplayHandle | None:
     "Saves wheel height to platform_json and stores in `mouse.state['wheel_height']`."
-    
-    layout = ipw.Layout(max_width='130px')
-    
-    prev_height = session.mouse.state.get('wheel_height', 0)
-    height_counter = ipw.BoundedFloatText(value=prev_height, min=0, max=10, step=0.1, description="Wheel height", layout=layout)
-    save_button = ipw.Button(description='Save', button_style='warning', layout=layout)
+
+    layout = ipw.Layout(max_width="130px")
+
+    prev_height = session.mouse.state.get("wheel_height", 0)
+    height_counter = ipw.BoundedFloatText(
+        value=prev_height,
+        min=0,
+        max=10,
+        step=0.1,
+        description="Wheel height",
+        layout=layout,
+    )
+    save_button = ipw.Button(description="Save", button_style="warning", layout=layout)
 
     def on_click(b):
         session.platform_json.wheel_height = height_counter.value
-        session.mouse.state['wheel_height'] = height_counter.value
-        save_button.button_style = 'success'
-        save_button.description = 'Saved'
+        session.mouse.state["wheel_height"] = height_counter.value
+        save_button.button_style = "success"
+        save_button.description = "Saved"
+
     save_button.on_click(on_click)
-    return IPython.display.display(ipw.VBox([height_counter,save_button]))
-    
+    return IPython.display.display(ipw.VBox([height_counter, save_button]))
+
 
-def di_widget(session: np_session.PipelineSession) -> IPython.display.DisplayHandle | None:
+def di_widget(
+    session: np_session.PipelineSession,
+) -> IPython.display.DisplayHandle | None:
     "Supply a path or a platform json instance. Saves a JSON file with the dye used in the session and a timestamp."
-    
+
     di_info: dict[str, int | str] = dict(
-        EndTime=0, StartTime=npxc.now(), dii_description="", times_dipped=0, previous_uses="",
+        EndTime=0,
+        StartTime=npxc.now(),
+        dii_description="",
+        times_dipped=0,
+        previous_uses="",
     )
     di_info.update(session.platform_json.DiINotes)
-    
-    layout = ipw.Layout(max_width='180px')
-    dipped_counter = ipw.IntText(value=0, min=0, max=99, description="Dipped count", layout=layout)
-    usage_counter = ipw.IntText(value=0, min=0, max=99, description="Previous uses", layout=layout)
-    dye_dropdown = ipw.Dropdown(options=['CM-DiI 100%', 'DiO'], layout=layout)
-    save_button = ipw.Button(description='Save', button_style='warning', layout=layout)
-    
+
+    layout = ipw.Layout(max_width="180px")
+    dipped_counter = ipw.IntText(
+        value=0, min=0, max=99, description="Dipped count", layout=layout
+    )
+    usage_counter = ipw.IntText(
+        value=0, min=0, max=99, description="Previous uses", layout=layout
+    )
+    dye_dropdown = ipw.Dropdown(options=["CM-DiI 100%", "DiO"], layout=layout)
+    save_button = ipw.Button(description="Save", button_style="warning", layout=layout)
+
     def update_di_info():
-        di_info['EndTime'] = npxc.now()
-        di_info['times_dipped'] = str(dipped_counter.value)
-        di_info['dii_description'] = str(dye_dropdown.value)
-        di_info['previous_uses'] = str(usage_counter.value)
-        
+        di_info["EndTime"] = npxc.now()
+        di_info["times_dipped"] = str(dipped_counter.value)
+        di_info["dii_description"] = str(dye_dropdown.value)
+        di_info["previous_uses"] = str(usage_counter.value)
+
     def on_click(b):
         update_di_info()
         session.platform_json.DiINotes = di_info
-        save_button.description = 'Saved'
-        save_button.button_style = 'success'
-        
+        save_button.description = "Saved"
+        save_button.button_style = "success"
+
     save_button.on_click(on_click)
-    return IPython.display.display(ipw.VBox([
-        dipped_counter, dye_dropdown, 
-        usage_counter, save_button]))
+    return IPython.display.display(
+        ipw.VBox([dipped_counter, dye_dropdown, usage_counter, save_button])
+    )
 
-    
-def dye_info_widget(session: np_session.PipelineSession) -> IPython.display.DisplayHandle | None:
+
+def dye_info_widget(
+    session: np_session.PipelineSession,
+) -> IPython.display.DisplayHandle | None:
     """
     - scan barcode or enter ID number for the dye used
     - change dye description if incorrect (DiI, DiO)
     - increment number of times probes were dipped this session
     - hit `Save` to store info in platform.json
     """
-    
+
     di_info: dict[str, int | str] = dict(
-        EndTime=0, StartTime=npxc.now(), dii_description="", times_dipped=0, previous_uses="",
+        EndTime=0,
+        StartTime=npxc.now(),
+        dii_description="",
+        times_dipped=0,
+        previous_uses="",
     )
     di_info.update(session.platform_json.DiINotes)
-    
+
     def width(w):
-        return ipw.Layout(max_width=f'{w}px')
-    
-    dye_id_entry = ipw.Text(value=None, description='Dye ID', layout=width(250), placeholder='Enter ID or scan barcode')
-    ipw.Button(description='Record single use', button_style='warning', layout=width(180))
-    first_usage = ipw.Text(value='', description="First use", layout=width(250), disabled=True)
-    dye_dropdown = ipw.Dropdown(description="Description:", options=np_session.Dye.descriptions, layout=width(180))
-    dipped_counter = ipw.IntText(value=int(di_info['times_dipped'] or 0), min=0, max=99, description="Dipped count", layout=width(150))
-    usage_counter = ipw.IntText(value=int(di_info['previous_uses'] or 0), min=0, max=99, description="Previous uses", layout=width(180), disabled=True)
-    save_button = ipw.Button(description='Save', button_style='warning', layout=width(180))
-    if (desc := di_info['dii_description']) in np_session.Dye.descriptions:
+        return ipw.Layout(max_width=f"{w}px")
+
+    dye_id_entry = ipw.Text(
+        value=None,
+        description="Dye ID",
+        layout=width(250),
+        placeholder="Enter ID or scan barcode",
+    )
+    ipw.Button(
+        description="Record single use", button_style="warning", layout=width(180)
+    )
+    first_usage = ipw.Text(
+        value="", description="First use", layout=width(250), disabled=True
+    )
+    dye_dropdown = ipw.Dropdown(
+        description="Description:",
+        options=np_session.Dye.descriptions,
+        layout=width(180),
+    )
+    dipped_counter = ipw.IntText(
+        value=int(di_info["times_dipped"] or 0),
+        min=0,
+        max=99,
+        description="Dipped count",
+        layout=width(150),
+    )
+    usage_counter = ipw.IntText(
+        value=int(di_info["previous_uses"] or 0),
+        min=0,
+        max=99,
+        description="Previous uses",
+        layout=width(180),
+        disabled=True,
+    )
+    save_button = ipw.Button(
+        description="Save", button_style="warning", layout=width(180)
+    )
+    if (desc := di_info["dii_description"]) in np_session.Dye.descriptions:
         dye_dropdown.value = desc
-        
+
     def update_display(_):
         dye = np_session.Dye(int(str(dye_id_entry.value)))
         dye_dropdown.value = dye.description
         usage_counter.value = dye.previous_uses
-        first_usage.value = f'{dye.first_use}'
-    dye_id_entry.observe(update_display, 'value')
-    
+        first_usage.value = f"{dye.first_use}"
+
+    dye_id_entry.observe(update_display, "value")
+
     def record_dye_usage():
         dye = np_session.Dye(int(str(dye_id_entry.value)))
         dye.description = dye_dropdown.value
         dye.increment_uses()
-        
+
     def update_di_info():
-        di_info['EndTime'] = npxc.now()
-        di_info['times_dipped'] = str(dipped_counter.value)
-        di_info['dii_description'] = str(dye_dropdown.value)
-        di_info['previous_uses'] = str(usage_counter.value)
-        
+        di_info["EndTime"] = npxc.now()
+        di_info["times_dipped"] = str(dipped_counter.value)
+        di_info["dii_description"] = str(dye_dropdown.value)
+        di_info["previous_uses"] = str(usage_counter.value)
+
     def on_click(b):
         update_di_info()
         record_dye_usage()
         session.platform_json.DiINotes = di_info
-        save_button.description = 'Saved'
-        save_button.button_style = 'success'
-        
+        save_button.description = "Saved"
+        save_button.button_style = "success"
+
     save_button.on_click(on_click)
-    return IPython.display.display(ipw.VBox([
-        dye_id_entry,
-        dipped_counter, dye_dropdown, 
-        usage_counter, first_usage, save_button]))
+    return IPython.display.display(
+        ipw.VBox(
+            [
+                dye_id_entry,
+                dipped_counter,
+                dye_dropdown,
+                usage_counter,
+                first_usage,
+                save_button,
+            ]
+        )
+    )
+
 
 def dye_widget(session_folder: pathlib.Path) -> IPython.display.DisplayHandle | None:
     "Supply a path - saves a JSON file with the dye used in the session and a timestamp."
 
     dict(
-        EndTime=0, StartTime=0, dii_description="DiI", times_dipped=0,
+        EndTime=0,
+        StartTime=0,
+        dii_description="DiI",
+        times_dipped=0,
     )
-        
+
     class DyeRecorder(np_services.JsonRecorder):
-        log_name = f'{session_folder.name}_dye.json'
+        log_name = f"{session_folder.name}_dye.json"
         log_root = session_folder
 
-    dye_dropdown = ipw.Dropdown(options=['DiI', 'DiO'])
-    save_button = ipw.Button(description='Save', button_style='warning')
+    dye_dropdown = ipw.Dropdown(options=["DiI", "DiO"])
+    save_button = ipw.Button(description="Save", button_style="warning")
+
     def on_click(b):
-        DyeRecorder.write(dict(dye=dye_dropdown.value, datetime=datetime.datetime.now(), time=time.time()))
-        save_button.button_style = 'success'
-        save_button.description = 'Saved'
+        DyeRecorder.write(
+            dict(
+                dye=dye_dropdown.value,
+                datetime=datetime.datetime.now(),
+                time=time.time(),
+            )
+        )
+        save_button.button_style = "success"
+        save_button.description = "Saved"
+
     save_button.on_click(on_click)
     return IPython.display.display(ipw.VBox([dye_dropdown, save_button]))
 
-ISICoords = list[dict[Literal['x', 'y', 'z'], float]]
-ISISpaces = dict[Literal['image_space', 'reticle_space'], ISICoords | None]
-ISITargets = dict[Literal['insertion_targets', 'intended_insertion', 'actual_insertion'], ISISpaces]
+
+ISICoords = list[dict[Literal["x", "y", "z"], float]]
+ISISpaces = dict[Literal["image_space", "reticle_space"], ISICoords | None]
+ISITargets = dict[
+    Literal["insertion_targets", "intended_insertion", "actual_insertion"], ISISpaces
+]
+
 
 def isi_targets(
     labtracks_mouse_id: str | int | np_session.LIMS2MouseInfo,
-)-> None | ISITargets: 
-    mouse = np_session.LIMS2MouseInfo(labtracks_mouse_id) if not isinstance(labtracks_mouse_id, np_session.LIMS2MouseInfo) else labtracks_mouse_id
+) -> None | ISITargets:
+    mouse = (
+        np_session.LIMS2MouseInfo(labtracks_mouse_id)
+        if not isinstance(labtracks_mouse_id, np_session.LIMS2MouseInfo)
+        else labtracks_mouse_id
+    )
     if (exp_id := mouse.isi_id) is None:
         return None
-    exps = mouse.isi_info['isi_experiments']
-    isi = [e for e in exps if e['id'] == exp_id]
-    return isi[0]['targets'] if isi else None
-    
+    exps = mouse.isi_info["isi_experiments"]
+    isi = [e for e in exps if e["id"] == exp_id]
+    return isi[0]["targets"] if isi else None
+
+
 def isi_widget(
-    labtracks_mouse_id: str | int | np_session.LIMS2MouseInfo, colormap: bool = False,
+    labtracks_mouse_id: str | int | np_session.LIMS2MouseInfo,
+    colormap: bool = False,
 ) -> IPython.display.DisplayHandle | None:
     """Displays ISI target map from lims (contours only), or colormap overlay if
     `show_colormap = True`."""
@@ -434,13 +539,13 @@ def isi_widget(
         mouse_info = np_session.LIMS2MouseInfo(labtracks_mouse_id)
     else:
         mouse_info = labtracks_mouse_id
-        mouse_info.fetch() # refresh in case targets were updated recently
+        mouse_info.fetch()  # refresh in case targets were updated recently
 
     if colormap:
         key = "isi_image_overlay_path"
     else:
         key = "target_map_image_path"
-    
+
     try:
         lims_path = mouse_info.isi_info[key]
     except ValueError:
@@ -457,30 +562,39 @@ def isi_widget(
         print(f"ISI map found for {mouse_info.np_id}:\n{path}")
         img = PIL.Image.open(path)
         if all_targets := isi_targets(mouse_info):
-            colors = {'insertion_targets': 'red', 'intended_insertion': 'yellow', 'actual_insertion': 'blue'}
+            colors = {
+                "insertion_targets": "red",
+                "intended_insertion": "yellow",
+                "actual_insertion": "blue",
+            }
             for targets, spaces in all_targets.items():
-                coords = spaces['image_space']
+                coords = spaces["image_space"]
                 if coords is None:
                     continue
                 draw = PIL.ImageDraw.Draw(img)
-                draw.line([(_['x'], _['y']) for _ in coords], 
-                        fill=colors[targets],
-                        width=3)
-        else: 
-            logger.debug("No ISI targets found for %r in lims, ISI experiment id %s", mouse_info, mouse_info.isi_id)
+                draw.line(
+                    [(_["x"], _["y"]) for _ in coords], fill=colors[targets], width=3
+                )
+        else:
+            logger.debug(
+                "No ISI targets found for %r in lims, ISI experiment id %s",
+                mouse_info,
+                mouse_info.isi_id,
+            )
         ## displaying img directly no longer works (due to jupyterlab 4.0?)
         # return IPython.display.display(img)
         membuf = io.BytesIO()
-        img.save(membuf, format="png") 
+        img.save(membuf, format="png")
         return IPython.display.display(ipw.VBox([ipw.Image(value=membuf.getvalue())]))
-        
 
 
 def insertion_notes_widget(session: np_session.PipelineSession):
-    
-    probes = 'ABCDEF'
+
+    probes = "ABCDEF"
+
     def probe(_):
-        return f'Probe{_}'
+        return f"Probe{_}"
+
     fields = (
         "FailedToInsert",
         # "ProbeLocationChanged",
@@ -488,86 +602,110 @@ def insertion_notes_widget(session: np_session.PipelineSession):
         # "ProbeBendingElsewhere",
     )
     # "NumAgarInsertions",
-    
+
     def get_notes(_):
-        return session.platform_json.InsertionNotes.get(probe(_), {}).get('Notes', '')
+        return session.platform_json.InsertionNotes.get(probe(_), {}).get("Notes", "")
+
     def get_field(_, field):
         return session.platform_json.InsertionNotes.get(probe(_), {}).get(field, None)
-    
-    def disp_str(s): # split PascalCase fieldname into 'Title case' words
-        matches = re.finditer('.+?(?:(?<=[a-z])(?=[A-Z])|(?<=[A-Z])(?=[A-Z][a-z])|$)', s)
-        return ' '.join([m.group(0) for m in matches]).lower().capitalize()
+
+    def disp_str(s):  # split PascalCase fieldname into 'Title case' words
+        matches = re.finditer(
+            ".+?(?:(?<=[a-z])(?=[A-Z])|(?<=[A-Z])(?=[A-Z][a-z])|$)", s
+        )
+        return " ".join([m.group(0) for m in matches]).lower().capitalize()
+
     def save_str(s):
-        return ''.join([_.capitalize() for _ in s.split(' ')])
-    
+        return "".join([_.capitalize() for _ in s.split(" ")])
+
     def row(*args):
         return ipw.HBox([*args])
+
     def probe_row(p):
-        return row(ipw.Text(value=get_notes(p), placeholder='Insertion notes', description=disp_str(probe(p).strip('Probe ')), layout=ipw.Layout(width='auto', min_width='400px')), *(ipw.Checkbox(value=get_field(p, field), description=disp_str(field)) for field in fields))
-    button = ipw.Button(description="Save", button_style='warning')
+        return row(
+            ipw.Text(
+                value=get_notes(p),
+                placeholder="Insertion notes",
+                description=disp_str(probe(p).strip("Probe ")),
+                layout=ipw.Layout(width="auto", min_width="400px"),
+            ),
+            *(
+                ipw.Checkbox(value=get_field(p, field), description=disp_str(field))
+                for field in fields
+            ),
+        )
+
+    button = ipw.Button(description="Save", button_style="warning")
     console = ipw.Output()
-    
+
     rows = [probe_row(p) for p in probes]
     widget = ipw.VBox([*rows, button, console])
-    
+
     def save(b):
         d = {}
         for letter, row in zip(probes, rows):
             p = d.get(probe(letter), {})
             for widget in row.children:
                 if isinstance(widget, ipw.Text):
-                    p['Notes'] = widget.value
+                    p["Notes"] = widget.value
                 elif isinstance(widget, ipw.Checkbox):
                     p[save_str(widget.description)] = widget.value
                 else:
                     continue
             if p:
-                d[probe(letter)] = p  
-        
-        session.platform_json.InsertionNotes = d 
+                d[probe(letter)] = p
+
+        session.platform_json.InsertionNotes = d
         with console:
-            print('Updated notes')
-        button.button_style = 'success'
-        
+            print("Updated notes")
+        button.button_style = "success"
+
     button.on_click(save)
     return IPython.display.display(widget)
 
 
 def probe_depth_widget(session: np_session.PipelineSession):
-    
-    probes = 'ABCDEF'
-    
+
+    probes = "ABCDEF"
+
     def coords():
         return session.platform_json.manipulator_coordinates
-    
+
     if not coords():
         logger.warning("No photodocs have been captured yet.")
-    
+
     def probe_coords(img):
         return coords().get(img, dict.fromkeys(probes, dict(x=None, y=None, z=None)))
+
     def field_str(s):
-        return '_'.join(s.split(' ')).lower() + '_surface_image' if s else ''
-    
+        return "_".join(s.split(" ")).lower() + "_surface_image" if s else ""
+
     selection = ipw.ToggleButtons(
-    options=[' '.join(_.strip('_surface_image').split('_')).capitalize() for _ in coords().keys()],
-    description='Depth',
-    disabled=False,
-    button_style='', # 'success', 'info', 'warning', 'danger' or ''
-    tooltips=[field_str(_) for _ in coords().keys()],
+        options=[
+            " ".join(_.strip("_surface_image").split("_")).capitalize()
+            for _ in coords().keys()
+        ],
+        description="Depth",
+        disabled=False,
+        button_style="",  # 'success', 'info', 'warning', 'danger' or ''
+        tooltips=[field_str(_) for _ in coords().keys()],
     )
-    
+
     def update(_):
         for probe in probes:
             depth = probe_coords(field_str(selection.value))[probe]["z"]
-            textbox[probe].value = f'{depth:6.1f}' if depth is not None else ''
-            
+            textbox[probe].value = f"{depth:6.1f}" if depth is not None else ""
+
     textbox = {
         probe: ipw.Text(
-        value='', description=probe, disabled=True,
-        layout=ipw.Layout(max_width='150px'),)
+            value="",
+            description=probe,
+            disabled=True,
+            layout=ipw.Layout(max_width="150px"),
+        )
         for probe in probes
     }
-    selection.observe(update, 'value')
+    selection.observe(update, "value")
     update(None)
     widget = ipw.VBox([selection, ipw.HBox([*textbox.values()])])
     return IPython.display.display(widget)
@@ -575,96 +713,106 @@ def probe_depth_widget(session: np_session.PipelineSession):
 
 def photodoc_widget(img_name: str) -> IPython.display.DisplayHandle | None:
     "Captures and displays snapshot from image camera, appending `img_name` to the filename."
-    image = ipw.Image(value=b'', format='png', width='80%', layout=ipw.Layout(visibility='hidden'))
-    widget = ipw.VBox([
-        image,
-        button := ipw.Button(description="Re-capture", button_style='warning'),
-        console := ipw.Output(),
-    ])
-    
+    image = ipw.Image(
+        value=b"", format="png", width="80%", layout=ipw.Layout(visibility="hidden")
+    )
+    widget = ipw.VBox(
+        [
+            image,
+            button := ipw.Button(description="Re-capture", button_style="warning"),
+            console := ipw.Output(),
+        ]
+    )
+
     def capture() -> pathlib.Path:
-        image.value = b''
-        image.layout.visibility = 'hidden'
-        button.button_style = ''
-        button.description = 'Capturing new image...'
+        image.value = b""
+        image.layout.visibility = "hidden"
+        button.button_style = ""
+        button.description = "Capturing new image..."
         button.disabled = True
         return npxc.photodoc(img_name)
-    
+
     def disp(img_path) -> None:
         image.value = img_path.read_bytes()
-        image.layout.visibility = 'visible'
-        button.button_style = 'warning'
-        button.description = 'Re-capture'
+        image.layout.visibility = "visible"
+        button.button_style = "warning"
+        button.description = "Re-capture"
         button.disabled = False
         with console:
             print(img_path)
-            
+
     def capture_and_display(*args):
         disp(capture())
-        
+
     button.on_click(capture_and_display)
-    
-    if (matches := [_ for _ in (np_services.Cam3d.data_files or np_services.ImageMVR.data_files or []) if img_name in _.stem]):
+
+    if matches := [
+        _
+        for _ in (np_services.Cam3d.data_files or np_services.ImageMVR.data_files or [])
+        if img_name in _.stem
+    ]:
         disp(sorted(matches)[-1])
     else:
         capture_and_display()
-    
+
     return IPython.display.display(widget)
 
+
 def probe_targeting_widget(session_folder) -> IPython.display.DisplayHandle | None:
     from np_probe_targets.implant_drawing import CurrentWeek, DRWeeklyTargets
+
     CurrentWeek.display()
     IPython.display.display(DRWeeklyTargets())
-    
+
+
 def quiet_mode_widget() -> IPython.display.DisplayHandle | None:
     """Displays a toggle button that switches logging level INFO <-> DEBUG and
     hides/shows tracebacks.
     """
     debug_mode_toggle = ipw.ToggleButton(
-            value=True,
-            description='Quiet mode is on',
-            disabled=False,
-            button_style='info', # 'success', 'info', 'warning', 'danger' or ''
-            icon='check',
-            tooltip='Quiet mode: tracebacks hidden, logging level set to INFO.',
-        )
-    
+        value=True,
+        description="Quiet mode is on",
+        disabled=False,
+        button_style="info",  # 'success', 'info', 'warning', 'danger' or ''
+        icon="check",
+        tooltip="Quiet mode: tracebacks hidden, logging level set to INFO.",
+    )
+
     def set_debug_mode(value: bool) -> None:
         if value:
             npxc.show_tracebacks()
             for handler in np_logging.getLogger().handlers:
                 if isinstance(handler, logging.StreamHandler):
-                    handler.setLevel('DEBUG')
+                    handler.setLevel("DEBUG")
         else:
             npxc.hide_tracebacks()
             for handler in np_logging.getLogger().handlers:
                 if isinstance(handler, logging.StreamHandler):
-                    handler.setLevel('INFO')
-                
+                    handler.setLevel("INFO")
+
     def on_click(b) -> None:
         if not debug_mode_toggle.value:
             set_debug_mode(True)
-            debug_mode_toggle.description = 'Quiet mode is off'
-            debug_mode_toggle.button_style = ''
-            debug_mode_toggle.icon = 'times'
+            debug_mode_toggle.description = "Quiet mode is off"
+            debug_mode_toggle.button_style = ""
+            debug_mode_toggle.icon = "times"
         else:
             set_debug_mode(False)
-            debug_mode_toggle.description = 'Quiet mode is on'
-            debug_mode_toggle.button_style = 'info'
-            debug_mode_toggle.icon = 'check'
-    
+            debug_mode_toggle.description = "Quiet mode is on"
+            debug_mode_toggle.button_style = "info"
+            debug_mode_toggle.icon = "check"
+
     debug_mode_toggle.observe(on_click)
-     
+
     return IPython.display.display(debug_mode_toggle)
 
 
 def task_select_widget(
     experiment,
 ) -> None:
-    """Select a task name for controlling behavior of TaskControl.
-    """
+    """Select a task name for controlling behavior of TaskControl."""
     experiment.task_name = experiment.preset_task_names[0]
-    
+
     task_dropdown = ipw.Select(
         options=tuple(experiment.preset_task_names),
         description="Presets",
@@ -676,7 +824,7 @@ def task_select_widget(
     )
     console = ipw.Output()
     with console:
-        if last_task:= experiment.mouse.state.get('last_task'):
+        if last_task := experiment.mouse.state.get("last_task"):
             print(f"{experiment.mouse} last task: {last_task}")
 
     def update(change):
@@ -698,8 +846,10 @@ def task_select_widget(
                 task_dropdown.value = None
         with console:
             print(f"Updated task: {experiment.task_name}")
-    task_dropdown.observe(update, names='value')
-    task_input_box.observe(update, names='value')
 
-    IPython.display.display(ipw.VBox([ipw.HBox([task_dropdown, task_input_box]), console]))
-    
+    task_dropdown.observe(update, names="value")
+    task_input_box.observe(update, names="value")
+
+    IPython.display.display(
+        ipw.VBox([ipw.HBox([task_dropdown, task_input_box]), console])
+    )