Update app.py

added dropdown menu button options for exporting csvs with all frames (ln 495-521)
talmolab · Dec 3, 2023 · 8635ec0 · 8635ec0
1 parent bedb879
commit 8635ec0
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/sleap/gui/app.py b/sleap/gui/app.py
@@ -496,15 +496,28 @@ def add_submenu_choices(menu, title, options, key):
         add_menu_item(
             export_csv_menu,
             "export_csv_current",
-            "Current Video...",
+            "Current Video (only frames with tracking)...",
             self.commands.exportCSVFile,
         )
         add_menu_item(
             export_csv_menu,
             "export_csv_all",
-            "All Videos...",
+            "All Videos (only frames with tracking)...",
             lambda: self.commands.exportCSVFile(all_videos=True),
         )
+
+        fileMenu.addSeparator()
+        add_menu_item(
+            export_csv_menu,
+            "export_csv_current",
+            "Current Video (all frames)...",
+            lambda: self.commands.exportCSVFile(all_frames=True),
+        )
+        add_menu_item(
+            export_csv_menu,
+            "export_csv_all",
+            "All Videos (all frames)...",
+            lambda: self.commands.exportCSVFile(all_frames=True, all_videos=True),
 
         add_menu_item(fileMenu, "export_nwb", "Export NWB...", self.commands.exportNWB)