Adding splitter and saving layout state when closing the app

src/gui/tabs/annotation_tab.py

@@ -12,6 +12,7 @@ from PySide6.QtWidgets import (
     QPushButton,
     QFileDialog,
     QMessageBox,
+    QSplitter,
 )
 from PySide6.QtCore import Qt, QSettings
 from pathlib import Path
@@ -43,25 +44,13 @@ class AnnotationTab(QWidget):
         """Setup user interface."""
         layout = QVBoxLayout()
 
-        # Image loading section
-        load_group = QGroupBox("Image Loading")
-        load_layout = QVBoxLayout()
+        # Main horizontal splitter to divide left (image) and right (controls)
+        self.main_splitter = QSplitter(Qt.Horizontal)
+        self.main_splitter.setHandleWidth(10)
 
-        # Load image button
-        button_layout = QHBoxLayout()
-        self.load_image_btn = QPushButton("Load Image")
-        self.load_image_btn.clicked.connect(self._load_image)
-        button_layout.addWidget(self.load_image_btn)
-        button_layout.addStretch()
-
-        load_layout.addLayout(button_layout)
-
-        # Image info label
-        self.image_info_label = QLabel("No image loaded")
-        load_layout.addWidget(self.image_info_label)
-
-        load_group.setLayout(load_layout)
-        layout.addWidget(load_group)
+        # { Left splitter for image display and zoom info
+        self.left_splitter = QSplitter(Qt.Vertical)
+        self.left_splitter.setHandleWidth(10)
 
         # Image display section
         display_group = QGroupBox("Image Display")
@@ -73,7 +62,21 @@ class AnnotationTab(QWidget):
         display_layout.addWidget(self.image_display_widget)
 
         display_group.setLayout(display_layout)
-        layout.addWidget(display_group)
+        self.left_splitter.addWidget(display_group)
+
+        # Zoom controls info
+        zoom_info = QLabel("Zoom: Mouse wheel or +/- keys to zoom in/out")
+        zoom_info.setStyleSheet("QLabel { color: #888; font-style: italic; }")
+        self.left_splitter.addWidget(zoom_info)
+        # }
+
+        # { Right splitter for annotation tools and controls
+        self.right_splitter = QSplitter(Qt.Vertical)
+        self.right_splitter.setHandleWidth(10)
+
+        # Image loading section
+        load_group = QGroupBox("Image Loading")
+        load_layout = QVBoxLayout()
 
         # Future features info
         info_group = QGroupBox("Annotation Tool (Future Feature)")
@@ -86,18 +89,41 @@ class AnnotationTab(QWidget):
             "- Export annotations to YOLO format\n"
             "- Annotation verification"
         )
+        info_label.setWordWrap(True)
         info_layout.addWidget(info_label)
         info_group.setLayout(info_layout)
 
-        layout.addWidget(info_group)
+        self.right_splitter.addWidget(info_group)
 
-        # Zoom controls info
-        zoom_info = QLabel("Zoom: Mouse wheel or +/- keys to zoom in/out")
-        zoom_info.setStyleSheet("QLabel { color: #888; font-style: italic; }")
-        layout.addWidget(zoom_info)
+        # Load image button
+        button_layout = QHBoxLayout()
+        self.load_image_btn = QPushButton("Load Image")
+        self.load_image_btn.clicked.connect(self._load_image)
+        button_layout.addWidget(self.load_image_btn)
+        button_layout.addStretch()
+        load_layout.addLayout(button_layout)
 
+        # Image info label
+        self.image_info_label = QLabel("No image loaded")
+        load_layout.addWidget(self.image_info_label)
+
+        load_group.setLayout(load_layout)
+        self.right_splitter.addWidget(load_group)
+        # }
+
+        # Add both splitters to the main horizontal splitter
+        self.main_splitter.addWidget(self.left_splitter)
+        self.main_splitter.addWidget(self.right_splitter)
+
+        # Set initial sizes: 75% for left (image), 25% for right (controls)
+        self.main_splitter.setSizes([750, 250])
+
+        layout.addWidget(self.main_splitter)
         self.setLayout(layout)
 
+        # Restore splitter positions from settings
+        self._restore_state()
+
     def _load_image(self):
         """Load and display an image file."""
         # Get last opened directory from QSettings
@@ -171,6 +197,49 @@ class AnnotationTab(QWidget):
         """Handle zoom level changes from the image display widget."""
         self._update_image_info()
 
+    def _restore_state(self):
+        """Restore splitter positions from settings."""
+        settings = QSettings("microscopy_app", "object_detection")
+
+        # Restore main splitter state
+        main_state = settings.value("annotation_tab/main_splitter_state")
+        if main_state:
+            self.main_splitter.restoreState(main_state)
+            logger.debug("Restored main splitter state")
+
+        # Restore left splitter state
+        left_state = settings.value("annotation_tab/left_splitter_state")
+        if left_state:
+            self.left_splitter.restoreState(left_state)
+            logger.debug("Restored left splitter state")
+
+        # Restore right splitter state
+        right_state = settings.value("annotation_tab/right_splitter_state")
+        if right_state:
+            self.right_splitter.restoreState(right_state)
+            logger.debug("Restored right splitter state")
+
+    def save_state(self):
+        """Save splitter positions to settings."""
+        settings = QSettings("microscopy_app", "object_detection")
+
+        # Save main splitter state
+        settings.setValue(
+            "annotation_tab/main_splitter_state", self.main_splitter.saveState()
+        )
+
+        # Save left splitter state
+        settings.setValue(
+            "annotation_tab/left_splitter_state", self.left_splitter.saveState()
+        )
+
+        # Save right splitter state
+        settings.setValue(
+            "annotation_tab/right_splitter_state", self.right_splitter.saveState()
+        )
+
+        logger.debug("Saved annotation tab splitter states")
+
     def refresh(self):
         """Refresh the tab."""
         pass