lazylabel-gui 1.0.8__py3-none-any.whl → 1.1.0__py3-none-any.whl

lazylabel/utils/custom_file_system_model.py

@@ -0,0 +1,132 @@
+from pathlib import Path
+from PyQt6.QtCore import Qt, QModelIndex, QDir
+from PyQt6.QtGui import QFileSystemModel, QBrush, QColor
+
+
+class CustomFileSystemModel(QFileSystemModel):
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self.setFilter(QDir.Filter.NoDotAndDotDot | QDir.Filter.Files)
+        self.setNameFilterDisables(False)
+        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 or deleted file."""
+        if not saved_file_path:
+            return
+
+        p = Path(saved_file_path)
+        base_name = p.stem
+
+        if p.suffix == ".npz":
+            if p.exists():
+                self.npz_files.add(base_name)
+            else:
+                self.npz_files.discard(base_name)
+        elif p.suffix == ".txt":
+            if p.exists():
+                self.txt_files.add(base_name)
+            else:
+                self.txt_files.discard(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 = str(Path(path)) if path else None
+        self.layoutChanged.emit()
+
+    def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
+        return 3
+
+    def headerData(
+        self,
+        section: int,
+        orientation: Qt.Orientation,
+        role: int = Qt.ItemDataRole.DisplayRole,
+    ):
+        if (
+            orientation == Qt.Orientation.Horizontal
+            and role == Qt.ItemDataRole.DisplayRole
+        ):
+            if section == 0:
+                return "File Name"
+            if section == 1:
+                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
+
+        if role == Qt.ItemDataRole.BackgroundRole:
+            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)

{lazylabel_gui-1.0.8.dist-info → lazylabel_gui-1.1.0.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: lazylabel-gui
-Version: 1.0.8
-Summary: An image segmentation GUI for generating mask tensors.
+Version: 1.1.0
+Summary: An image segmentation GUI for generating ML ready mask tensors and annotations.
 Author-email: "Deniz N. Cakan" <deniz.n.cakan@gmail.com>
 License: MIT License
         
@@ -48,7 +48,8 @@ Requires-Dist: tqdm>=4.67.1
 Dynamic: license-file
 
 # <img src="https://raw.githubusercontent.com/dnzckn/LazyLabel/main/src/lazylabel/demo_pictures/logo2.png" alt="LazyLabel Logo" style="height:60px; vertical-align:middle;" /> <img src="https://raw.githubusercontent.com/dnzckn/LazyLabel/main/src/lazylabel/demo_pictures/logo_black.png" alt="LazyLabel Cursive" style="height:60px; vertical-align:middle;" />
-LazyLabel is an intuitive, AI-assisted image segmentation tool. It uses Meta's Segment Anything Model (SAM) for quick, precise mask generation, alongside advanced polygon editing for fine-tuned control. Outputs are saved in a clean, one-hot encoded `.npz` format for easy machine learning integration and in YOLO `.txt` format.
+
+LazyLabel is an intuitive, AI-assisted image segmentation tool built with a modern, modular architecture. It uses Meta's Segment Anything Model (SAM) for quick, precise mask generation, alongside advanced polygon editing for fine-tuned control. Features comprehensive model management, customizable hotkeys, and outputs in clean, one-hot encoded `.npz` format for easy machine learning integration.
 
 Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#installation) and [Segment-Anything-UI](https://github.com/branislavhesko/segment-anything-ui/tree/main).
 
@@ -58,19 +59,33 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
 
 ## ✨ Core Features
 
-* **AI-Powered Segmentation**: Generate masks with simple left-click (positive) and right-click (negative) interactions.
-* **Vector Polygon Tool**: Full control to draw, edit, and reshape polygons. Drag vertices or move entire shapes.
-* **Advanced Class Management**: Assign multiple segments to a single class ID for organized labeling.
-* **Intuitive Editing & Refinement**: Select, merge, and re-order segments.
-* **Interactive UI**: Color-coded segments, sortable lists, and hover highlighting.
-* **Smart I/O**: Loads existing `.npz` masks; saves work as clean, one-hot encoded outputs.
+### **AI-Powered Segmentation**
+* Generate masks with simple left-click (positive) and right-click (negative) interactions
+* Multiple SAM model support with easy switching
+* Custom model loading from any directory
+
+### **Advanced Editing Tools**
+* **Vector Polygon Tool**: Full control to draw, edit, and reshape polygons
+* **Vertex Editing**: Drag vertices or move entire shapes with precision
+* **Selection & Merging**: Select, merge, and re-order segments intuitively
+
+### **Professional Workflow**
+* **Customizable Hotkeys**: Personalize keyboard shortcuts for all functions
+* **Advanced Class Management**: Assign multiple segments to single class IDs
+* **Smart I/O**: Load existing `.npz` masks; save as clean, one-hot encoded outputs
+* **Interactive UI**: Color-coded segments, sortable lists, and hover highlighting
+
+### **Modern Architecture**
+* **Modular Design**: Clean, maintainable codebase with separated concerns
+* **Model Management**: Dedicated model storage and switching system
+* **Persistent Settings**: User preferences saved between sessions
 
 ---
 
 ## 🚀 Getting Started
 
 ### Prerequisites
-**Python 3.10**
+**Python 3.10+**
 
 ### Installation
 
@@ -90,7 +105,7 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
     git clone https://github.com/dnzckn/LazyLabel.git
     cd LazyLabel
     ```
-2.  Install in editable mode, which links the installed package to your source directory:
+2.  Install in editable mode:
     ```bash
     pip install -e .
     ```
@@ -99,12 +114,20 @@ Inspired by [LabelMe](https://github.com/wkentaro/labelme?tab=readme-ov-file#ins
     lazylabel-gui
     ```
 
-**Note**: On the first run, the application will automatically download the SAM model checkpoint (~2.5 GB) from Meta's repository to a local cache. This is a one-time download.
+### Model Management
+* **Default Storage**: Models are stored in `src/lazylabel/models/` directory
+* **Custom Models**: Click "Browse Models" to select custom model folders  
+* **Model Switching**: Use the dropdown to switch between available models
+* **Auto-Detection**: Application automatically detects all `.pth` files in selected directories
+
+**Note**: On the first run, the application will automatically download the SAM model checkpoint (~2.5 GB) from Meta's repository to the models directory. This is a one-time download.
 
 ---
 
 ## ⌨️ Controls & Keybinds
 
+> **💡 Tip**: All hotkeys are fully customizable! Click the "Hotkeys" button in the control panel to personalize your shortcuts.
+
 ### Modes
 | Key | Action |
 |---|---|
@@ -143,5 +166,32 @@ Each channel is a binary mask for a class, combining all assigned segments into
 
 ---
 
+## 🏗️ Architecture
+
+LazyLabel features a modern, modular architecture designed for maintainability and extensibility:
+
+* **Modular Design**: Clean separation between UI, business logic, and configuration
+* **Signal-Based Communication**: Loose coupling between components using PyQt signals
+* **Persistent Configuration**: User settings and preferences saved between sessions
+* **Extensible Model System**: Easy integration of new SAM models and types
+
+For detailed technical documentation, see [ARCHITECTURE.md](src/lazylabel/ARCHITECTURE.md).
+
+---
+
+## ⌨️ Hotkey Customization
+
+LazyLabel includes a comprehensive hotkey management system:
+
+* **Full Customization**: Personalize keyboard shortcuts for all 27+ functions
+* **Category Organization**: Hotkeys organized by function (Modes, Actions, Navigation, etc.)
+* **Primary & Secondary Keys**: Set multiple shortcuts for the same action
+* **Persistent Settings**: Custom hotkeys saved between sessions
+* **Conflict Prevention**: System prevents duplicate key assignments
+
+For complete hotkey documentation, see [HOTKEY_FEATURE.md](src/lazylabel/HOTKEY_FEATURE.md).
+
+---
+
 ## ☕ Support LazyLabel
 [If you found LazyLabel helpful, consider supporting the project!](https://buymeacoffee.com/dnzckn)

lazylabel_gui-1.1.0.dist-info/RECORD

@@ -0,0 +1,36 @@
+lazylabel/__init__.py,sha256=XTAPk-88MdkJX5IKg0f51CxNlARQ-SZIiqLneV6B3II,197
+lazylabel/main.py,sha256=EQdIfp4GX3Qt9Hz1KFldxKiqjG0Wm0-GmuECdfGwL_8,426
+lazylabel/config/__init__.py,sha256=ahAF6cneEVmq2zyGwydGgGAOzJKkv_aumq5cWmokOms,261
+lazylabel/config/hotkeys.py,sha256=LrQ4nmLWVOZvNOe4mEARFU3Wd3bZ7nvvsxGeFY6N9wA,7618
+lazylabel/config/paths.py,sha256=d6lJ0rwdlnLjrhR5fIzb3Pc9Fhl4lsGIf2IZNmi4vZ8,1320
+lazylabel/config/settings.py,sha256=IkUGz-FdDapF71k_LRpUOLHScglGeYjdLbbELGJ4hOs,1860
+lazylabel/core/__init__.py,sha256=YIj3IAI6Er5nN7gSuo9E8eoQKbBbSq2Jy2prtU8Fp50,230
+lazylabel/core/file_manager.py,sha256=cXaSPSUiqTm57hfpUxkuzgqApOZCK1RAJdfjmKRJKRA,4734
+lazylabel/core/model_manager.py,sha256=Prski8UZCSj6Rkk5uCtJwhsxSSnZPos-RZwM5CgF0KY,3442
+lazylabel/core/segment_manager.py,sha256=mrP6QqeHRFvX4HaX3hUsesnfWqAwUjLVfkxB8Wdy3cY,5097
+lazylabel/models/__init__.py,sha256=qH0EvkWsou0boS85yM6DfRhJnrPOLc-QzWI0grAwnRI,89
+lazylabel/models/sam_model.py,sha256=NPC_Zwc973jos1hV-EfUTlnX44QREgjI-QT2nDd4NUY,6408
+lazylabel/ui/__init__.py,sha256=i5hgblzrydTkFSJDiFyfRZMNl4Z4J2dumuwx3bWEPvA,266
+lazylabel/ui/control_panel.py,sha256=LKPfIgS12rzMbORunUpoHFGKEtEjsJTHtlU1BUAcD7g,8742
+lazylabel/ui/editable_vertex.py,sha256=pMuXvlCkbguUsFisEzj3bC2UZ_xKqfGuM8kyPug5yws,1852
+lazylabel/ui/hotkey_dialog.py,sha256=3QkthYil_SWYWxRGL_U8cZCsQ4y-6SExQZXJuE5aO8s,15252
+lazylabel/ui/hoverable_pixelmap_item.py,sha256=kJFOp7WXiyHpNf7l73TZjiob85jgP30b5MZvu_z5L3c,728
+lazylabel/ui/hoverable_polygon_item.py,sha256=hQKax3vzhtrdB5ThwTINjOCwOYy718zw5YZPLCfLnGM,1251
+lazylabel/ui/main_window.py,sha256=DVBmTtGd8EoiN-Qhl_Cx3o9ajNENyAf-OnYnHTs3mOI,52596
+lazylabel/ui/numeric_table_widget_item.py,sha256=dQUlIFu9syCxTGAHVIlmbgkI7aJ3f3wmDPBz1AGK9Bg,283
+lazylabel/ui/photo_viewer.py,sha256=PNgm0gU2gnIqvRkrGlQugdobGsKwAi3m3X6ZF487lCo,2055
+lazylabel/ui/reorderable_class_table.py,sha256=4c-iuSkPcmk5Aey5n2zz49O85x9TQPujKG-JLxtuBCo,2406
+lazylabel/ui/right_panel.py,sha256=m5PBRtzwB2x8CKtLzRwVYAsp1Mq-hvIGl7KEGMCcLrg,9988
+lazylabel/ui/widgets/__init__.py,sha256=O4FFoTYecZ_dUIpNjqNBVmG5b3HLhv-gYNW7cevbtsE,264
+lazylabel/ui/widgets/adjustments_widget.py,sha256=Az8GZgu4Uvhm6nEnwjiYDyShOqwieinjFnTOiUfxdzo,3940
+lazylabel/ui/widgets/model_selection_widget.py,sha256=X3qVH90yCaCpLDauj-DLWJgAwAkAVQrzhG9X5mESa-o,3590
+lazylabel/ui/widgets/settings_widget.py,sha256=8zhjLxUxqFqxqMYDzkXWD1Ye5V-a7HNhPjPJjYBxeZw,4361
+lazylabel/utils/__init__.py,sha256=SX_aZvmFojIJ4Lskat9ly0dmFBXgN7leBdmc68aDLpg,177
+lazylabel/utils/custom_file_system_model.py,sha256=zK0Z4LY2hzYVtycDWsCLEhXZ0zKZz-Qy5XGwHYrqazQ,4867
+lazylabel/utils/utils.py,sha256=sYSCoXL27OaLgOZaUkCAhgmKZ7YfhR3Cc5F8nDIa3Ig,414
+lazylabel_gui-1.1.0.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
+lazylabel_gui-1.1.0.dist-info/METADATA,sha256=HT1KUNRWJsGUjtU8OhcEcIyK8W9TiMqB_00cmpQElfE,8473
+lazylabel_gui-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+lazylabel_gui-1.1.0.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
+lazylabel_gui-1.1.0.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
+lazylabel_gui-1.1.0.dist-info/RECORD,,

lazylabel/controls.py

@@ -1,261 +0,0 @@
-from PyQt6.QtWidgets import (
-    QWidget,
-    QVBoxLayout,
-    QPushButton,
-    QLabel,
-    QFrame,
-    QTableWidget,
-    QTreeView,
-    QAbstractItemView,
-    QHBoxLayout,
-    QComboBox,
-    QHeaderView,
-    QCheckBox,
-    QSlider,
-    QGroupBox,
-    QSplitter,
-)
-from PyQt6.QtCore import Qt
-from .reorderable_class_table import ReorderableClassTable
-
-
-class ControlPanel(QWidget):
-    def __init__(self, parent=None):
-        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)
-        main_layout.addWidget(self.mode_label)
-
-        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)")
-        main_layout.addWidget(self.btn_sam_mode)
-        main_layout.addWidget(self.btn_polygon_mode)
-        main_layout.addWidget(self.btn_selection_mode)
-
-        main_layout.addSpacing(20)
-        line1 = QFrame()
-        line1.setFrameShape(QFrame.Shape.HLine)
-        main_layout.addWidget(line1)
-        main_layout.addSpacing(10)
-
-        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)")
-        main_layout.addWidget(self.btn_fit_view)
-        main_layout.addWidget(self.btn_clear_points)
-
-        main_layout.addSpacing(10)
-
-        settings_group = QGroupBox("Settings")
-        settings_layout = QVBoxLayout()
-
-        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)
-        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_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)
-
-        sliders_group.setLayout(sliders_layout)
-        main_layout.addWidget(sliders_group)
-
-        main_layout.addStretch()
-
-        self.notification_label = QLabel("")
-        font = self.notification_label.font()
-        font.setItalic(True)
-        self.notification_label.setFont(font)
-        self.notification_label.setStyleSheet("color: #ffa500;")
-        self.notification_label.setWordWrap(True)
-        main_layout.addWidget(self.notification_label)
-
-        self.device_label = QLabel("Device: Unknown")
-        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)
-        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 Widget ---
-        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)
-        v_splitter.addWidget(file_explorer_widget)
-
-        # --- Segment List Widget ---
-        segment_widget = QWidget()
-        segment_layout = QVBoxLayout(segment_widget)
-        segment_layout.setContentsMargins(0, 0, 0, 0)
-
-        class_filter_layout = QHBoxLayout()
-        class_filter_layout.addWidget(QLabel("Filter Class:"))
-        self.class_filter_combo = QComboBox()
-        self.class_filter_combo.setToolTip("Filter segments list by class")
-        class_filter_layout.addWidget(self.class_filter_combo)
-        segment_layout.addLayout(class_filter_layout)
-
-        self.segment_table = QTableWidget()
-        self.segment_table.setColumnCount(3)
-        self.segment_table.setHorizontalHeaderLabels(
-            ["Segment ID", "Class ID", "Alias"]
-        )
-        self.segment_table.horizontalHeader().setSectionResizeMode(
-            QHeaderView.ResizeMode.Stretch
-        )
-        self.segment_table.setSelectionBehavior(
-            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()
-        self.btn_merge_selection = QPushButton("Merge to Class")
-        self.btn_merge_selection.setToolTip(
-            "Merge selected segments into a single class (M)"
-        )
-        self.btn_delete_selection = QPushButton("Delete")
-        self.btn_delete_selection.setToolTip(
-            "Delete selected segments (Delete/Backspace)"
-        )
-        segment_action_layout.addWidget(self.btn_merge_selection)
-        segment_action_layout.addWidget(self.btn_delete_selection)
-        segment_layout.addLayout(segment_action_layout)
-        v_splitter.addWidget(segment_widget)
-
-        # --- Class Table Widget ---
-        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(
-            "Double-click to set class aliases and drag to reorder channels for saving."
-        )
-        self.class_table.setColumnCount(2)
-        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)
-        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,72 +0,0 @@
-import os
-from PyQt6.QtCore import Qt, QModelIndex, QDir
-from PyQt6.QtGui import QFileSystemModel, QBrush, QColor
-
-
-class CustomFileSystemModel(QFileSystemModel):
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        self.setFilter(QDir.Filter.NoDotAndDotDot | QDir.Filter.Files)
-        self.setNameFilterDisables(False)
-        self.setNameFilters(["*.png", "*.jpg", "*.jpeg", "*.tiff", "*.tif"])
-        self.highlighted_path = None
-
-    def set_highlighted_path(self, path):
-        self.highlighted_path = os.path.normpath(path) if path else None
-        self.layoutChanged.emit()
-
-    def columnCount(self, parent: QModelIndex = QModelIndex()) -> int:
-        return 3
-
-    def headerData(
-        self,
-        section: int,
-        orientation: Qt.Orientation,
-        role: int = Qt.ItemDataRole.DisplayRole,
-    ):
-        if (
-            orientation == Qt.Orientation.Horizontal
-            and role == Qt.ItemDataRole.DisplayRole
-        ):
-            if section == 0:
-                return "File Name"
-            if section == 1:
-                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
-
-        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))
-
-        if index.column() > 0 and role == Qt.ItemDataRole.CheckStateRole:
-            filePath = self.filePath(index.siblingAtColumn(0))
-            base_path = os.path.splitext(filePath)[0]
-
-            if index.column() == 1:
-                check_path = base_path + ".npz"
-            elif index.column() == 2:
-                check_path = base_path + ".txt"
-            else:
-                return None
-
-            return (
-                Qt.CheckState.Checked
-                if os.path.exists(check_path)
-                else Qt.CheckState.Unchecked
-            )
-
-        if index.column() > 0 and role == Qt.ItemDataRole.DisplayRole:
-            return ""
-
-        return super().data(index, role)