simba-uw-tf-dev 4.5.8__py3-none-any.whl → 4.7.1__py3-none-any.whl

simba/third_party_label_appenders/transform/coco_keypoints_to_yolo.py

@@ -54,15 +54,15 @@ class COCOKeypoints2Yolo:
     :return: None
 
     :example:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:\cvat_annotations\frames\coco_keypoints_1\s1\annotations\s1.json", img_dir=r"D:\cvat_annotations\frames\simon", save_dir=r"D:\cvat_annotations\frames\yolo_keypoints", clahe=True)
+    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:/cvat_annotations/frames/coco_keypoints_1/s1/annotations/s1.json", img_dir=r"D:/cvat_annotations/frames/simon", save_dir=r"D:/cvat_annotations/frames/yolo_keypoints", clahe=True)
     >>> runner.run()
 
     :example II:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:\cvat_annotations\frames\coco_keypoints_1\merged.json", img_dir=r"D:\cvat_annotations\frames", save_dir=r"D:\cvat_annotations\frames\yolo", clahe=False)
+    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:/cvat_annotations/frames/coco_keypoints_1/merged.json", img_dir=r"D:/cvat_annotations/frames", save_dir=r"D:/cvat_annotations/frames/yolo", clahe=False)
     >>> runner.run()
 
     :example III:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"E:\netholabs_videos\mosaics\subset\to_annotate\2d_mosaic_batch_1.json", img_dir=r"E:\netholabs_videos\mosaics\subset\to_annotate", save_dir=r"E:\netholabs_videos\mosaics\yolo_mdl", clahe=False)
+    >>> runner = COCOKeypoints2Yolo(coco_path=r"E:/netholabs_videos/mosaics/subset/to_annotate/2d_mosaic_batch_1.json", img_dir=r"E:/netholabs_videos/mosaics/subset/to_annotate", save_dir=r"E:/netholabs_videos/mosaics/yolo_mdl", clahe=False)
     >>> runner.run()
 
      :references:

simba/third_party_label_appenders/transform/coco_keypoints_to_yolo_bbox.py

@@ -58,11 +58,11 @@ class COCOKeypoints2YoloBbox:
     :return: None
 
     :example:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:\cvat_annotations\frames\coco_keypoints_1\s1\annotations\s1.json", img_dir=r"D:\cvat_annotations\frames\simon", save_dir=r"D:\cvat_annotations\frames\yolo_keypoints", clahe=True)
+    >>> runner = COCOKeypoints2YoloBbox(coco_path=r"D:/cvat_annotations/frames/coco_keypoints_1/s1/annotations/s1.json", img_dir=r"D:/cvat_annotations/frames/simon", save_dir=r"D:/cvat_annotations/frames/yolo_keypoints", clahe=True)
     >>> runner.run()
 
     :example II:
-    >>> runner = COCOKeypoints2Yolo(coco_path=r"D:\cvat_annotations\frames\coco_keypoints_1\merged.json", img_dir=r"D:\cvat_annotations\frames", save_dir=r"D:\cvat_annotations\frames\yolo", clahe=False)
+    >>> runner = COCOKeypoints2YoloBbox(coco_path=r"D:/cvat_annotations/frames/coco_keypoints_1/merged.json", img_dir=r"D:/cvat_annotations/frames", save_dir=r"D:/cvat_annotations/frames/yolo", clahe=False)
     >>> runner.run()
 
      :references:

simba/ui/pop_ups/clf_add_remove_print_pop_up.py

@@ -7,21 +7,22 @@ from typing import Union
 from simba.mixins.config_reader import ConfigReader
 from simba.mixins.pop_up_mixin import PopUpMixin
 from simba.pose_processors.pose_reset import PoseResetter
-from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, DropDownMenu,
-                                        Entry_Box, FileSelect, SimbaButton,
+from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, Entry_Box,
+                                        FileSelect, SimbaButton, SimBADropDown,
                                         TwoOptionQuestionPopUp)
 from simba.utils.checks import check_str
 from simba.utils.enums import ConfigKey, Keys, Links
-from simba.utils.errors import DuplicationError
+from simba.utils.errors import DuplicationError, NoDataError
 from simba.utils.printing import stdout_success, stdout_trash
 from simba.utils.read_write import tabulate_clf_info
 
 
 class AddClfPopUp(PopUpMixin, ConfigReader):
-    def __init__(self, config_path: Union[str, os.PathLike]):
-        PopUpMixin.__init__(self, config_path=config_path, title="ADD CLASSIFIER")
+    def __init__(self,
+                 config_path: Union[str, os.PathLike]):
+        PopUpMixin.__init__(self, config_path=config_path, title="ADD CLASSIFIER", icon='plus')
         ConfigReader.__init__(self, config_path=config_path, read_video_info=False)
-        self.clf_eb = Entry_Box(self.main_frm, "CLASSIFIER NAME:", "15")
+        self.clf_eb = Entry_Box(parent=self.main_frm, fileDescription="CLASSIFIER NAME:", labelwidth=25, entry_box_width=30, justify='center', img='decision_tree_small')
         add_btn = SimbaButton(parent=self.main_frm, txt="ADD CLASSIFIER", cmd=self.run, img='rocket')
         self.clf_eb.grid(row=0, column=0, sticky=NW)
         add_btn.grid(row=1, column=0, sticky=NW)
@@ -43,36 +44,42 @@ class AddClfPopUp(PopUpMixin, ConfigReader):
 
 class RemoveAClassifierPopUp(PopUpMixin, ConfigReader):
     def __init__(self, config_path: Union[str, os.PathLike]):
-        PopUpMixin.__init__(self, title="Warning: Remove classifier(s) settings")
+
         ConfigReader.__init__(self, config_path=config_path, read_video_info=False)
+        if not isinstance(self.clf_names, (list, tuple)) or len(self.clf_names) < 1:
+            raise NoDataError(msg='The SimBA project has no classifiers: Cannot remove a classifier.', source=self.__class__.__name__)
+        PopUpMixin.__init__(self, title="WARNING: REMOVE CLASSIFIER", icon='trash_red')
         self.remove_clf_frm = CreateLabelFrameWithIcon( parent=self.main_frm, header="SELECT A CLASSIFIER TO REMOVE", icon_name=Keys.DOCUMENTATION.value, icon_link=Links.REMOVE_CLF.value)
-        self.clf_dropdown = DropDownMenu(self.remove_clf_frm, "Classifier", self.clf_names, "12")
-        self.clf_dropdown.setChoices(self.clf_names[0])
-
+        self.clf_dropdown = SimBADropDown(parent=self.remove_clf_frm, dropdown_options=self.clf_names, label_width=20, dropdown_width=40, label='CLASSIFIER:', value=self.clf_names[0])
         run_btn = SimbaButton(parent=self.main_frm, txt="REMOVE CLASSIFIER", cmd=self.run, img='trash')
-        self.remove_clf_frm.grid(row=0, sticky=W)
-        self.clf_dropdown.grid(row=0, sticky=W)
+        self.remove_clf_frm.grid(row=0, column=0, sticky=W)
+        self.clf_dropdown.grid(row=0, column=0, sticky=W)
         run_btn.grid(row=1, pady=10)
 
     def run(self):
-        for i in range(len(self.clf_names)):
-            self.config.remove_option("SML settings", f"model_path_{i+1}")
-            self.config.remove_option("SML settings", f"target_name_{i+1}")
-            self.config.remove_option("threshold_settings", f"threshold_{i+1}")
-            self.config.remove_option("Minimum_bout_lengths", f"min_bout_{i+1}")
-        self.clf_names.remove(self.clf_dropdown.getChoices())
-        self.config.set("SML settings", "no_targets", str(len(self.clf_names)))
-
-        for clf_cnt, clf_name in enumerate(self.clf_names):
-            self.config.set("SML settings", f"model_path_{clf_cnt + 1}", "")
-            self.config.set("SML settings", f"target_name_{clf_cnt + 1}", clf_name)
-            self.config.set("threshold_settings", f"threshold_{clf_cnt + 1}", "None")
-            self.config.set("Minimum_bout_lengths", f"min_bout_{clf_cnt + 1}", "None")
-
-        with open(self.config_path, "w") as f:
-            self.config.write(f)
-
-        stdout_trash(msg=f"{self.clf_dropdown.getChoices()} classifier removed from SimBA project.", source=self.__class__.__name__)
+        clf_to_remove = self.clf_dropdown.get_value()
+        question = TwoOptionQuestionPopUp(title="WARNING!", question=f"Do you want to remove the {clf_to_remove} \nclassifier from the SimBA project?", option_one="YES", option_two="NO")
+        if question.selected_option == "YES":
+            for i in range(len(self.clf_names)):
+                self.config.remove_option("SML settings", f"model_path_{i+1}")
+                self.config.remove_option("SML settings", f"target_name_{i+1}")
+                self.config.remove_option("threshold_settings", f"threshold_{i+1}")
+                self.config.remove_option("Minimum_bout_lengths", f"min_bout_{i+1}")
+            self.clf_names.remove(self.clf_dropdown.getChoices())
+            self.config.set("SML settings", "no_targets", str(len(self.clf_names)))
+
+            for clf_cnt, clf_name in enumerate(self.clf_names):
+                self.config.set("SML settings", f"model_path_{clf_cnt + 1}", "")
+                self.config.set("SML settings", f"target_name_{clf_cnt + 1}", clf_name)
+                self.config.set("threshold_settings", f"threshold_{clf_cnt + 1}", "None")
+                self.config.set("Minimum_bout_lengths", f"min_bout_{clf_cnt + 1}", "None")
+
+            with open(self.config_path, "w") as f:
+                self.config.write(f)
+
+            stdout_trash(msg=f"{self.clf_dropdown.getChoices()} classifier removed from SimBA project.", source=self.__class__.__name__)
+        else:
+            pass
 
 
 # _ = RemoveAClassifierPopUp(config_path='/Users/simon/Desktop/envs/troubleshooting/Two_animals_16bps/project_folder/project_config.ini')

simba/ui/pop_ups/clf_plot_pop_up.py

@@ -49,8 +49,8 @@ class SklearnVisualizationPopUp(PopUpMixin, ConfigReader):
         pose_palettes = Options.PALETTE_OPTIONS_CATEGORICAL.value + Options.PALETTE_OPTIONS.value
         PopUpMixin.__init__(self, title="VISUALIZE CLASSIFICATION (SKLEARN) RESULTS", icon='photos')
         bp_threshold_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="BODY-PART VISUALIZATION THRESHOLD", icon_name='threshold', icon_link=Links.SKLEARN_PLOTS.value, padx=5, pady=5, relief='solid')
-        self.bp_threshold_lbl = SimBALabel(parent=bp_threshold_frm, txt="Body-parts detected below the set threshold won't be shown in the output videos.", font=Formats.FONT_REGULAR_ITALICS.value)
-        self.bp_threshold_entry = Entry_Box(parent=bp_threshold_frm, fileDescription='BODY-PART PROBABILITY THRESHOLD: ', labelwidth=40, entry_box_width=15, value=0.00, img='green_dice')
+        self.bp_threshold_lbl = SimBALabel(parent=bp_threshold_frm, txt="Body-parts detected below the set threshold won't be shown in the output videos (use 0.0 to see all body-part predictions)", font=Formats.FONT_REGULAR_ITALICS.value)
+        self.bp_threshold_entry = Entry_Box(parent=bp_threshold_frm, fileDescription='BODY-PART PROBABILITY THRESHOLD: ', labelwidth=40, entry_box_width=15, value=0.00, img='green_dice', justify='center')
         self.get_bp_probability_threshold()
 
         bp_threshold_frm.grid(row=0, column=0, sticky=NW)

simba/ui/pop_ups/egocentric_alignment_pop_up.py

@@ -5,13 +5,12 @@ from typing import Union
 from simba.data_processors.egocentric_aligner import EgocentricalAligner
 from simba.mixins.config_reader import ConfigReader
 from simba.mixins.pop_up_mixin import PopUpMixin
-from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, DropDownMenu,
-                                        FolderSelect, SimbaCheckbox,
+from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, FolderSelect,
                                         SimBADropDown)
 from simba.utils.checks import check_if_dir_exists, check_nvidea_gpu_available
 from simba.utils.enums import Keys, Links
 from simba.utils.errors import InvalidInputError, NoDataError, SimBAGPUError
-from simba.utils.lookups import get_color_dict
+from simba.utils.lookups import find_closest_string, get_color_dict
 from simba.utils.read_write import (find_all_videos_in_directory,
                                     find_files_of_filetypes_in_directory,
                                     get_fn_ext, str_2_bool)
@@ -31,19 +30,19 @@ class EgocentricAlignPopUp(ConfigReader, PopUpMixin):
         ConfigReader.__init__(self, config_path=config_path, read_video_info=False, create_logger=False)
         self.clr_dict = get_color_dict()
         gpu_status = NORMAL if check_nvidea_gpu_available() else DISABLED
-
-
         settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="SETTINGS", icon_name=Keys.DOCUMENTATION.value, icon_link=Links.VIDEO_TOOLS.value)
-        self.data_dir = FolderSelect(settings_frm, "DATA DIRECTORY:", lblwidth=45, initialdir=self.outlier_corrected_dir, lbl_icon='folder')
-        self.videos_dir = FolderSelect(settings_frm, "VIDEO DIRECTORY:", lblwidth=45, initialdir=self.video_dir, lbl_icon='folder')
-        self.save_dir = FolderSelect(settings_frm, "SAVE DIRECTORY:", lblwidth=45, initialdir=self.video_dir, lbl_icon='folder')
+        self.data_dir = FolderSelect(settings_frm, "DATA DIRECTORY:", lblwidth=45, initialdir=self.outlier_corrected_dir, lbl_icon='folder', tooltip_key='EGOCENTRIC_DATA_DIR')
+        self.videos_dir = FolderSelect(settings_frm, "VIDEO DIRECTORY:", lblwidth=45, initialdir=self.video_dir, lbl_icon='folder', tooltip_key='EGOCENTRIC_VIDEO_DIR')
+        self.save_dir = FolderSelect(settings_frm, "SAVE DIRECTORY:", lblwidth=45, initialdir=self.project_path, lbl_icon='folder', tooltip_key='SAVE_DIR')
+        default_center = find_closest_string(target='center', string_list=self.body_parts_lst)[0]
+        default_direction = find_closest_string(target='nose', string_list=self.body_parts_lst)[0]
 
-        self.center_anchor_dropdown = SimBADropDown(parent=settings_frm, label="CENTER ANCHOR:", dropdown_options=self.body_parts_lst, label_width=45, dropdown_width=45, img='center', value=self.body_parts_lst[0])
-        self.direction_anchor_dropdown = SimBADropDown(parent=settings_frm, label="DIRECTION ANCHOR:", dropdown_options=self.body_parts_lst, label_width=45, dropdown_width=45, img='direction', value=self.body_parts_lst[1])
-        self.direction_dropdown = SimBADropDown(parent=settings_frm, label="DIRECTION:", dropdown_options=list(range(0, 361)), label_width=45, dropdown_width=45, img='direction_2', value=0)
-        self.fill_clr_dropdown = SimBADropDown(parent=settings_frm, label="ROTATION COLOR:", dropdown_options=list(self.clr_dict.keys()), label_width=45, dropdown_width=45, img='fill', value='Black')
-        self.core_cnt_dropdown = SimBADropDown(parent=settings_frm, label="CPU COUNT:", dropdown_options=list(range(1, self.cpu_cnt + 1)), label_width=45, dropdown_width=45, img='cpu_small', value=self.cpu_cnt)
-        self.gpu_dropdown = SimBADropDown(parent=settings_frm, label='USE GPU:', dropdown_options=['TRUE', 'FALSE'], label_width=45, dropdown_width=45, img='gpu_3', value='FALSE', state=gpu_status)
+        self.center_anchor_dropdown = SimBADropDown(parent=settings_frm, label="CENTER ANCHOR:", dropdown_options=self.body_parts_lst, label_width=45, dropdown_width=45, img='center', value=default_center, tooltip_key='EGOCENTRIC_ANCHOR')
+        self.direction_anchor_dropdown = SimBADropDown(parent=settings_frm, label="DIRECTION ANCHOR:", dropdown_options=self.body_parts_lst, label_width=45, dropdown_width=45, img='direction', value=default_direction, tooltip_key='EGOCENTRIC_DIRECTION_ANCHOR')
+        self.direction_dropdown = SimBADropDown(parent=settings_frm, label="DIRECTION:", dropdown_options=list(range(0, 361)), label_width=45, dropdown_width=45, img='direction_2', value=0, tooltip_key='EGOCENTRIC_DIRECTION')
+        self.fill_clr_dropdown = SimBADropDown(parent=settings_frm, label="ROTATION COLOR:", dropdown_options=list(self.clr_dict.keys()), label_width=45, dropdown_width=45, img='fill', value='Black', tooltip_key='ROTATE_FILL_COLOR')
+        self.core_cnt_dropdown = SimBADropDown(parent=settings_frm, label="CPU COUNT:", dropdown_options=list(range(1, self.cpu_cnt + 1)), label_width=45, dropdown_width=45, img='cpu_small', value=int(self.cpu_cnt/2), tooltip_key='CORE_COUNT')
+        self.gpu_dropdown = SimBADropDown(parent=settings_frm, label='USE GPU:', dropdown_options=['TRUE', 'FALSE'], label_width=45, dropdown_width=45, img='gpu_3', value='FALSE', state=gpu_status, tooltip_key='USE_GPU')
 
         settings_frm.grid(row=0, column=0, sticky=NW)
         self.data_dir.grid(row=0, column=0, sticky=NW)
@@ -62,9 +61,10 @@ class EgocentricAlignPopUp(ConfigReader, PopUpMixin):
         data_dir, video_dir = self.data_dir.folder_path, self.videos_dir.folder_path
         save_dir = self.save_dir.folder_path
         gpu = str_2_bool(self.gpu_dropdown.get_value())
-        check_if_dir_exists(in_dir=data_dir);
-        check_if_dir_exists(in_dir=video_dir);
+        check_if_dir_exists(in_dir=data_dir)
+        check_if_dir_exists(in_dir=video_dir)
         check_if_dir_exists(in_dir=save_dir)
+        core_cnt = int(self.core_cnt_dropdown.get_value())
         if (save_dir == data_dir) or (save_dir == video_dir):
             raise InvalidInputError(msg='The save directory cannot be the same as the data/video directories',source=self.__class__.__name__)
         center_anchor, direction_anchor = self.center_anchor_dropdown.getChoices(), self.direction_anchor_dropdown.getChoices()
@@ -72,12 +72,10 @@ class EgocentricAlignPopUp(ConfigReader, PopUpMixin):
         direction = int(self.direction_dropdown.getChoices())
         if gpu and not check_nvidea_gpu_available():
             raise SimBAGPUError(msg='No NVIDEA GPU detected.', source=self.__class__.__name__)
-        data_file_paths = find_files_of_filetypes_in_directory(directory=data_dir, extensions=[f'.{self.file_type}'],
-                                                               raise_error=True)
+        data_file_paths = find_files_of_filetypes_in_directory(directory=data_dir, extensions=[f'.{self.file_type}'], raise_error=True)
         data_file_paths = [os.path.join(data_dir, x) for x in data_file_paths]
         data_file_names = [get_fn_ext(filepath=x)[1] for x in data_file_paths]
-        video_file_paths = list(
-            find_all_videos_in_directory(directory=video_dir, as_dict=True, raise_error=True).values())
+        video_file_paths = list( find_all_videos_in_directory(directory=video_dir, as_dict=True, raise_error=True).values())
         video_file_names = [get_fn_ext(filepath=x)[1] for x in video_file_paths]
         missing_video_files = [x for x in video_file_names if x not in data_file_names]
         if len(missing_video_files) > 0: raise NoDataError(
@@ -89,6 +87,7 @@ class EgocentricAlignPopUp(ConfigReader, PopUpMixin):
                                       anchor_2=direction_anchor,
                                       direction=direction,
                                       anchor_location=None,
+                                      core_cnt=core_cnt,
                                       fill_clr=self.clr_dict[fill_clr],
                                       verbose=True,
                                       gpu=gpu,
@@ -97,4 +96,4 @@ class EgocentricAlignPopUp(ConfigReader, PopUpMixin):
         aligner.run()
 
 
-#_ = EgocentricAlignPopUp(config_path=r"C:\troubleshooting\mitra\project_folder\project_config.ini")
+#_ = EgocentricAlignPopUp(config_path=r"D:\troubleshooting\mitra\project_folder\project_config.ini")

simba/ui/pop_ups/fsttc_pop_up.py

@@ -7,25 +7,29 @@ from typing import Union
 from simba.data_processors.fsttc_calculator import FSTTCCalculator
 from simba.mixins.config_reader import ConfigReader
 from simba.mixins.pop_up_mixin import PopUpMixin
-from simba.ui.tkinter_functions import CreateLabelFrameWithIcon, Entry_Box
+from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, Entry_Box,
+                                        SimbaCheckbox)
 from simba.utils.checks import check_int
-from simba.utils.enums import Defaults, Formats, Keys, Links
+from simba.utils.enums import Formats, Keys, Links
 from simba.utils.errors import CountError
 
 
 class FSTTCPopUp(PopUpMixin, ConfigReader):
-    def __init__(self, config_path: Union[str, os.PathLike]):
+    def __init__(self,
+                 config_path: Union[str, os.PathLike]):
 
-        PopUpMixin.__init__(self, title="FORWARD SPIKE TIME TILING COEFFICIENTS", icon='dependency')
         ConfigReader.__init__(self, config_path=config_path, read_video_info=False)
+        if len(self.clf_names) < 2:
+            raise CountError(msg=f'Cannot compute FSTTC: Needs project with at least 2 classified behaviors, got {len(self.clf_names)}', source=self.__class__.__name__)
+
+        PopUpMixin.__init__(self, title="FORWARD SPIKE TIME TILING COEFFICIENTS", icon='dependency')
         settings_frm = CreateLabelFrameWithIcon( parent=self.main_frm, header="SETTINGS", icon_name=Keys.DOCUMENTATION.value, icon_link=Links.FSTTC.value,)
-        self.time_delta_eb = Entry_Box(settings_frm, "TIME-DELTA", "10", validation="numeric")
-        self.graph_cb_var = BooleanVar(value=False)
-        graph_cb = Checkbutton(settings_frm, font=Formats.FONT_REGULAR.value, text="CREATE GRAPH", variable=self.graph_cb_var)
-        self.join_bouts_within_delta_var = BooleanVar(value=False)
-        join_bouts_within_delta_cb = Checkbutton(settings_frm, text="JOIN BOUTS WITHIN TIME-DELTA", font=Formats.FONT_REGULAR.value, variable=self.join_bouts_within_delta_var,)
-        self.time_delta_at_onset_var = BooleanVar(value=False)
-        time_delta_at_onset_cb = Checkbutton( settings_frm, text="TIME-DELTA AT BOUT START", font=Formats.FONT_REGULAR.value, variable=self.time_delta_at_onset_var)
+        self.time_delta_eb = Entry_Box(parent=settings_frm, fileDescription="TIME-DELTA", labelwidth=25, validation="numeric", entry_box_width=20, img='timer_2')
+
+        graph_cb, self.graph_cb_var = SimbaCheckbox(parent=settings_frm, font=Formats.FONT_REGULAR.value, val=False, txt_img='line_chart_blue', txt="CREATE GRAPH")
+        join_bouts_within_delta_cb, self.join_bouts_within_delta_var = SimbaCheckbox(parent=settings_frm, font=Formats.FONT_REGULAR.value, val=False, txt_img='join_purple', txt="JOIN BOUTS WITHIN TIME-DELTA")
+        time_delta_at_onset_cb, self.time_delta_at_onset_var = SimbaCheckbox(parent=settings_frm, font=Formats.FONT_REGULAR.value, val=False, txt_img='play', txt="TIME-DELTA AT BOUT START")
+
         settings_frm.grid(row=0, column=0, sticky=NW)
         self.time_delta_eb.grid(row=0, column=0, sticky="NW")
         join_bouts_within_delta_cb.grid(row=1, column=0, sticky="NW")
@@ -43,22 +47,20 @@ class FSTTCPopUp(PopUpMixin, ConfigReader):
                 targets.append(behaviour)
 
         if len(targets) < 2:
-            raise CountError(
-                msg="FORWARD SPIKE TIME TILING COEFFICIENTS REQUIRE 2 OR MORE BEHAVIORS.",
-                source=self.__class__.__name__,
-            )
-
-        fsttc_calculator = FSTTCCalculator(
-            config_path=self.config_path,
-            time_window=self.time_delta_eb.entry_get,
-            join_bouts_within_delta=self.join_bouts_within_delta_var.get(),
-            time_delta_at_onset=self.time_delta_at_onset_var.get(),
-            behavior_lst=targets,
-            create_graphs=self.graph_cb_var.get(),
-        )
+            raise CountError(msg="FORWARD SPIKE TIME TILING COEFFICIENTS REQUIRE 2 OR MORE BEHAVIORS.", source=self.__class__.__name__,)
+
+        fsttc_calculator = FSTTCCalculator(config_path=self.config_path,
+                                           time_window=self.time_delta_eb.entry_get,
+                                           join_bouts_within_delta=self.join_bouts_within_delta_var.get(),
+                                           time_delta_at_onset=self.time_delta_at_onset_var.get(),
+                                           behavior_lst=targets,
+                                           create_graphs=self.graph_cb_var.get())
         fsttc_calculator.run()
 
 
-#ExtractAnnotationFramesPopUp(config_path=)
+
 
 #_ = FSTTCPopUp(config_path=r"C:\troubleshooting\multi_animal_dlc_two_c57\project_folder\project_config.ini")
+
+#_ = FSTTCPopUp(config_path=r"C:\troubleshooting\mitra\project_folder\project_config.ini")
+

simba/ui/pop_ups/gantt_pop_up.py

@@ -13,6 +13,7 @@ from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, SimbaButton,
 from simba.utils.checks import check_if_filepath_list_is_empty
 from simba.utils.enums import Formats, Links, Options
 from simba.utils.errors import NoSpecifiedOutputError
+from simba.utils.lookups import get_fonts
 from simba.utils.read_write import find_files_of_filetypes_in_directory
 
 
@@ -29,7 +30,9 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
         check_if_filepath_list_is_empty(filepaths=self.machine_results_paths,error_msg=f"SIMBA ERROR: Zero files found in the {self.machine_results_dir} directory. Create classification results before visualizing gantt charts",)
         palettes = Options.PALETTE_OPTIONS_CATEGORICAL.value + Options.PALETTE_OPTIONS.value
         self.data_paths = find_files_of_filetypes_in_directory(directory=self.machine_results_dir, extensions=[f'.{self.file_type}'], as_dict=True)
-        max_file_name_len = max(len(k) for k in self.data_paths) + 5
+        max_file_name_len, fonts = max(len(k) for k in self.data_paths) + 5, list(get_fonts(sort_alphabetically=True).keys())
+        fonts.insert(0, 'AUTO')
+        default_font = 'Arial' if 'Arial' in fonts else 'AUTO'
         PopUpMixin.__init__(self, config_path=config_path, title="VISUALIZE GANTT PLOTS", icon='gantt_small')
 
         self.style_settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="STYLE SETTINGS", icon_name='settings', icon_link=Links.GANTT_PLOTS.value, relief='solid', padx=5, pady=5)
@@ -39,6 +42,15 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
         self.palette_dropdown = SimBADropDown(parent=self.style_settings_frm, dropdown_options=palettes, label='COLOR PALETTE: ', label_width=30, dropdown_width=30, value='Set1', img='palette_small')
         self.time_format_dropdown = SimBADropDown(parent=self.style_settings_frm, dropdown_options=['SECONDS', 'HH:MM:SS'], label='X-AXIS TIME FORMAT: ', label_width=30, dropdown_width=30, value='SECONDS', img='timer_2')
         self.core_dropdown = SimBADropDown(parent=self.style_settings_frm, dropdown_options=list(range(1, self.cpu_cnt+1)), label='CPU CORES: ', label_width=30, dropdown_width=30, value=int(self.cpu_cnt/2), img='cpu_small')
+        self.font_dropdown = SimBADropDown(parent=self.style_settings_frm, dropdown_options=fonts, label='FONT: ', label_width=30, dropdown_width=30, value=default_font, img='font')
+
+
+        self.clf_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="BEHAVIORS", icon_name='forest', icon_link=Links.GANTT_PLOTS.value, relief='solid', padx=5, pady=5)
+        self.clf_choices = {}
+        for cnt, clf_name in enumerate(self.clf_names):
+            gantt_frames_cb, self.gantt_frames_var = SimbaCheckbox(parent=self.clf_frm, txt=clf_name, val=True)
+            self.clf_choices[clf_name] = self.gantt_frames_var
+            gantt_frames_cb.grid(row=cnt, column=0, sticky=NW)
 
         self.settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="VISUALIZATION SETTINGS", icon_name='eye', icon_link=Links.GANTT_PLOTS.value, relief='solid', padx=5, pady=5)
         gantt_frames_cb, self.gantt_frames_var = SimbaCheckbox(parent=self.settings_frm, txt='CREATE FRAMES', txt_img='frames', val=False)
@@ -57,18 +69,20 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
         self.font_rotation_dropdown.grid(row=2, sticky=NW)
         self.palette_dropdown.grid(row=3, sticky=NW)
         self.time_format_dropdown.grid(row=4, sticky=NW)
-        self.core_dropdown.grid(row=5, sticky=NW)
+        self.font_dropdown.grid(row=5, sticky=NW)
+        self.core_dropdown.grid(row=6, sticky=NW)
 
-        self.settings_frm.grid(row=1, sticky=NW, padx=10, pady=10)
+        self.clf_frm.grid(row=1, sticky=NW, padx=10, pady=10)
+        self.settings_frm.grid(row=2, sticky=NW, padx=10, pady=10)
         gantt_videos_cb.grid(row=0, sticky=NW)
         gantt_frames_cb.grid(row=1, sticky=W)
         gantt_last_frame_cb.grid(row=2, sticky=NW)
 
-        self.run_single_video_frm.grid(row=2, sticky=NW)
+        self.run_single_video_frm.grid(row=3, sticky=NW)
         self.run_single_video_btn.grid(row=0, sticky=NW)
         self.single_video_dropdown.grid(row=0, column=1, sticky=NW)
 
-        self.run_multiple_videos.grid(row=3, sticky=NW)
+        self.run_multiple_videos.grid(row=4, sticky=NW)
         self.run_multiple_video_btn.grid(row=0, sticky=NW)
 
         self.main_frm.mainloop()
@@ -85,8 +99,14 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
         video_setting = self.gantt_videos_var.get()
         last_frm_setting = self.gantt_last_frame_var.get()
         palette = self.palette_dropdown.get_value()
+        font = self.font_dropdown.get_value()
+        font = None if font == 'AUTO' else font
         hhmmss = True if self.time_format_dropdown.get_value() == 'HH:MM:SS' else False
-
+        clf_names = []
+        for clf_name, clf_val in self.clf_choices.items():
+            if clf_val.get(): clf_names.append(clf_name)
+        if len(clf_names) < 1:
+            raise NoSpecifiedOutputError(msg="Select AT LEAST one behavior name.")
         if not frame_setting and not video_setting and not last_frm_setting:
             raise NoSpecifiedOutputError(msg="SIMBA ERROR: Please select gantt videos, frames, and/or last frame.")
 
@@ -103,6 +123,8 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
                                                      data_paths=data_paths,
                                                      width=width,
                                                      height=height,
+                                                     font=font,
+                                                     clf_names=clf_names,
                                                      font_size=font_size,
                                                      font_rotation=font_rotation,
                                                      core_cnt=core_cnt,
@@ -116,7 +138,9 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
                                                       last_frm_setting=last_frm_setting,
                                                       data_paths=data_paths,
                                                       width=width,
+                                                      font=font,
                                                       height=height,
+                                                      clf_names=clf_names,
                                                       font_size=font_size,
                                                       font_rotation=font_rotation,
                                                       palette=palette)
@@ -124,6 +148,7 @@ class GanttPlotPopUp(PopUpMixin, ConfigReader):
 
 
 #_ = GanttPlotPopUp(config_path=r"D:\troubleshooting\maplight_ri\project_folder\project_config.ini")
+#_ = GanttPlotPopUp(config_path=r"C:\troubleshooting\mitra\project_folder\project_config.ini")
 # _ = GanttPlotPopUp(config_path=r"/Users/simon/Desktop/envs/simba/troubleshooting/two_black_animals_14bp/project_folder/project_config.ini")
 
  # _ = GanttPlotPopUp(config_path=r"C:\troubleshooting\RAT_NOR\project_folder\project_config.ini")

simba/ui/pop_ups/interpolate_pop_up.py

@@ -35,8 +35,6 @@ class InterpolatePopUp(PopUpMixin, ConfigReader):
         self.config_path = config_path
 
         self.settings_frm= CreateLabelFrameWithIcon(parent=self.main_frm, header="SETTINGS", icon_name='settings')
-
-        #self.settings_frm = LabelFrame(self.main_frm, text="SETTINGS", font=Formats.FONT_HEADER.value)
         instruction_lbl_1 = SimBALabel(parent=self.settings_frm, txt=INSTRUCTIONS_LBL_1, font=Formats.FONT_REGULAR_ITALICS.value, justify='center')
         self.type_dropdown = SimBADropDown(parent=self.settings_frm, dropdown_options=['MISSING BODY-PARTS', 'MISSING ANIMALS'], label="INTERPOLATION TYPE:", label_width=35, value='MISSING BODY-PARTS', dropdown_width=35, img='file_type')
         self.method_dropdown = SimBADropDown(parent=self.settings_frm, dropdown_options=['NEAREST', 'LINEAR', 'QUADRATIC'], label="INTERPOLATION METHOD:", label_width=35, value='NEAREST', dropdown_width=35, img='equation_small')
@@ -50,7 +48,7 @@ class InterpolatePopUp(PopUpMixin, ConfigReader):
 
         self.single_file_frm = LabelFrame(self.main_frm, text="INTERPOLATE SINGLE DATA FILE", font=Formats.FONT_HEADER.value)
         instruction_lbl_single = SimBALabel(parent=self.single_file_frm, txt=INSTRUCTIONS_LBL_2, font=Formats.FONT_REGULAR_ITALICS.value, justify='center')
-        self.selected_file = FileSelect(self.single_file_frm, "DATA PATH:", lblwidth=35, file_types=[("VIDEO FILE", ".csv .parquet")], initialdir=self.project_path)
+        self.selected_file = FileSelect(self.single_file_frm, "DATA PATH:", lblwidth=35, file_types=[("VIDEO FILE", ".csv .parquet")], initialdir=self.project_path, lbl_icon='file')
         self.run_btn_single = SimbaButton(parent=self.single_file_frm, txt="RUN SINGLE DATA FILE INTERPOLATION", img='rocket', txt_clr="blue", font=Formats.FONT_REGULAR.value, cmd=self.run, cmd_kwargs={'multiple': False})
 
         self.single_file_frm.grid(row=1, column=0, sticky=NW)
@@ -60,7 +58,7 @@ class InterpolatePopUp(PopUpMixin, ConfigReader):
 
         self.multiple_file_frm = LabelFrame(self.main_frm, text="INTERPOLATE DIRECTORY OF DATA", font=Formats.FONT_HEADER.value)
         instruction_lbl_multiple = SimBALabel(parent=self.multiple_file_frm, txt=INSTRUCTIONS_LBL_3, font=Formats.FONT_REGULAR_ITALICS.value, justify='center')
-        self.selected_dir = FolderSelect(self.multiple_file_frm, "SELECT DIRECTORY OF DATA FILES:", lblwidth=35, initialdir=self.project_path)
+        self.selected_dir = FolderSelect(self.multiple_file_frm, "SELECT DIRECTORY OF DATA FILES:", lblwidth=35, initialdir=self.project_path, lbl_icon='folder')
         self.run_btn_multiple = SimbaButton(parent=self.multiple_file_frm, txt="RUN DATA DIRECTORY INTERPOLATION", img='rocket', txt_clr="blue", font=Formats.FONT_REGULAR.value, cmd=self.run, cmd_kwargs={'multiple': True})
         self.multiple_file_frm.grid(row=2, column=0, sticky=NW)
         instruction_lbl_multiple.grid(row=0, column=0, sticky=NW)

simba/ui/pop_ups/kleinberg_pop_up.py

@@ -8,12 +8,13 @@ from simba.data_processors.kleinberg_calculator import KleinbergCalculator
 from simba.mixins.config_reader import ConfigReader
 from simba.mixins.pop_up_mixin import PopUpMixin
 from simba.ui.tkinter_functions import (CreateLabelFrameWithIcon, Entry_Box,
-                                        SimbaButton, SimBADropDown)
+                                        SimbaButton, SimBADropDown, SimBALabel)
 from simba.utils.checks import check_float, check_int
 from simba.utils.enums import Formats, Keys, Links
 from simba.utils.errors import NoChoosenClassifierError, NoDataError
-from simba.utils.read_write import str_2_bool
+from simba.utils.read_write import get_current_time, str_2_bool
 
+INSTRUCTIONS_TXT = 'Results in the project_folder/csv/machine_results folder are overwritten.\n If saving the originals, the original un-smoothened data is saved in a subdirectory of \nthe project_folder/csv/machine_results folder'
 
 class KleinbergPopUp(PopUpMixin, ConfigReader):
     def __init__(self,
@@ -24,61 +25,59 @@ class KleinbergPopUp(PopUpMixin, ConfigReader):
             raise NoDataError(msg=f'Cannot perform Kleinberg smoothing: No data files found in {self.machine_results_dir} directory', source=self.__class__.__name__)
         PopUpMixin.__init__(self, title="APPLY KLEINBERG BEHAVIOR CLASSIFICATION SMOOTHING", icon='smooth')
         kleinberg_settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header="KLEINBERG SETTINGS", icon_name=Keys.DOCUMENTATION.value, icon_link=Links.KLEINBERG.value)
-        self.k_sigma = Entry_Box(kleinberg_settings_frm, fileDescription="SIGMA", img='sigma', value='2', justify='center', labelwidth=35, entry_box_width=35)
-        self.k_gamma = Entry_Box(kleinberg_settings_frm, fileDescription="GAMMA", img='gamma', value='0.3', justify='center', labelwidth=35, entry_box_width=35)
-        self.k_hierarchy = Entry_Box(kleinberg_settings_frm, fileDescription="HIERARCHY", value=1, img='hierarchy_2', justify='center', labelwidth=35, entry_box_width=35)
-        self.h_search_dropdown = SimBADropDown(parent=kleinberg_settings_frm, dropdown_options=['TRUE', 'FALSE'], label="HIERACHICAL SEARCH", value='FALSE', img='hierarchy', label_width=35, dropdown_width=35)
+        self.k_sigma = Entry_Box(kleinberg_settings_frm, fileDescription="SIGMA", img='sigma', value='2', justify='center', labelwidth=35, entry_box_width=35, tooltip_key='KLEINBERG_SIGMA')
+        self.k_gamma = Entry_Box(kleinberg_settings_frm, fileDescription="GAMMA", img='gamma', value='0.3', justify='center', labelwidth=35, entry_box_width=35, tooltip_key='KLEINBERG_GAMMA')
+        self.k_hierarchy = Entry_Box(kleinberg_settings_frm, fileDescription="HIERARCHY", value=1, img='hierarchy_2', justify='center', labelwidth=35, entry_box_width=35, validation='numeric', tooltip_key='KLEINBERG_HIERARCHY')
+        self.h_search_dropdown = SimBADropDown(parent=kleinberg_settings_frm, dropdown_options=['TRUE', 'FALSE'], label="HIERARCHICAL SEARCH", value='FALSE', img='hierarchy', label_width=35, dropdown_width=35, tooltip_key='KLEINBERG_HIERARCHY_SEARCH')
+        self.save_originals_dropdown = SimBADropDown(parent=kleinberg_settings_frm, dropdown_options=['TRUE', 'FALSE'], label="SAVE ORIGINAL DATA:", value='TRUE', img='save', label_width=35, dropdown_width=35, tooltip_key='KLEINBERG_SAVE_ORIGINALS')
+        self.instructions_lbl = SimBALabel(parent=kleinberg_settings_frm, txt=INSTRUCTIONS_TXT, justify='center', txt_clr='blue', font=Formats.FONT_REGULAR_ITALICS.value)
 
 
-        kleinberg_table_frame = LabelFrame(self.main_frm, text="CHOOSE CLASSIFIER(S) FOR KLEINBERG SMOOTHING", font=Formats.FONT_HEADER.value)
+        kleinberg_table_frame = CreateLabelFrameWithIcon(parent=self.main_frm, header="CHOOSE CLASSIFIER(S) FOR KLEINBERG SMOOTHING", icon_name=Keys.DOCUMENTATION.value, icon_link=Links.KLEINBERG.value)
         clf_var_dict, clf_cb_dict = {}, {}
         for clf_cnt, clf in enumerate(self.clf_names):
             clf_var_dict[clf] = BooleanVar()
             clf_cb_dict[clf] = Checkbutton(kleinberg_table_frame, text=clf, font=Formats.FONT_REGULAR.value, variable=clf_var_dict[clf])
-            clf_cb_dict[clf].grid(row=clf_cnt, sticky=NW)
+            clf_cb_dict[clf].grid(row=clf_cnt, column=0, sticky=NW)
 
         run_kleinberg_btn = SimbaButton(parent=self.main_frm, txt="APPLY KLEINBERG SMOOTHER", img='rocket', txt_clr="blue", font=Formats.FONT_REGULAR.value, cmd=self.run_kleinberg, cmd_kwargs={'behaviors_dict': lambda: clf_var_dict, 'hierarchical_search': lambda: str_2_bool(self.h_search_dropdown.get_value())})
-        kleinberg_settings_frm.grid(row=0, sticky=W, padx=10)
-        self.k_sigma.grid(row=0, sticky=W)
-        self.k_gamma.grid(row=1, sticky=W)
-        self.k_hierarchy.grid(row=2, sticky=W)
-        self.h_search_dropdown.grid(row=3, column=0, sticky=W)
-        kleinberg_table_frame.grid(row=1, pady=10, padx=10)
-        run_kleinberg_btn.grid(row=2)
+        kleinberg_settings_frm.grid(row=0, sticky=W, pady=(15, 0))
+        self.instructions_lbl.grid(row=0, sticky=W)
+        self.k_sigma.grid(row=1, sticky=W)
+        self.k_gamma.grid(row=2, sticky=W)
+        self.k_hierarchy.grid(row=3, sticky=W)
+        self.h_search_dropdown.grid(row=4, column=0, sticky=W)
+        self.save_originals_dropdown.grid(row=5, column=0, sticky=W)
+        kleinberg_table_frame.grid(row=1, column=0, sticky=NW, pady=(15, 0))
+        run_kleinberg_btn.grid(row=2, column=0, sticky=NW, pady=(15, 0))
         self.main_frm.mainloop()
 
     def run_kleinberg(self, behaviors_dict: dict, hierarchical_search: bool):
         targets = []
         for behaviour, behavior_val in behaviors_dict.items():
-            if behavior_val.get():
-                targets.append(behaviour)
+            if behavior_val.get(): targets.append(behaviour)
 
         if len(targets) == 0:
             raise NoChoosenClassifierError(source=self.__class__.__name__)
 
-        check_int(name="Hierarchy", value=self.k_hierarchy.entry_get)
-        check_float(name="Sigma", value=self.k_sigma.entry_get)
-        check_float(name="Gamma", value=self.k_gamma.entry_get)
-
-        try:
-            print(
-                "Applying kleinberg hyperparameter Setting: Sigma: {}, Gamma: {}, Hierarchy: {}".format(
-                    str(self.k_sigma.entry_get),
-                    str(self.k_gamma.entry_get),
-                    str(self.k_hierarchy.entry_get),
-                )
-            )
-        except:
-            print("Please insert accurate values for all hyperparameters.")
-
-        kleinberg_analyzer = KleinbergCalculator(
-            config_path=self.config_path,
-            classifier_names=targets,
-            sigma=self.k_sigma.entry_get,
-            gamma=self.k_gamma.entry_get,
-            hierarchy=self.k_hierarchy.entry_get,
-            hierarchical_search=hierarchical_search,
-        )
+        k_hierarchy = self.k_hierarchy.entry_get
+        k_sigma = self.k_sigma.entry_get
+        k_gamma = self.k_gamma.entry_get
+        save_originals = str_2_bool(self.save_originals_dropdown.get_value())
+
+        check_int(name="Hierarchy", value=k_hierarchy, min_value=1, allow_negative=False, allow_zero=False, raise_error=True)
+        check_float(name="Sigma", value=k_sigma, allow_negative=False, allow_zero=False, raise_error=True)
+        check_float(name="Gamma", value=k_gamma, allow_negative=False, allow_zero=False, raise_error=True)
+
+        print(f"[{get_current_time()}] Applying kleinberg hyperparameter Setting: Sigma: {k_sigma}, Gamma: {k_gamma}, Hierarchy: {k_hierarchy}")
+
+        kleinberg_analyzer = KleinbergCalculator(config_path=self.config_path,
+                                                 classifier_names=targets,
+                                                 sigma=float(k_sigma),
+                                                 gamma=float(k_gamma),
+                                                 hierarchy=int(k_hierarchy),
+                                                 hierarchical_search=hierarchical_search,
+                                                 save_originals=save_originals)
         kleinberg_analyzer.run()
 
 

simba/ui/pop_ups/multiple_videos_to_frames_popup.py

@@ -19,17 +19,16 @@ class MultipleVideos2FramesPopUp(PopUpMixin):
         PopUpMixin.__init__(self, title="EXTRACT FRAMES FROM SINGLE VIDEO", icon='frames')
         core_cnt = find_core_cnt()[0]
         settings_frm = CreateLabelFrameWithIcon(parent=self.main_frm, header='SETTINGS', icon_name='settings', padx=5, pady=5, relief='solid')
-
-        self.video_dir = FolderSelect(parent=settings_frm, folderDescription="VIDEO DIRECTORY:", lblwidth=50)
-        self.save_dir = FolderSelect(parent=settings_frm, folderDescription="SAVE DIRECTORY:", lblwidth=50)
-        self.core_cnt_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=list(range(1, core_cnt+1)), label='CORE COUNT: ', label_width=50, value=int(core_cnt/2))
-        self.quality_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=list(range(1, 101)), label='IMAGE QUALITY: ', label_width=50, value=90, state='disabled')
-        self.img_format = SimBADropDown(parent=settings_frm, dropdown_options=['jpeg', 'png', 'webp'], label='IMAGE FORMAT: ', label_width=50, value='png', command=self._inactivate_quality)
-        self.verbose_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='VERBOSE: ', label_width=50, value='TRUE')
-        self.greyscale_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='GREYSCALE: ', label_width=50, value='FALSE')
-        self.bw_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='BLACK & WHITE: ',label_width=50, value='FALSE')
-        self.clahe_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='CLAHE: ', label_width=50, value='FALSE')
-        self.include_fn_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='INCLUDE VIDEO NAME IN IMAGE NAMES: ',label_width=50, value='FALSE')
+        self.video_dir = FolderSelect(parent=settings_frm, folderDescription="VIDEO DIRECTORY:", lblwidth=50, lbl_icon='folder')
+        self.save_dir = FolderSelect(parent=settings_frm, folderDescription="SAVE DIRECTORY:", lblwidth=50, lbl_icon='folder')
+        self.core_cnt_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=list(range(1, core_cnt+1)), label='CORE COUNT: ', label_width=50, value=int(core_cnt/2), img='cpu_small')
+        self.quality_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=list(range(1, 101)), label='IMAGE QUALITY: ', label_width=50, value=90, state='disabled', img='pct_2')
+        self.img_format = SimBADropDown(parent=settings_frm, dropdown_options=['jpeg', 'png', 'webp'], label='IMAGE FORMAT: ', label_width=50, value='png', command=self._inactivate_quality, img='file_type')
+        self.verbose_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='VERBOSE: ', label_width=50, value='TRUE', img='verbose')
+        self.greyscale_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='GREYSCALE: ', label_width=50, value='FALSE', img='grey')
+        self.bw_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='BLACK & WHITE: ',label_width=50, value='FALSE', img='black_and_white')
+        self.clahe_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='CLAHE: ', label_width=50, value='FALSE', img='clahe')
+        self.include_fn_dropdown = SimBADropDown(parent=settings_frm, dropdown_options=['TRUE', 'FALSE'], label='INCLUDE VIDEO NAME IN IMAGE NAMES: ',label_width=50, value='FALSE', img='id_card_2')
 
         settings_frm.grid(row=0, column=0, sticky=NW, padx=10, pady=10)
         self.video_dir.grid(row=0, column=0, sticky=NW)