lazylabel-gui 1.0.7__py3-none-any.whl → 1.0.9__py3-none-any.whl

lazylabel/controls.py

@@ -11,6 +11,9 @@ from PyQt6.QtWidgets import (
     QComboBox,
     QHeaderView,
     QCheckBox,
+    QSlider,
+    QGroupBox,
+    QSplitter,
 )
 from PyQt6.QtCore import Qt
 from .reorderable_class_table import ReorderableClassTable
@@ -21,88 +24,176 @@ class ControlPanel(QWidget):
         super().__init__(parent)
         layout = QVBoxLayout(self)
         layout.setAlignment(Qt.AlignmentFlag.AlignTop)
+
+        toggle_layout = QHBoxLayout()
+        self.btn_toggle_visibility = QPushButton("< Hide")
+        self.btn_toggle_visibility.setToolTip("Hide this panel")
+        toggle_layout.addWidget(self.btn_toggle_visibility)
+        toggle_layout.addStretch()
+        layout.addLayout(toggle_layout)
+
+        self.main_controls_widget = QWidget()
+        main_layout = QVBoxLayout(self.main_controls_widget)
+        main_layout.setContentsMargins(0, 0, 0, 0)
+
         self.mode_label = QLabel("Mode: Points")
         font = self.mode_label.font()
         font.setPointSize(14)
         font.setBold(True)
         self.mode_label.setFont(font)
-        layout.addWidget(self.mode_label)
+        main_layout.addWidget(self.mode_label)
 
-        # Mode Buttons
         self.btn_sam_mode = QPushButton("Point Mode (1)")
         self.btn_sam_mode.setToolTip("Switch to Point Mode for AI segmentation (1)")
         self.btn_polygon_mode = QPushButton("Polygon Mode (2)")
         self.btn_polygon_mode.setToolTip("Switch to Polygon Drawing Mode (2)")
         self.btn_selection_mode = QPushButton("Selection Mode (E)")
         self.btn_selection_mode.setToolTip("Toggle segment selection (E)")
-        layout.addWidget(self.btn_sam_mode)
-        layout.addWidget(self.btn_polygon_mode)
-        layout.addWidget(self.btn_selection_mode)
+        main_layout.addWidget(self.btn_sam_mode)
+        main_layout.addWidget(self.btn_polygon_mode)
+        main_layout.addWidget(self.btn_selection_mode)
 
-        layout.addSpacing(20)
+        main_layout.addSpacing(20)
         line1 = QFrame()
         line1.setFrameShape(QFrame.Shape.HLine)
-        layout.addWidget(line1)
-        layout.addSpacing(10)
+        main_layout.addWidget(line1)
+        main_layout.addSpacing(10)
 
-        # Action Buttons
         self.btn_fit_view = QPushButton("Fit View (.)")
         self.btn_fit_view.setToolTip("Reset image zoom and pan to fit the view (.)")
         self.btn_clear_points = QPushButton("Clear Clicks (C)")
         self.btn_clear_points.setToolTip("Clear current temporary points/vertices (C)")
-        layout.addWidget(self.btn_fit_view)
-        layout.addWidget(self.btn_clear_points)
+        main_layout.addWidget(self.btn_fit_view)
+        main_layout.addWidget(self.btn_clear_points)
 
-        layout.addSpacing(10)
+        main_layout.addSpacing(10)
+
+        settings_group = QGroupBox("Settings")
+        settings_layout = QVBoxLayout()
 
-        # Settings
         self.chk_auto_save = QCheckBox("Auto-Save on Navigate")
         self.chk_auto_save.setToolTip(
             "Automatically save work when using arrow keys to change images."
         )
         self.chk_auto_save.setChecked(True)
-        layout.addWidget(self.chk_auto_save)
+        settings_layout.addWidget(self.chk_auto_save)
+
+        self.chk_save_npz = QCheckBox("Save .npz")
+        self.chk_save_npz.setChecked(True)
+        self.chk_save_npz.setToolTip(
+            "Save the final mask as a compressed NumPy NPZ file."
+        )
+        settings_layout.addWidget(self.chk_save_npz)
+
+        self.chk_save_txt = QCheckBox("Save .txt")
+        self.chk_save_txt.setChecked(True)
+        self.chk_save_txt.setToolTip(
+            "Save bounding box annotations in YOLO TXT format."
+        )
+        settings_layout.addWidget(self.chk_save_txt)
+
+        self.chk_yolo_use_alias = QCheckBox("Save YOLO with Class Aliases")
+        self.chk_yolo_use_alias.setToolTip(
+            "If checked, saves YOLO .txt files using class alias names instead of numeric IDs.\nThis is useful when a separate .yaml or .names file defines the classes."
+        )
+        self.chk_yolo_use_alias.setChecked(True)
+        settings_layout.addWidget(self.chk_yolo_use_alias)
+
+        self.chk_save_class_aliases = QCheckBox("Save Class Aliases (.json)")
+        self.chk_save_class_aliases.setToolTip(
+            "Save class aliases to a companion JSON file."
+        )
+        self.chk_save_class_aliases.setChecked(False)
+        settings_layout.addWidget(self.chk_save_class_aliases)
+
+        settings_group.setLayout(settings_layout)
+        main_layout.addWidget(settings_group)
+
+        sliders_group = QGroupBox("Adjustments")
+        sliders_layout = QVBoxLayout()
+
+        self.size_label = QLabel("Annotation Size: 1.0x")
+        self.size_slider = QSlider(Qt.Orientation.Horizontal)
+        self.size_slider.setRange(1, 50)
+        self.size_slider.setValue(10)
+        self.size_slider.setToolTip("Adjusts the size of points and lines (Ctrl +/-)")
+        sliders_layout.addWidget(self.size_label)
+        sliders_layout.addWidget(self.size_slider)
+
+        sliders_layout.addSpacing(10)
+
+        self.pan_label = QLabel("Pan Speed: 1.0x")
+        self.pan_slider = QSlider(Qt.Orientation.Horizontal)
+        self.pan_slider.setRange(1, 100)
+        self.pan_slider.setValue(10)
+        self.pan_slider.setToolTip(
+            "Adjusts the speed of WASD panning. Hold Shift for 5x boost."
+        )
+        sliders_layout.addWidget(self.pan_label)
+        sliders_layout.addWidget(self.pan_slider)
+
+        sliders_layout.addSpacing(10)
+
+        self.join_label = QLabel("Polygon Join Distance: 2px")
+        self.join_slider = QSlider(Qt.Orientation.Horizontal)
+        self.join_slider.setRange(1, 10)
+        self.join_slider.setValue(2)
+        self.join_slider.setToolTip("The pixel distance to 'snap' a polygon closed.")
+        sliders_layout.addWidget(self.join_label)
+        sliders_layout.addWidget(self.join_slider)
 
-        layout.addStretch()
+        sliders_group.setLayout(sliders_layout)
+        main_layout.addWidget(sliders_group)
+
+        main_layout.addStretch()
 
-        # Notification Label
         self.notification_label = QLabel("")
         font = self.notification_label.font()
         font.setItalic(True)
         self.notification_label.setFont(font)
-        self.notification_label.setStyleSheet(
-            "color: #ffa500;"
-        )  # Orange color for visibility
+        self.notification_label.setStyleSheet("color: #ffa500;")
         self.notification_label.setWordWrap(True)
-        layout.addWidget(self.notification_label)
+        main_layout.addWidget(self.notification_label)
 
-        # Device Label
         self.device_label = QLabel("Device: Unknown")
-        layout.addWidget(self.device_label)
+        main_layout.addWidget(self.device_label)
+
+        layout.addWidget(self.main_controls_widget)
         self.setFixedWidth(250)
 
 
 class RightPanel(QWidget):
     def __init__(self, parent=None):
         super().__init__(parent)
-        layout = QVBoxLayout(self)
+        self.v_layout = QVBoxLayout(self)
+
+        toggle_layout = QHBoxLayout()
+        toggle_layout.addStretch()
+        self.btn_toggle_visibility = QPushButton("Hide >")
+        self.btn_toggle_visibility.setToolTip("Hide this panel")
+        toggle_layout.addWidget(self.btn_toggle_visibility)
+        self.v_layout.addLayout(toggle_layout)
+
+        self.main_controls_widget = QWidget()
+        main_layout = QVBoxLayout(self.main_controls_widget)
+        main_layout.setContentsMargins(0, 0, 0, 0)
+
+        v_splitter = QSplitter(Qt.Orientation.Vertical)
 
-        # File Explorer
-        file_explorer_layout = QVBoxLayout()
+        file_explorer_widget = QWidget()
+        file_explorer_layout = QVBoxLayout(file_explorer_widget)
+        file_explorer_layout.setContentsMargins(0, 0, 0, 0)
         self.btn_open_folder = QPushButton("Open Image Folder")
         self.btn_open_folder.setToolTip("Open a directory of images")
         self.file_tree = QTreeView()
         file_explorer_layout.addWidget(self.btn_open_folder)
         file_explorer_layout.addWidget(self.file_tree)
-        layout.addLayout(file_explorer_layout)
+        v_splitter.addWidget(file_explorer_widget)
 
-        # Status Label
-        self.status_label = QLabel("")
-        self.status_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
-        layout.addWidget(self.status_label)
+        segment_widget = QWidget()
+        segment_layout = QVBoxLayout(segment_widget)
+        segment_layout.setContentsMargins(0, 0, 0, 0)
 
-        # Segment Table
-        segment_layout = QVBoxLayout()
         class_filter_layout = QHBoxLayout()
         class_filter_layout.addWidget(QLabel("Filter Class:"))
         self.class_filter_combo = QComboBox()
@@ -112,7 +203,9 @@ class RightPanel(QWidget):
 
         self.segment_table = QTableWidget()
         self.segment_table.setColumnCount(3)
-        self.segment_table.setHorizontalHeaderLabels(["Index", "Class ID", "Type"])
+        self.segment_table.setHorizontalHeaderLabels(
+            ["Segment ID", "Class ID", "Alias"]
+        )
         self.segment_table.horizontalHeader().setSectionResizeMode(
             QHeaderView.ResizeMode.Stretch
         )
@@ -120,6 +213,7 @@ class RightPanel(QWidget):
             QAbstractItemView.SelectionBehavior.SelectRows
         )
         self.segment_table.setSortingEnabled(True)
+        self.segment_table.setEditTriggers(QAbstractItemView.EditTrigger.NoEditTriggers)
         segment_layout.addWidget(self.segment_table)
 
         segment_action_layout = QHBoxLayout()
@@ -134,30 +228,38 @@ class RightPanel(QWidget):
         segment_action_layout.addWidget(self.btn_merge_selection)
         segment_action_layout.addWidget(self.btn_delete_selection)
         segment_layout.addLayout(segment_action_layout)
-        layout.addLayout(segment_layout, 2)
+        v_splitter.addWidget(segment_widget)
 
-        # Class Table
-        class_layout = QVBoxLayout()
+        class_widget = QWidget()
+        class_layout = QVBoxLayout(class_widget)
+        class_layout.setContentsMargins(0, 0, 0, 0)
         class_layout.addWidget(QLabel("Class Order:"))
         self.class_table = ReorderableClassTable()
         self.class_table.setToolTip(
-            "Set class aliases and drag to reorder channels for saving."
+            "Double-click to set class aliases and drag to reorder channels for saving."
         )
         self.class_table.setColumnCount(2)
-        self.class_table.setHorizontalHeaderLabels(["Alias", "Channel Index"])
+        self.class_table.setHorizontalHeaderLabels(["Alias", "Class ID"])
         self.class_table.horizontalHeader().setSectionResizeMode(
             0, QHeaderView.ResizeMode.Stretch
         )
         self.class_table.horizontalHeader().setSectionResizeMode(
             1, QHeaderView.ResizeMode.ResizeToContents
         )
-
+        self.class_table.setEditTriggers(QAbstractItemView.EditTrigger.DoubleClicked)
         class_layout.addWidget(self.class_table)
         self.btn_reassign_classes = QPushButton("Reassign Class IDs")
         self.btn_reassign_classes.setToolTip(
             "Re-index class channels based on the current order in this table"
         )
         class_layout.addWidget(self.btn_reassign_classes)
-        layout.addLayout(class_layout, 1)
+        v_splitter.addWidget(class_widget)
+
+        main_layout.addWidget(v_splitter)
+
+        self.status_label = QLabel("")
+        self.status_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
+        main_layout.addWidget(self.status_label)
 
+        self.v_layout.addWidget(self.main_controls_widget)
         self.setFixedWidth(350)

lazylabel/custom_file_system_model.py

@@ -1,4 +1,4 @@
-import os
+from pathlib import Path
 from PyQt6.QtCore import Qt, QModelIndex, QDir
 from PyQt6.QtGui import QFileSystemModel, QBrush, QColor
 
@@ -11,13 +11,71 @@ class CustomFileSystemModel(QFileSystemModel):
         self.setNameFilters(["*.png", "*.jpg", "*.jpeg", "*.tiff", "*.tif"])
         self.highlighted_path = None
 
+        self.npz_files = set()
+        self.txt_files = set()
+
+    def setRootPath(self, path: str) -> QModelIndex:
+        self._scan_directory(path)
+        return super().setRootPath(path)
+
+    def _scan_directory(self, path: str):
+        """Scans the directory once and caches the basenames of .npz and .txt files."""
+        self.npz_files.clear()
+        self.txt_files.clear()
+        if not path:
+            return
+
+        directory = Path(path)
+        if not directory.is_dir():
+            return
+
+        try:
+            for file_path in directory.iterdir():
+                if file_path.suffix == ".npz":
+                    self.npz_files.add(file_path.stem)
+                elif file_path.suffix == ".txt":
+                    self.txt_files.add(file_path.stem)
+        except OSError:
+            pass
+
+    def update_cache_for_path(self, saved_file_path: str):
+        """Incrementally updates the cache and the view for a newly saved file."""
+        if not saved_file_path:
+            return
+
+        p = Path(saved_file_path)
+        base_name = p.stem
+
+        if p.suffix == ".npz":
+            self.npz_files.add(base_name)
+        elif p.suffix == ".txt":
+            self.txt_files.add(base_name)
+        else:
+            return
+
+        # Find the model index for the corresponding image file to refresh its row
+        # This assumes the image file is in the same directory (the root path)
+        root_path = Path(self.rootPath())
+        for image_ext in self.nameFilters():  # e.g., '*.png', '*.jpg'
+            # Construct full path to the potential image file
+            image_file = root_path / (base_name + image_ext.replace("*", ""))
+            index = self.index(str(image_file))
+
+            if index.isValid() and index.row() != -1:
+                # Found the corresponding image file, emit signal to refresh its checkmarks
+                index_col1 = self.index(index.row(), 1, index.parent())
+                index_col2 = self.index(index.row(), 2, index.parent())
+                self.dataChanged.emit(
+                    index_col1, index_col2, [Qt.ItemDataRole.CheckStateRole]
+                )
+                break
+
     def set_highlighted_path(self, path):
-        self.highlighted_path = os.path.normpath(path) if path else None
-        # Trigger repaint of the entire view
+        self.highlighted_path = str(Path(path)) if path else None
         self.layoutChanged.emit()
 
     def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
-        return 2
+        return 3
 
     def headerData(
         self,
@@ -32,32 +90,37 @@ class CustomFileSystemModel(QFileSystemModel):
             if section == 0:
                 return "File Name"
             if section == 1:
-                return "Mask"
+                return ".npz"
+            if section == 2:
+                return ".txt"
         return super().headerData(section, orientation, role)
 
     def data(self, index: QModelIndex, role: int = Qt.ItemDataRole.DisplayRole):
         if not index.isValid():
             return None
 
-        # Handle the temporary highlight for saving
         if role == Qt.ItemDataRole.BackgroundRole:
-            filePath = os.path.normpath(self.filePath(index))
-            if (
-                self.highlighted_path
-                and os.path.splitext(filePath)[0]
-                == os.path.splitext(self.highlighted_path)[0]
-            ):
-                return QBrush(QColor(40, 80, 40))  # Dark green highlight
-
-        if index.column() == 1:
-            if role == Qt.ItemDataRole.CheckStateRole:
-                filePath = self.filePath(index.siblingAtColumn(0))
-                mask_path = os.path.splitext(filePath)[0] + ".npz"
-                return (
-                    Qt.CheckState.Checked
-                    if os.path.exists(mask_path)
-                    else Qt.CheckState.Unchecked
-                )
-            return None
+            filePath = self.filePath(index)
+            if self.highlighted_path:
+                p_file = Path(filePath)
+                p_highlight = Path(self.highlighted_path)
+                if p_file.with_suffix("") == p_highlight.with_suffix(""):
+                    return QBrush(QColor(40, 80, 40))
+
+        if index.column() > 0 and role == Qt.ItemDataRole.CheckStateRole:
+            fileName = self.fileName(index.siblingAtColumn(0))
+            base_name = Path(fileName).stem
+
+            if index.column() == 1:
+                exists = base_name in self.npz_files
+            elif index.column() == 2:
+                exists = base_name in self.txt_files
+            else:
+                return None
+
+            return Qt.CheckState.Checked if exists else Qt.CheckState.Unchecked
+
+        if index.column() > 0 and role == Qt.ItemDataRole.DisplayRole:
+            return ""
 
         return super().data(index, role)