Auto-apply downloaded POET weights

Update the POET weights dialog flow to react to `weights_downloaded` instead of manual selection. Downloaded weights are now persisted as the last path, the backend is switched to POET, the model path field is populated automatically, and a status message is shown. The bounding-box coordinate controls were also repositioned in the form to sit with the bbox settings block for clearer UI grouping.

Author: C-Achard

dlclivegui/gui/main_window.py

@@ -433,11 +433,9 @@ def _build_stats_layout(self, stats_widget: QWidget) -> QGridLayout:
 
     def _action_manage_poet_weights(self) -> None:
         if self._poet_weights_dialog is None:
-            dialog = PoetWeightsDialog(
-                self,
-                initial_path=(self._get_last_poet_weights_path()),
-            )
-            dialog.weights_selected.connect(self._on_poet_weights_selected)
+            dialog = PoetWeightsDialog(self)
+
+            dialog.weights_downloaded.connect(self._on_poet_weights_downloaded)
             dialog.finished.connect(self._clear_poet_weights_dialog)
 
             self._poet_weights_dialog = dialog
@@ -446,6 +444,19 @@ def _action_manage_poet_weights(self) -> None:
         self._poet_weights_dialog.raise_()
         self._poet_weights_dialog.activateWindow()
 
+    def _on_poet_weights_downloaded(
+        self,
+        path: str,
+    ) -> None:
+        self._set_last_poet_weights_path(path)
+        self._set_backend("poet")
+        self.model_path_edit.setText(path)
+
+        self.statusBar().showMessage(
+            f"POET weights ready: {path}",
+            5000,
+        )
+
     def _on_poet_weights_selected(
         self,
         path: str,
@@ -910,6 +921,33 @@ def _build_viz_group(self) -> QGroupBox:
         )
         form.addRow(bbox_settings)
 
+        bbox_layout = QHBoxLayout()
+        self.bbox_x0_spin = ScrubSpinBox()
+        self.bbox_x0_spin.setRange(0, 7680)
+        self.bbox_x0_spin.setPrefix("x0:")
+        self.bbox_x0_spin.setValue(0)
+        bbox_layout.addWidget(self.bbox_x0_spin)
+
+        self.bbox_y0_spin = ScrubSpinBox()
+        self.bbox_y0_spin.setRange(0, 4320)
+        self.bbox_y0_spin.setPrefix("y0:")
+        self.bbox_y0_spin.setValue(0)
+        bbox_layout.addWidget(self.bbox_y0_spin)
+
+        self.bbox_x1_spin = ScrubSpinBox()
+        self.bbox_x1_spin.setRange(0, 7680)
+        self.bbox_x1_spin.setPrefix("x1:")
+        self.bbox_x1_spin.setValue(100)
+        bbox_layout.addWidget(self.bbox_x1_spin)
+
+        self.bbox_y1_spin = ScrubSpinBox()
+        self.bbox_y1_spin.setRange(0, 4320)
+        self.bbox_y1_spin.setPrefix("y1:")
+        self.bbox_y1_spin.setValue(100)
+        bbox_layout.addWidget(self.bbox_y1_spin)
+
+        form.addRow("Coordinates", bbox_layout)
+
         # Skeleton overlay
         self.show_skeleton_checkbox = QCheckBox("Display skeleton")
         self.show_skeleton_checkbox.setChecked(False)
@@ -950,33 +988,6 @@ def _build_viz_group(self) -> QGroupBox:
             self.skeleton_thickness_spin,
         )
 
-        bbox_layout = QHBoxLayout()
-        self.bbox_x0_spin = ScrubSpinBox()
-        self.bbox_x0_spin.setRange(0, 7680)
-        self.bbox_x0_spin.setPrefix("x0:")
-        self.bbox_x0_spin.setValue(0)
-        bbox_layout.addWidget(self.bbox_x0_spin)
-
-        self.bbox_y0_spin = ScrubSpinBox()
-        self.bbox_y0_spin.setRange(0, 4320)
-        self.bbox_y0_spin.setPrefix("y0:")
-        self.bbox_y0_spin.setValue(0)
-        bbox_layout.addWidget(self.bbox_y0_spin)
-
-        self.bbox_x1_spin = ScrubSpinBox()
-        self.bbox_x1_spin.setRange(0, 7680)
-        self.bbox_x1_spin.setPrefix("x1:")
-        self.bbox_x1_spin.setValue(100)
-        bbox_layout.addWidget(self.bbox_x1_spin)
-
-        self.bbox_y1_spin = ScrubSpinBox()
-        self.bbox_y1_spin.setRange(0, 4320)
-        self.bbox_y1_spin.setPrefix("y1:")
-        self.bbox_y1_spin.setValue(100)
-        bbox_layout.addWidget(self.bbox_y1_spin)
-
-        form.addRow("Coordinates", bbox_layout)
-
         return group
 
     # ------------------------------------------------------------------ signals