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

lazylabel/ui/widgets/status_bar.py

@@ -0,0 +1,109 @@
+"""Status bar widget for displaying active messages."""
+
+from PyQt6.QtWidgets import QStatusBar, QLabel
+from PyQt6.QtCore import QTimer, pyqtSignal, Qt
+from PyQt6.QtGui import QFont
+
+
+class StatusBar(QStatusBar):
+    """Custom status bar for displaying messages and app status."""
+
+    def __init__(self, parent=None):
+        super().__init__(parent)
+        self._message_timer = QTimer()
+        self._message_timer.timeout.connect(self._clear_temporary_message)
+        self._setup_ui()
+
+    def _setup_ui(self):
+        """Setup the status bar UI."""
+        # Set a reasonable height for the status bar
+        self.setFixedHeight(25)
+
+        # Main message label (centered)
+        self.message_label = QLabel()
+        self.message_label.setStyleSheet("color: #ffa500; padding: 2px 5px;")
+        self.message_label.setAlignment(Qt.AlignmentFlag.AlignCenter)
+        font = QFont()
+        font.setPointSize(9)
+        self.message_label.setFont(font)
+
+        # Add the message label as the main widget
+        self.addWidget(self.message_label, 1)  # stretch factor 1
+
+        # Permanent status label (right side)
+        self.permanent_label = QLabel()
+        self.permanent_label.setStyleSheet("color: #888; padding: 2px 5px;")
+        font = QFont()
+        font.setPointSize(9)
+        self.permanent_label.setFont(font)
+        self.addPermanentWidget(self.permanent_label)
+
+        # Default state
+        self.set_ready_message()
+
+    def show_message(self, message: str, duration: int = 5000):
+        """Show a temporary message for specified duration."""
+        self.message_label.setText(message)
+        self.message_label.setStyleSheet("color: #ffa500; padding: 2px 5px;")
+
+        # Stop any existing timer
+        self._message_timer.stop()
+
+        # Start new timer if duration > 0
+        if duration > 0:
+            self._message_timer.start(duration)
+
+    def show_error_message(self, message: str, duration: int = 8000):
+        """Show an error message with red color."""
+        self.message_label.setText(f"Error: {message}")
+        self.message_label.setStyleSheet("color: #ff6b6b; padding: 2px 5px;")
+
+        # Stop any existing timer
+        self._message_timer.stop()
+
+        # Start new timer if duration > 0
+        if duration > 0:
+            self._message_timer.start(duration)
+
+    def show_success_message(self, message: str, duration: int = 3000):
+        """Show a success message with green color."""
+        self.message_label.setText(message)
+        self.message_label.setStyleSheet("color: #51cf66; padding: 2px 5px;")
+
+        # Stop any existing timer
+        self._message_timer.stop()
+
+        # Start new timer if duration > 0
+        if duration > 0:
+            self._message_timer.start(duration)
+
+    def show_warning_message(self, message: str, duration: int = 5000):
+        """Show a warning message with yellow color."""
+        self.message_label.setText(f"Warning: {message}")
+        self.message_label.setStyleSheet("color: #ffd43b; padding: 2px 5px;")
+
+        # Stop any existing timer
+        self._message_timer.stop()
+
+        # Start new timer if duration > 0
+        if duration > 0:
+            self._message_timer.start(duration)
+
+    def set_permanent_message(self, message: str):
+        """Set a permanent message (usually for status info)."""
+        self.permanent_label.setText(message)
+
+    def set_ready_message(self):
+        """Set the default ready message."""
+        self.message_label.setText("")  # Blank instead of "Ready"
+        self.message_label.setStyleSheet("color: #888; padding: 2px 5px;")
+        self._message_timer.stop()
+
+    def _clear_temporary_message(self):
+        """Clear temporary message and return to ready state."""
+        self.set_ready_message()
+        self._message_timer.stop()
+
+    def clear_message(self):
+        """Immediately clear any message."""
+        self.set_ready_message()

lazylabel/utils/__init__.py

@@ -1,6 +1,6 @@
-"""Utility modules."""
-
-from .utils import mask_to_pixmap
-from .custom_file_system_model import CustomFileSystemModel
-
+"""Utility modules."""
+
+from .utils import mask_to_pixmap
+from .custom_file_system_model import CustomFileSystemModel
+
 __all__ = ['mask_to_pixmap', 'CustomFileSystemModel']

lazylabel/utils/custom_file_system_model.py

@@ -1,132 +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)
+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/utils/utils.py

@@ -1,12 +1,12 @@
-import numpy as np
-from PyQt6.QtGui import QImage, QPixmap
-
-
-def mask_to_pixmap(mask, color, alpha=150):
-    colored_mask = np.zeros((mask.shape[0], mask.shape[1], 4), dtype=np.uint8)
-    colored_mask[mask, :3] = color
-    colored_mask[mask, 3] = alpha
-    image = QImage(
-        colored_mask.data, mask.shape[1], mask.shape[0], QImage.Format.Format_RGBA8888
-    )
-    return QPixmap.fromImage(image)
+import numpy as np
+from PyQt6.QtGui import QImage, QPixmap
+
+
+def mask_to_pixmap(mask, color, alpha=150):
+    colored_mask = np.zeros((mask.shape[0], mask.shape[1], 4), dtype=np.uint8)
+    colored_mask[mask, :3] = color
+    colored_mask[mask, 3] = alpha
+    image = QImage(
+        colored_mask.data, mask.shape[1], mask.shape[0], QImage.Format.Format_RGBA8888
+    )
+    return QPixmap.fromImage(image)