celldetective 1.5.0b7__py3-none-any.whl → 1.5.0b9__py3-none-any.whl

celldetective/gui/settings/_settings_event_model_training.py

@@ -417,8 +417,7 @@ class SettingsEventDetectionModelTraining(CelldetectiveSettingsPanel):
         self.signals = ["--"] + num_cols_pairs + num_cols_reference + num_cols_neighbor
 
         for cb in self.ch_norm.channel_cbs:
-            cb.clear()
-            cb.addItems(self.signals)
+            self.ch_norm.add_items_truncated(cb, self.signals)
 
     def fill_available_neighborhoods(self):
 
@@ -541,15 +540,36 @@ class SettingsEventDetectionModelTraining(CelldetectiveSettingsPanel):
         self.model_length_slider.setValue(int(signal_length))
         self.model_length_slider.setEnabled(False)
 
-        for c, cb in zip(channels, self.ch_norm.channel_cbs):
-            index = cb.findText(c)
+        try:
+            norm_perc = data["normalization_percentile"]
+            if isinstance(norm_perc, bool):
+                norm_perc = [norm_perc] * len(channels)
+            norm_val = data["normalization_values"]
+            if len(norm_val) == 2 and isinstance(norm_val[0], float):
+                norm_val = [norm_val] * len(channels)
+            norm_clip = data["normalization_clip"]
+            if isinstance(norm_clip, bool):
+                norm_clip = [norm_clip] * len(channels)
+        except Exception:
+            norm_perc = [True] * len(channels)
+            norm_val = [[0.1, 99.99]] * len(channels)
+            norm_clip = [False] * len(channels)
+
+        for k, (c, cb) in enumerate(zip(channels, self.ch_norm.channel_cbs)):
+            index = cb.findData(c)
             cb.setCurrentIndex(index)
 
-        if len(channels) < len(self.ch_norm.channel_cbs):
-            for k in range(len(self.ch_norm.channel_cbs) - len(channels)):
-                self.ch_norm.channel_cbs[len(channels) + k].setCurrentIndex(0)
-                self.ch_norm.channel_cbs[len(channels) + k].setEnabled(False)
-        self.ch_norm.add_col_btn.setEnabled(False)
+            # Set normalization mode
+            if self.ch_norm.normalization_mode[k] != norm_perc[k]:
+                self.ch_norm.switch_normalization_mode(k)
+
+            # Set clipping mode
+            if self.ch_norm.clip_option[k] != norm_clip[k]:
+                self.ch_norm.switch_clipping_mode(k)
+
+            # Set normalization values
+            self.ch_norm.normalization_min_value_le[k].setText(str(norm_val[k][0]))
+            self.ch_norm.normalization_max_value_le[k].setText(str(norm_val[k][1]))
 
     def adjust_scroll_area(self):
         """
@@ -581,7 +601,7 @@ class SettingsEventDetectionModelTraining(CelldetectiveSettingsPanel):
 
         channels = []
         for i in range(len(self.ch_norm.channel_cbs)):
-            channels.append(self.ch_norm.channel_cbs[i].currentText())
+            channels.append(self.ch_norm.channel_cbs[i].currentData())
 
         slots_to_keep = np.where(np.array(channels) != "--")[0]
         while "--" in channels:
@@ -750,10 +770,8 @@ class SettingsEventDetectionModelTraining(CelldetectiveSettingsPanel):
 
             # Assuming signal/event models directory. Verifying path would be better but this fits pattern.
             # If exact attribute unknown, use os.path.dirname logic or config methods.
-            # Safe bet: self.parent_window.signal_models_dir based on refresh call.
-            model_path = os.path.join(
-                self.parent_window.signal_models_dir, self.modelname_le.text()
-            )
+            # Safe bet: self.signal_models_dir based on init.
+            model_path = os.path.join(self.signal_models_dir, self.modelname_le.text())
             if os.path.exists(model_path):
                 time.sleep(0.5)
                 shutil.rmtree(model_path)

celldetective/gui/settings/_settings_segmentation_model_training.py

@@ -355,7 +355,7 @@ class SettingsSegmentationModelTraining(CelldetectiveSettingsPanel):
             not current_name in models
             and not self.spatial_calib_le.text() == ""
             and not np.all(
-                [cb.currentText() == "--" for cb in self.ch_norm.channel_cbs]
+                [cb.currentData() == "--" for cb in self.ch_norm.channel_cbs]
             )
         ):
             self.submit_btn.setEnabled(True)
@@ -370,7 +370,7 @@ class SettingsSegmentationModelTraining(CelldetectiveSettingsPanel):
                 self.submit_warning.setText(
                     "Please provide a valid spatial calibration..."
                 )
-            elif np.all([cb.currentText() == "--" for cb in self.ch_norm.channel_cbs]):
+            elif np.all([cb.currentData() == "--" for cb in self.ch_norm.channel_cbs]):
                 self.submit_warning.setText("Please provide valid channels...")
 
     def rescale_slider(self):
@@ -459,7 +459,7 @@ class SettingsSegmentationModelTraining(CelldetectiveSettingsPanel):
 
         for k in range(len(self.diamWidget.cellpose_channel_cb)):
             ch = self.diamWidget.cellpose_channel_cb[k].currentText()
-            idx = self.ch_norm.channel_cbs[k].findText(ch)
+            idx = self.ch_norm.channel_cbs[k].findData(ch)
             self.ch_norm.channel_cbs[k].setCurrentIndex(idx)
 
         self.diamWidget.close()
@@ -561,7 +561,7 @@ class SettingsSegmentationModelTraining(CelldetectiveSettingsPanel):
             self.cellpose_model.setChecked(True)
 
         for c, cb in zip(channels, self.ch_norm.channel_cbs):
-            index = cb.findText(c)
+            index = cb.findData(c)
             cb.setCurrentIndex(index)
 
         for i in range(len(channels)):
@@ -622,7 +622,7 @@ class SettingsSegmentationModelTraining(CelldetectiveSettingsPanel):
 
         channels = []
         for i in range(len(self.ch_norm.channel_cbs)):
-            channels.append(self.ch_norm.channel_cbs[i].currentText())
+            channels.append(self.ch_norm.channel_cbs[i].currentData())
 
         slots_to_keep = np.where(np.array(channels) != "--")[0]
         while "--" in channels:

celldetective/gui/settings/_settings_signal_annotator.py

@@ -100,11 +100,6 @@ class SettingsSignalAnnotator(CelldetectiveSettingsPanel):
         hbox_frac.addWidget(self.fraction_slider, 80)
         sub_layout.addLayout(hbox_frac)
 
-        hbox_interval = QHBoxLayout()
-        hbox_interval.addWidget(self._interval_lbl, 20)
-        hbox_interval.addWidget(self.interval_slider, 80)
-        sub_layout.addLayout(hbox_interval)
-
         self._layout.addLayout(sub_layout)
 
         self._layout.addWidget(self.submit_btn)
@@ -118,7 +113,6 @@ class SettingsSignalAnnotator(CelldetectiveSettingsPanel):
 
         self._modality_lbl = QLabel("Modality: ")
         self._fraction_lbl = QLabel("fraction: ")
-        self._interval_lbl = QLabel("interval [ms]: ")
 
         self.gs_btn = QRadioButton("grayscale")
         self.gs_btn.setChecked(True)
@@ -173,14 +167,6 @@ class SettingsSignalAnnotator(CelldetectiveSettingsPanel):
         self.fraction_slider.setRange(0.1, 1)
         self.fraction_slider.setValue(0.25)
 
-        self.interval_slider = QLabeledSlider()
-        self.interval_slider.setSingleStep(1)
-        self.interval_slider.setTickInterval(1)
-        self.interval_slider.setSingleStep(1)
-        self.interval_slider.setOrientation(Qt.Horizontal)
-        self.interval_slider.setRange(1, 1000)
-        self.interval_slider.setValue(1)
-
     def enable_channels(self):
         """
         Enable three channels when RGB mode is checked.
@@ -255,7 +241,6 @@ class SettingsSignalAnnotator(CelldetectiveSettingsPanel):
             "rgb_mode": self.rgb_btn.isChecked(),
             "percentile_mode": self.percentile_mode,
             "fraction": float(self.fraction_slider.value()),
-            "interval": int(self.interval_slider.value()),
             "log": self.log_option,
         }
         max_i = 3 if self.rgb_btn.isChecked() else 1
@@ -322,10 +307,6 @@ class SettingsSignalAnnotator(CelldetectiveSettingsPanel):
                     fraction = instructions["fraction"]
                     self.fraction_slider.setValue(fraction)
 
-                if "interval" in instructions:
-                    interval = instructions["interval"]
-                    self.interval_slider.setValue(interval)
-
                 if "log" in instructions:
                     self.log_option = not instructions["log"]
                     self.switch_to_log()

celldetective/gui/survival_ui.py

@@ -19,11 +19,13 @@ from celldetective.utils.data_cleaning import extract_cols_from_table_list
 from celldetective.utils.parsing import _extract_labels_from_config
 from celldetective.utils.data_loaders import load_experiment_tables
 import numpy as np
+import pandas as pd
 import os
 import matplotlib.pyplot as plt
 
 plt.rcParams["svg.fonttype"] = "none"
 from glob import glob
+from celldetective import get_logger
 from celldetective.gui.base.styles import Styles
 from celldetective.gui.base.components import CelldetectiveWidget
 from matplotlib import colormaps
@@ -32,6 +34,8 @@ from celldetective.relative_measurements import expand_pair_table
 import matplotlib.cm
 from celldetective.neighborhood import extract_neighborhood_in_pair_table
 
+logger = get_logger(__name__)
+
 
 class ConfigSurvival(CelldetectiveWidget):
     """
@@ -225,7 +229,9 @@ class ConfigSurvival(CelldetectiveWidget):
                 )
             )
             if not tables_pairs:
-                print("No pair table found... please compute the pair measurements...")
+                logger.warning(
+                    "No pair table found. Please compute the pair measurements."
+                )
                 return None
             self.cols_pairs = extract_cols_from_table_list(tables_pairs)
 
@@ -264,7 +270,7 @@ class ConfigSurvival(CelldetectiveWidget):
                     and str(self.population_neigh) not in c
                 ]
 
-            print(f"{self.neighborhood_keys=}")
+            logger.debug(f"Neighborhood keys: {self.neighborhood_keys}")
 
             time_idx = np.array(
                 [s.startswith("t_") or s.startswith("t0") for s in self.cols_pairs]
@@ -281,8 +287,8 @@ class ConfigSurvival(CelldetectiveWidget):
 
             try:
                 time_columns = list(self.all_columns[time_idx])
-            except:
-                print("no column starts with t")
+            except (IndexError, KeyError):
+                logger.warning("No column starts with 't_' for time reference.")
                 self.auto_close = True
                 return None
 
@@ -295,8 +301,15 @@ class ConfigSurvival(CelldetectiveWidget):
 
     def process_survival(self):
 
+        # Validate that reference time and time of interest are different
+        time_of_reference = self.cbs[1].currentText()
+        time_of_interest = self.cbs[2].currentText()
+        if time_of_reference == time_of_interest:
+            generic_message("Reference time and time of interest must be different.")
+            return None
+
         self.FrameToMin = float(self.time_calibration_le.text().replace(",", "."))
-        self.time_of_interest = self.cbs[2].currentText()
+        self.time_of_interest = time_of_interest
         if self.time_of_interest == "t0":
             self.class_of_interest = "class"
         elif self.time_of_interest.startswith("t0"):
@@ -312,8 +325,22 @@ class ConfigSurvival(CelldetectiveWidget):
                 query_text = self.query_le.text()
                 if query_text != "":
                     self.df = self.df.query(query_text)
+            except pd.errors.UndefinedVariableError as e:
+                logger.warning(f"Query failed - undefined variable: {e}")
+                generic_message(
+                    f"Query error: column not found.\n{e}\n\nPlease check your column names."
+                )
+                return None
+            except SyntaxError as e:
+                logger.warning(f"Query failed - syntax error: {e}")
+                generic_message(
+                    f"Query syntax error: {e}\n\nCheck your query format (e.g., 'column > 100')."
+                )
+                return None
             except Exception as e:
-                print(e, " The query is misunderstood and will not be applied...")
+                logger.warning(f"Query not applied: {e}")
+                generic_message(f"Query could not be applied: {e}")
+                return None
 
             self.interpret_pos_location()
 
@@ -391,11 +418,12 @@ class ConfigSurvival(CelldetectiveWidget):
                     / self.FrameToMin
                 )
                 if not 0 < cut_observation_time <= (self.df["FRAME"].max()):
-                    print("Invalid cut time (larger than movie length)... Not applied.")
+                    logger.warning(
+                        "Invalid cut time (larger than movie length). Not applied."
+                    )
                     cut_observation_time = None
         except Exception as e:
-            print(f"{e=}")
-            pass
+            logger.debug(f"Cut time parsing error: {e}")
 
         pairs = False
         if self.neighborhood_keys is not None:
@@ -403,7 +431,7 @@ class ConfigSurvival(CelldetectiveWidget):
 
         # Per position survival
         for block, movie_group in self.df.groupby(["well", "position"]):
-            print(f"{block=}")
+            logger.debug(f"Processing block: {block}")
             ks = compute_survival(
                 movie_group,
                 self.class_of_interest,
@@ -413,7 +441,7 @@ class ConfigSurvival(CelldetectiveWidget):
                 cut_observation_time=cut_observation_time,
                 pairs=pairs,
             )
-            print(f"{ks=}")
+            logger.debug(f"Survival fit result: {ks}")
             if ks is not None:
                 self.df_pos_info.loc[
                     self.df_pos_info["pos_path"] == block[1], "survival_fit"

celldetective/gui/tableUI.py

@@ -21,8 +21,13 @@ from celldetective.gui.gui_utils import (
 )
 from celldetective.gui.base.figure_canvas import FigureCanvas
 from celldetective.gui.base.utils import center_window
-from celldetective.gui.table_ops._maths import DifferentiateColWidget, OperationOnColsWidget, CalibrateColWidget, \
-    AbsColWidget, LogColWidget
+from celldetective.gui.table_ops._maths import (
+    DifferentiateColWidget,
+    OperationOnColsWidget,
+    CalibrateColWidget,
+    AbsColWidget,
+    LogColWidget,
+)
 from celldetective.gui.table_ops._merge_one_hot import MergeOneHotWidget
 from celldetective.gui.table_ops._query_table import QueryWidget
 from celldetective.gui.table_ops._rename_col import RenameColWidget
@@ -55,7 +60,7 @@ class PivotTableUI(CelldetectiveWidget):
         self.mode = mode
 
         self.setWindowTitle(title)
-        print("tab to show: ", self.data)
+        logger.debug(f"Pivot table to show: {self.data.shape}")
 
         self.table = QTableView(self)
 
@@ -219,9 +224,35 @@ class TableUI(CelldetectiveMainWindow):
 
         try:
             self.fig.tight_layout()
-        except:
+        except AttributeError:
+            # fig not yet created
             pass
 
+    def _get_selected_columns(self, max_cols=None):
+        """
+        Get selected column names from the table view.
+
+        Parameters
+        ----------
+        max_cols : int, optional
+            Maximum number of columns to return. Returns all if None.
+
+        Returns
+        -------
+        list
+            List of selected column names.
+        """
+        x = self.table_view.selectedIndexes()
+        col_idx = np.unique(np.array([l.column() for l in x]))
+        cols = np.array(list(self.data.columns))
+        result = []
+        if len(col_idx) > 0:
+            for i in col_idx:
+                result.append(str(cols[i]))
+        if max_cols is not None:
+            return result[:max_cols]
+        return result
+
     def _create_actions(self):
 
         self.save_as = QAction("&Save as...", self)
@@ -540,8 +571,8 @@ class TableUI(CelldetectiveMainWindow):
         self.renameWidget.show()
 
     def save_as_csv_inplace_per_pos(self):
-
-        print("Saving each table in its respective position folder...")
+        """Save each position's table in its respective folder."""
+        logger.info("Saving each table in its respective position folder...")
         for pos, pos_group in self.data.groupby(["position"]):
             invalid_cols = [
                 c for c in list(pos_group.columns) if c.startswith("Unnamed")
@@ -555,26 +586,12 @@ class TableUI(CelldetectiveMainWindow):
                 ),
                 index=False,
             )
-        print("Done...")
+        logger.info("Done saving tables.")
 
     def divide_signals(self):
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_col1 = str(cols[col_idx[0]])
-                if len(col_idx) > 1:
-                    selected_col2 = str(cols[col_idx[1]])
-                else:
-                    selected_col2 = None
-            else:
-                selected_col1 = None
-                selected_col2 = None
-        else:
-            selected_col1 = None
-            selected_col2 = None
+        selected = self._get_selected_columns(max_cols=2)
+        selected_col1 = selected[0] if len(selected) > 0 else None
+        selected_col2 = selected[1] if len(selected) > 1 else None
 
         self.divWidget = OperationOnColsWidget(
             self, column1=selected_col1, column2=selected_col2, operation="divide"
@@ -582,23 +599,9 @@ class TableUI(CelldetectiveMainWindow):
         self.divWidget.show()
 
     def multiply_signals(self):
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_col1 = str(cols[col_idx[0]])
-                if len(col_idx) > 1:
-                    selected_col2 = str(cols[col_idx[1]])
-                else:
-                    selected_col2 = None
-            else:
-                selected_col1 = None
-                selected_col2 = None
-        else:
-            selected_col1 = None
-            selected_col2 = None
+        selected = self._get_selected_columns(max_cols=2)
+        selected_col1 = selected[0] if len(selected) > 0 else None
+        selected_col2 = selected[1] if len(selected) > 1 else None
 
         self.mulWidget = OperationOnColsWidget(
             self, column1=selected_col1, column2=selected_col2, operation="multiply"
@@ -606,23 +609,9 @@ class TableUI(CelldetectiveMainWindow):
         self.mulWidget.show()
 
     def add_signals(self):
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_col1 = str(cols[col_idx[0]])
-                if len(col_idx) > 1:
-                    selected_col2 = str(cols[col_idx[1]])
-                else:
-                    selected_col2 = None
-            else:
-                selected_col1 = None
-                selected_col2 = None
-        else:
-            selected_col1 = None
-            selected_col2 = None
+        selected = self._get_selected_columns(max_cols=2)
+        selected_col1 = selected[0] if len(selected) > 0 else None
+        selected_col2 = selected[1] if len(selected) > 1 else None
 
         self.addiWidget = OperationOnColsWidget(
             self, column1=selected_col1, column2=selected_col2, operation="add"
@@ -630,23 +619,9 @@ class TableUI(CelldetectiveMainWindow):
         self.addiWidget.show()
 
     def subtract_signals(self):
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_col1 = str(cols[col_idx[0]])
-                if len(col_idx) > 1:
-                    selected_col2 = str(cols[col_idx[1]])
-                else:
-                    selected_col2 = None
-            else:
-                selected_col1 = None
-                selected_col2 = None
-        else:
-            selected_col1 = None
-            selected_col2 = None
+        selected = self._get_selected_columns(max_cols=2)
+        selected_col1 = selected[0] if len(selected) > 0 else None
+        selected_col2 = selected[1] if len(selected) > 1 else None
 
         self.subWidget = OperationOnColsWidget(
             self, column1=selected_col1, column2=selected_col2, operation="subtract"
@@ -654,56 +629,22 @@ class TableUI(CelldetectiveMainWindow):
         self.subWidget.show()
 
     def differenciate_selected_feature(self):
-
-        # check only one col selected and assert is numerical
-        # open widget to select window parameters, directionality
-        # create new col
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_col = str(cols[col_idx[0]])
-            else:
-                selected_col = None
-        else:
-            selected_col = None
-
+        """Open widget to differentiate the selected column."""
+        selected = self._get_selected_columns(max_cols=1)
+        selected_col = selected[0] if selected else None
         self.diffWidget = DifferentiateColWidget(self, selected_col)
         self.diffWidget.show()
 
     def take_log_of_selected_feature(self):
-
-        # check only one col selected and assert is numerical
-        # open widget to select window parameters, directionality
-        # create new col
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_col = str(cols[col_idx[0]])
-            else:
-                selected_col = None
-        else:
-            selected_col = None
-
+        """Open widget to take log of the selected column."""
+        selected = self._get_selected_columns(max_cols=1)
+        selected_col = selected[0] if selected else None
         self.LogWidget = LogColWidget(self, selected_col)
         self.LogWidget.show()
 
     def merge_classification_features(self):
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-
-        col_selection = []
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_cols = cols[col_idx]
-                col_selection.extend(selected_cols)
+        """Open widget to merge selected classification columns."""
+        col_selection = self._get_selected_columns()
 
         # Lazy load MergeGroupWidget
         from celldetective.gui.table_ops._merge_groups import MergeGroupWidget
@@ -712,38 +653,16 @@ class TableUI(CelldetectiveMainWindow):
         self.merge_classification_widget.show()
 
     def calibrate_selected_feature(self):
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_col = str(cols[col_idx[0]])
-            else:
-                selected_col = None
-        else:
-            selected_col = None
-
+        """Open widget to calibrate the selected column."""
+        selected = self._get_selected_columns(max_cols=1)
+        selected_col = selected[0] if selected else None
         self.calWidget = CalibrateColWidget(self, selected_col)
         self.calWidget.show()
 
     def take_abs_of_selected_feature(self):
-
-        # check only one col selected and assert is numerical
-        # open widget to select window parameters, directionality
-        # create new col
-
-        x = self.table_view.selectedIndexes()
-        col_idx = np.unique(np.array([l.column() for l in x]))
-        if isinstance(col_idx, (list, np.ndarray)):
-            cols = np.array(list(self.data.columns))
-            if len(col_idx) > 0:
-                selected_col = str(cols[col_idx[0]])
-            else:
-                selected_col = None
-        else:
-            selected_col = None
-
+        """Open widget to take absolute value of the selected column."""
+        selected = self._get_selected_columns(max_cols=1)
+        selected_col = selected[0] if selected else None
         self.absWidget = AbsColWidget(self, selected_col)
         self.absWidget.show()
 
@@ -991,7 +910,8 @@ class TableUI(CelldetectiveMainWindow):
             y = column_names[unique_cols]
             idx = self.y_cb.findText(y)
             self.y_cb.setCurrentIndex(idx)
-        except:
+        except (IndexError, KeyError):
+            # No column selected or invalid selection
             pass
 
         hbox = QHBoxLayout()
@@ -1018,7 +938,8 @@ class TableUI(CelldetectiveMainWindow):
             if hasattr(matplotlib.cm, str(cm).lower()):
                 try:
                     self.cmap_cb.addColormap(cm.lower())
-                except:
+                except Exception:
+                    # Some colormaps may fail to add
                     pass
 
         hbox = QHBoxLayout()
@@ -1057,7 +978,7 @@ class TableUI(CelldetectiveMainWindow):
                 cmap(i / len(self.data[self.hue_variable].unique()))
                 for i in range(len(self.data[self.hue_variable].unique()))
             ]
-        except:
+        except (KeyError, ZeroDivisionError):
             colors = None
 
         if self.hue_cb.currentText() == "--":