lazylabel-gui 1.1.1__py3-none-any.whl → 1.1.2__py3-none-any.whl

lazylabel/ui/widgets/adjustments_widget.py

@@ -1,107 +1,108 @@
-"""Adjustments widget for sliders and controls."""
-
-from PyQt6.QtWidgets import QWidget, QVBoxLayout, QLabel, QSlider, QGroupBox
-from PyQt6.QtCore import Qt, pyqtSignal
-
-
-class AdjustmentsWidget(QWidget):
-    """Widget for adjustment controls."""
-    
-    annotation_size_changed = pyqtSignal(int)
-    pan_speed_changed = pyqtSignal(int)
-    join_threshold_changed = pyqtSignal(int)
-    
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        self._setup_ui()
-        self._connect_signals()
-    
-    def _setup_ui(self):
-        """Setup the UI layout."""
-        group = QGroupBox("Adjustments")
-        layout = QVBoxLayout(group)
-        
-        # Annotation size
-        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 +/-)")
-        layout.addWidget(self.size_label)
-        layout.addWidget(self.size_slider)
-        
-        layout.addSpacing(10)
-        
-        # Pan speed
-        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."
-        )
-        layout.addWidget(self.pan_label)
-        layout.addWidget(self.pan_slider)
-        
-        layout.addSpacing(10)
-        
-        # Polygon join threshold
-        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.")
-        layout.addWidget(self.join_label)
-        layout.addWidget(self.join_slider)
-        
-        # Main layout
-        main_layout = QVBoxLayout(self)
-        main_layout.setContentsMargins(0, 0, 0, 0)
-        main_layout.addWidget(group)
-    
-    def _connect_signals(self):
-        """Connect internal signals."""
-        self.size_slider.valueChanged.connect(self._on_size_changed)
-        self.pan_slider.valueChanged.connect(self._on_pan_changed)
-        self.join_slider.valueChanged.connect(self._on_join_changed)
-    
-    def _on_size_changed(self, value):
-        """Handle annotation size change."""
-        multiplier = value / 10.0
-        self.size_label.setText(f"Annotation Size: {multiplier:.1f}x")
-        self.annotation_size_changed.emit(value)
-    
-    def _on_pan_changed(self, value):
-        """Handle pan speed change."""
-        multiplier = value / 10.0
-        self.pan_label.setText(f"Pan Speed: {multiplier:.1f}x")
-        self.pan_speed_changed.emit(value)
-    
-    def _on_join_changed(self, value):
-        """Handle join threshold change."""
-        self.join_label.setText(f"Polygon Join Distance: {value}px")
-        self.join_threshold_changed.emit(value)
-    
-    def get_annotation_size(self):
-        """Get current annotation size value."""
-        return self.size_slider.value()
-    
-    def set_annotation_size(self, value):
-        """Set annotation size value."""
-        self.size_slider.setValue(value)
-    
-    def get_pan_speed(self):
-        """Get current pan speed value."""
-        return self.pan_slider.value()
-    
-    def set_pan_speed(self, value):
-        """Set pan speed value."""
-        self.pan_slider.setValue(value)
-    
-    def get_join_threshold(self):
-        """Get current join threshold value."""
-        return self.join_slider.value()
-    
-    def set_join_threshold(self, value):
-        """Set join threshold value."""
-        self.join_slider.setValue(value)
+"""Adjustments widget for sliders and controls."""
+
+from PyQt6.QtWidgets import QWidget, QVBoxLayout, QLabel, QSlider, QGroupBox
+from PyQt6.QtCore import Qt, pyqtSignal
+
+
+class AdjustmentsWidget(QWidget):
+    """Widget for adjustment controls."""
+    
+    annotation_size_changed = pyqtSignal(int)
+    pan_speed_changed = pyqtSignal(int)
+    join_threshold_changed = pyqtSignal(int)
+    
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._setup_ui()
+        self._connect_signals()
+    
+    def _setup_ui(self):
+        """Setup the UI layout."""
+        group = QGroupBox("Adjustments")
+        layout = QVBoxLayout(group)
+        
+        # Annotation size
+        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 +/-)")
+        layout.addWidget(self.size_label)
+        layout.addWidget(self.size_slider)
+        
+        layout.addSpacing(10)
+        
+        # Pan speed
+        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."
+        )
+        layout.addWidget(self.pan_label)
+        layout.addWidget(self.pan_slider)
+        
+        layout.addSpacing(10)
+        
+        # Polygon join threshold
+        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.")
+        layout.addWidget(self.join_label)
+        layout.addWidget(self.join_slider)
+        
+        # Main layout
+        main_layout = QVBoxLayout(self)
+        main_layout.setContentsMargins(0, 0, 0, 0)
+        main_layout.addWidget(group)
+    
+    def _connect_signals(self):
+        """Connect internal signals."""
+        self.size_slider.valueChanged.connect(self._on_size_changed)
+        self.pan_slider.valueChanged.connect(self._on_pan_changed)
+        self.join_slider.valueChanged.connect(self._on_join_changed)
+    
+    def _on_size_changed(self, value):
+        """Handle annotation size change."""
+        multiplier = value / 10.0
+        self.size_label.setText(f"Annotation Size: {multiplier:.1f}x")
+        self.annotation_size_changed.emit(value)
+    
+    def _on_pan_changed(self, value):
+        """Handle pan speed change."""
+        multiplier = value / 10.0
+        self.pan_label.setText(f"Pan Speed: {multiplier:.1f}x")
+        self.pan_speed_changed.emit(value)
+    
+    def _on_join_changed(self, value):
+        """Handle join threshold change."""
+        self.join_label.setText(f"Polygon Join Distance: {value}px")
+        self.join_threshold_changed.emit(value)
+    
+    def get_annotation_size(self):
+        """Get current annotation size value."""
+        return self.size_slider.value()
+    
+    def set_annotation_size(self, value):
+        """Set annotation size value."""
+        self.size_slider.setValue(value)
+    
+    def get_pan_speed(self):
+        """Get current pan speed value."""
+        return self.pan_slider.value()
+    
+    def set_pan_speed(self, value):
+        """Set pan speed value."""
+        self.pan_slider.setValue(value)
+    
+    def get_join_threshold(self):
+        """Get current join threshold value."""
+        return self.join_slider.value()
+    
+    def set_join_threshold(self, value):
+        """Set join threshold value."""
+        self.join_slider.setValue(value)
+        self.join_label.setText(f"Polygon Join Distance: {value}px")

lazylabel/ui/widgets/model_selection_widget.py

@@ -1,94 +1,94 @@
-"""Model selection widget."""
-
-from PyQt6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel, QComboBox, QGroupBox
-from PyQt6.QtCore import pyqtSignal
-from typing import List, Tuple
-
-
-class ModelSelectionWidget(QWidget):
-    """Widget for model selection and management."""
-    
-    browse_requested = pyqtSignal()
-    refresh_requested = pyqtSignal()
-    model_selected = pyqtSignal(str)
-    
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        self._setup_ui()
-        self._connect_signals()
-    
-    def _setup_ui(self):
-        """Setup the UI layout."""
-        group = QGroupBox("Model Selection")
-        layout = QVBoxLayout(group)
-        
-        # Buttons
-        button_layout = QHBoxLayout()
-        self.btn_browse = QPushButton("Browse Models")
-        self.btn_browse.setToolTip("Browse for a folder containing .pth model files")
-        self.btn_refresh = QPushButton("Refresh")
-        self.btn_refresh.setToolTip("Refresh the list of available models")
-        
-        button_layout.addWidget(self.btn_browse)
-        button_layout.addWidget(self.btn_refresh)
-        layout.addLayout(button_layout)
-        
-        # Model combo
-        layout.addWidget(QLabel("Available Models:"))
-        self.model_combo = QComboBox()
-        self.model_combo.setToolTip("Select a .pth model file to use")
-        self.model_combo.addItem("Default (vit_h)")
-        layout.addWidget(self.model_combo)
-        
-        # Current model label
-        self.current_model_label = QLabel("Current: Default SAM Model")
-        self.current_model_label.setWordWrap(True)
-        self.current_model_label.setStyleSheet("color: #90EE90; font-style: italic;")
-        layout.addWidget(self.current_model_label)
-        
-        # Main layout
-        main_layout = QVBoxLayout(self)
-        main_layout.setContentsMargins(0, 0, 0, 0)
-        main_layout.addWidget(group)
-    
-    def _connect_signals(self):
-        """Connect internal signals."""
-        self.btn_browse.clicked.connect(self.browse_requested)
-        self.btn_refresh.clicked.connect(self.refresh_requested)
-        self.model_combo.currentTextChanged.connect(self.model_selected)
-    
-    def populate_models(self, models: List[Tuple[str, str]]):
-        """Populate the models combo box.
-        
-        Args:
-            models: List of (display_name, full_path) tuples
-        """
-        self.model_combo.blockSignals(True)
-        self.model_combo.clear()
-        
-        # Add default option
-        self.model_combo.addItem("Default (vit_h)")
-        
-        # Add custom models
-        for display_name, full_path in models:
-            self.model_combo.addItem(display_name, full_path)
-        
-        self.model_combo.blockSignals(False)
-    
-    def set_current_model(self, model_name: str):
-        """Set the current model display."""
-        self.current_model_label.setText(model_name)
-    
-    def get_selected_model_path(self) -> str:
-        """Get the path of the currently selected model."""
-        current_index = self.model_combo.currentIndex()
-        if current_index <= 0:  # Default option
-            return ""
-        return self.model_combo.itemData(current_index) or ""
-    
-    def reset_to_default(self):
-        """Reset selection to default model."""
-        self.model_combo.blockSignals(True)
-        self.model_combo.setCurrentIndex(0)
-        self.model_combo.blockSignals(False)
+"""Model selection widget."""
+
+from PyQt6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QPushButton, QLabel, QComboBox, QGroupBox
+from PyQt6.QtCore import pyqtSignal
+from typing import List, Tuple
+
+
+class ModelSelectionWidget(QWidget):
+    """Widget for model selection and management."""
+    
+    browse_requested = pyqtSignal()
+    refresh_requested = pyqtSignal()
+    model_selected = pyqtSignal(str)
+    
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._setup_ui()
+        self._connect_signals()
+    
+    def _setup_ui(self):
+        """Setup the UI layout."""
+        group = QGroupBox("Model Selection")
+        layout = QVBoxLayout(group)
+        
+        # Buttons
+        button_layout = QHBoxLayout()
+        self.btn_browse = QPushButton("Browse Models")
+        self.btn_browse.setToolTip("Browse for a folder containing .pth model files")
+        self.btn_refresh = QPushButton("Refresh")
+        self.btn_refresh.setToolTip("Refresh the list of available models")
+        
+        button_layout.addWidget(self.btn_browse)
+        button_layout.addWidget(self.btn_refresh)
+        layout.addLayout(button_layout)
+        
+        # Model combo
+        layout.addWidget(QLabel("Available Models:"))
+        self.model_combo = QComboBox()
+        self.model_combo.setToolTip("Select a .pth model file to use")
+        self.model_combo.addItem("Default (vit_h)")
+        layout.addWidget(self.model_combo)
+        
+        # Current model label
+        self.current_model_label = QLabel("Current: Default SAM Model")
+        self.current_model_label.setWordWrap(True)
+        self.current_model_label.setStyleSheet("color: #90EE90; font-style: italic;")
+        layout.addWidget(self.current_model_label)
+        
+        # Main layout
+        main_layout = QVBoxLayout(self)
+        main_layout.setContentsMargins(0, 0, 0, 0)
+        main_layout.addWidget(group)
+    
+    def _connect_signals(self):
+        """Connect internal signals."""
+        self.btn_browse.clicked.connect(self.browse_requested)
+        self.btn_refresh.clicked.connect(self.refresh_requested)
+        self.model_combo.currentTextChanged.connect(self.model_selected)
+    
+    def populate_models(self, models: List[Tuple[str, str]]):
+        """Populate the models combo box.
+        
+        Args:
+            models: List of (display_name, full_path) tuples
+        """
+        self.model_combo.blockSignals(True)
+        self.model_combo.clear()
+        
+        # Add default option
+        self.model_combo.addItem("Default (vit_h)")
+        
+        # Add custom models
+        for display_name, full_path in models:
+            self.model_combo.addItem(display_name, full_path)
+        
+        self.model_combo.blockSignals(False)
+    
+    def set_current_model(self, model_name: str):
+        """Set the current model display."""
+        self.current_model_label.setText(model_name)
+    
+    def get_selected_model_path(self) -> str:
+        """Get the path of the currently selected model."""
+        current_index = self.model_combo.currentIndex()
+        if current_index <= 0:  # Default option
+            return ""
+        return self.model_combo.itemData(current_index) or ""
+    
+    def reset_to_default(self):
+        """Reset selection to default model."""
+        self.model_combo.blockSignals(True)
+        self.model_combo.setCurrentIndex(0)
+        self.model_combo.blockSignals(False)
         self.set_current_model("Current: Default SAM Model")

lazylabel/ui/widgets/settings_widget.py

@@ -1,106 +1,106 @@
-"""Settings widget for save options."""
-
-from PyQt6.QtWidgets import QWidget, QVBoxLayout, QCheckBox, QGroupBox
-from PyQt6.QtCore import pyqtSignal
-
-
-class SettingsWidget(QWidget):
-    """Widget for application settings."""
-    
-    settings_changed = pyqtSignal()
-    
-    def __init__(self, parent=None):
-        super().__init__(parent)
-        self._setup_ui()
-        self._connect_signals()
-    
-    def _setup_ui(self):
-        """Setup the UI layout."""
-        group = QGroupBox("Settings")
-        layout = QVBoxLayout(group)
-        
-        # Auto-save
-        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)
-        
-        # Save NPZ
-        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."
-        )
-        layout.addWidget(self.chk_save_npz)
-        
-        # Save TXT
-        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."
-        )
-        layout.addWidget(self.chk_save_txt)
-        
-        # YOLO with aliases
-        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.\n"
-            "This is useful when a separate .yaml or .names file defines the classes."
-        )
-        self.chk_yolo_use_alias.setChecked(True)
-        layout.addWidget(self.chk_yolo_use_alias)
-        
-        # Save class aliases
-        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)
-        layout.addWidget(self.chk_save_class_aliases)
-        
-        # Main layout
-        main_layout = QVBoxLayout(self)
-        main_layout.setContentsMargins(0, 0, 0, 0)
-        main_layout.addWidget(group)
-    
-    def _connect_signals(self):
-        """Connect internal signals."""
-        self.chk_save_npz.stateChanged.connect(self._handle_save_checkbox_change)
-        self.chk_save_txt.stateChanged.connect(self._handle_save_checkbox_change)
-        
-        # Connect all checkboxes to settings changed signal
-        for checkbox in [self.chk_auto_save, self.chk_save_npz, self.chk_save_txt,
-                        self.chk_yolo_use_alias, self.chk_save_class_aliases]:
-            checkbox.stateChanged.connect(self.settings_changed)
-    
-    def _handle_save_checkbox_change(self):
-        """Ensure at least one save format is selected."""
-        is_npz_checked = self.chk_save_npz.isChecked()
-        is_txt_checked = self.chk_save_txt.isChecked()
-        
-        if not is_npz_checked and not is_txt_checked:
-            sender = self.sender()
-            if sender == self.chk_save_npz:
-                self.chk_save_txt.setChecked(True)
-            else:
-                self.chk_save_npz.setChecked(True)
-    
-    def get_settings(self):
-        """Get current settings as dictionary."""
-        return {
-            'auto_save': self.chk_auto_save.isChecked(),
-            'save_npz': self.chk_save_npz.isChecked(),
-            'save_txt': self.chk_save_txt.isChecked(),
-            'yolo_use_alias': self.chk_yolo_use_alias.isChecked(),
-            'save_class_aliases': self.chk_save_class_aliases.isChecked(),
-        }
-    
-    def set_settings(self, settings):
-        """Set settings from dictionary."""
-        self.chk_auto_save.setChecked(settings.get('auto_save', True))
-        self.chk_save_npz.setChecked(settings.get('save_npz', True))
-        self.chk_save_txt.setChecked(settings.get('save_txt', True))
-        self.chk_yolo_use_alias.setChecked(settings.get('yolo_use_alias', True))
+"""Settings widget for save options."""
+
+from PyQt6.QtWidgets import QWidget, QVBoxLayout, QCheckBox, QGroupBox
+from PyQt6.QtCore import pyqtSignal
+
+
+class SettingsWidget(QWidget):
+    """Widget for application settings."""
+    
+    settings_changed = pyqtSignal()
+    
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._setup_ui()
+        self._connect_signals()
+    
+    def _setup_ui(self):
+        """Setup the UI layout."""
+        group = QGroupBox("Settings")
+        layout = QVBoxLayout(group)
+        
+        # Auto-save
+        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)
+        
+        # Save NPZ
+        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."
+        )
+        layout.addWidget(self.chk_save_npz)
+        
+        # Save TXT
+        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."
+        )
+        layout.addWidget(self.chk_save_txt)
+        
+        # YOLO with aliases
+        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.\n"
+            "This is useful when a separate .yaml or .names file defines the classes."
+        )
+        self.chk_yolo_use_alias.setChecked(True)
+        layout.addWidget(self.chk_yolo_use_alias)
+        
+        # Save class aliases
+        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)
+        layout.addWidget(self.chk_save_class_aliases)
+        
+        # Main layout
+        main_layout = QVBoxLayout(self)
+        main_layout.setContentsMargins(0, 0, 0, 0)
+        main_layout.addWidget(group)
+    
+    def _connect_signals(self):
+        """Connect internal signals."""
+        self.chk_save_npz.stateChanged.connect(self._handle_save_checkbox_change)
+        self.chk_save_txt.stateChanged.connect(self._handle_save_checkbox_change)
+        
+        # Connect all checkboxes to settings changed signal
+        for checkbox in [self.chk_auto_save, self.chk_save_npz, self.chk_save_txt,
+                        self.chk_yolo_use_alias, self.chk_save_class_aliases]:
+            checkbox.stateChanged.connect(self.settings_changed)
+    
+    def _handle_save_checkbox_change(self):
+        """Ensure at least one save format is selected."""
+        is_npz_checked = self.chk_save_npz.isChecked()
+        is_txt_checked = self.chk_save_txt.isChecked()
+        
+        if not is_npz_checked and not is_txt_checked:
+            sender = self.sender()
+            if sender == self.chk_save_npz:
+                self.chk_save_txt.setChecked(True)
+            else:
+                self.chk_save_npz.setChecked(True)
+    
+    def get_settings(self):
+        """Get current settings as dictionary."""
+        return {
+            'auto_save': self.chk_auto_save.isChecked(),
+            'save_npz': self.chk_save_npz.isChecked(),
+            'save_txt': self.chk_save_txt.isChecked(),
+            'yolo_use_alias': self.chk_yolo_use_alias.isChecked(),
+            'save_class_aliases': self.chk_save_class_aliases.isChecked(),
+        }
+    
+    def set_settings(self, settings):
+        """Set settings from dictionary."""
+        self.chk_auto_save.setChecked(settings.get('auto_save', True))
+        self.chk_save_npz.setChecked(settings.get('save_npz', True))
+        self.chk_save_txt.setChecked(settings.get('save_txt', True))
+        self.chk_yolo_use_alias.setChecked(settings.get('yolo_use_alias', True))
         self.chk_save_class_aliases.setChecked(settings.get('save_class_aliases', False))