Add DeepLabCut import to menu.

sleap/gui/app.py

@@ -256,6 +256,12 @@ def add_submenu_choices(menu, title, options, key):
             "COCO dataset...",
             self.commands.importCoco,
         )
+        add_menu_item(
+            import_types_menu,
+            "import_dlc",
+            "DeepLabCut dataset...",
+            self.commands.importDLC,
+        )
         add_menu_item(
             import_types_menu,
             "import_dpk",

sleap/gui/commands.py

@@ -199,6 +199,10 @@ def importCoco(self):
         """Imports COCO datasets."""
         self.execute(ImportCoco)
 
+    def importDLC(self):
+        """Imports DeepLabCut datasets."""
+        self.execute(ImportDeepLabCut)
+
     def importLEAP(self):
         """Imports LEAP matlab datasets."""
         self.execute(ImportLEAP)
@@ -581,6 +585,37 @@ def ask(context: "CommandContext", params: dict) -> bool:
         return True
 
 
+class ImportDeepLabCut(AppCommand):
+    @staticmethod
+    def do_action(context: "CommandContext", params: dict):
+
+        labels = Labels.load_deeplabcut_csv(
+            filename=params["filename"]
+        )
+
+        new_window = context.app.__class__()
+        new_window.showMaximized()
+        new_window.loadLabelsObject(labels=labels)
+
+    @staticmethod
+    def ask(context: "CommandContext", params: dict) -> bool:
+        filters = ["CSV (*.csv)"]
+
+        filename, selected_filter = FileDialog.open(
+            context.app,
+            dir=None,
+            caption="Import DeepLabCut dataset...",
+            filter=";;".join(filters),
+        )
+
+        if len(filename) == 0:
+            return False
+
+        params["filename"] = filename
+
+        return True
+
+
 class SaveProjectAs(AppCommand):
     @staticmethod
     def _try_save(context, labels: Labels, filename: str):