try except statements added to IMRO uploader to prevent app crashing

Author: m-beau

channelmap_generator/__init__.py

@@ -7,6 +7,6 @@
 
 from .backend import *
 
-__version__ = "0.3.5"
+__version__ = "0.3.6"
 __author__ = "Maxime Beau"
 __email__ = "maxime@princeton.edu"

channelmap_generator/gui/gui.py

@@ -688,14 +688,18 @@ def apply_uploaded_imro(self):
             imro_file_content = imro_file_content.decode("utf-8")
         imro_list = imro.parse_imro_file(imro_file_content.strip())
 
-        (imro_electrodes,
-         self.probe_type,  # probe_type value is a monitored param - simply setting its value will update the plot
-         self.probe_subtype,
-         self.reference_id,
-         self.ap_gain_input.value,
-         self.lf_gain_input.value,
-         self.hardware_hp_filter_on,
-        ) = imro.parse_imro_list(imro_list)
+        try:
+            (imro_electrodes,
+            self.probe_type,  # probe_type value is a monitored param - simply setting its value will update the plot
+            self.probe_subtype,
+            self.reference_id,
+            self.ap_gain_input.value,
+            self.lf_gain_input.value,
+            self.hardware_hp_filter_on,
+            ) = imro.parse_imro_list(imro_list)
+        except:
+            pn.state.notifications.error("Failed to parse uploaded imro file.")
+            return
 
         self.electrodes.clear_selection()
         for shank_id, electrode_id in imro_electrodes:

channelmap_generator/utils/imro.py

@@ -181,17 +181,21 @@ def parse_imro_list(imro_list):
 def ref_id_to_string(reference_id, probe_subtype):
     ref_map = {v: k for k, v in REF_ELECTRODES[probe_subtype].items() if k != "Join Tips"}
     if reference_id not in ref_map:
-        pn.state.notifications.error(f"Unexpected reference id {reference_id} for probe subtype {probe_subtype}!",
+        error_message = f"Unexpected reference id {reference_id} for probe subtype {probe_subtype}!"
+        pn.state.notifications.error(error_message,
                                     duration=10_000)
-    assert reference_id in ref_map,\
-            f"Unexpected reference id {reference_id} for probe subtype {probe_subtype}!"
+        raise AssertionError(error_message)
     return ref_map[reference_id]
 
+
 def check_entry_elements(n_expected_elements, entry, probe_type):
     n_entry_elements = len(entry)
     if n_entry_elements != n_expected_elements:
-        pn.state.notifications.error(f"Corrupt .imro file - IMRO table entries for {probe_type} probes should have {n_expected_elements} elements, not {n_entry_elements}.",
+        error_message = f"Corrupt .imro file - IMRO table entries for {probe_type} probes should have {n_expected_elements} elements, not {n_entry_elements}."
+        pn.state.notifications.error(error_message,
                                 duration=10_000)
+        raise AssertionError(error_message)
+
 
 def generate_imro_channelmap(
     probe_type,