UX refinements

Author: zindy

src/main/java/qupath/ext/xgboost/XGBoostTrainer.java

@@ -8,6 +8,7 @@
 import qupath.lib.objects.PathObject;
 import qupath.lib.projects.ProjectImageEntry;
 import qupath.lib.objects.PathObjectTools;
+import java.util.Objects;
 
 
 import java.awt.image.BufferedImage;
@@ -105,7 +106,15 @@ public static void train(
             List<PathObject> validDets = new ArrayList<>(detClass.keySet());
 
             if (validDets.isEmpty()) {
-                log.accept("  " + entry.getImageName() + ": 0 training objects found");
+                long annoCount = hierarchy.getAnnotationObjects().stream()
+                        .filter(a -> a.getPathClass() != null && classNames.contains(a.getPathClass().toString()))
+                        .count();
+                long detCount = hierarchy.getDetectionObjects().size();
+                long conflicts = detClass.values().stream().filter(Objects::isNull).count();
+                log.accept("  " + entry.getImageName() + ": 0 training objects found"
+                        + " (matching annotations: " + annoCount
+                        + ", detections: " + detCount
+                        + ", conflicts discarded: " + conflicts + ")");
                 continue;
             }
 
@@ -175,9 +184,10 @@ record FeatureScore(String name, double score, int idx) {}
                       String.format("  %-50s  %.0f", fs.name(), fs.score())));
 
         // ── 6. Select top-N features ────────────────────────────────────────────
-        List<FeatureScore> selected = (topNFeatures > 0 && topNFeatures < nFeatures)
+        List<FeatureScore> selected = (topNFeatures > 0 && topNFeatures < ranked.size())
                 ? ranked.subList(0, topNFeatures)
                 : ranked;
+
         int[] selIdx = selected.stream().mapToInt(FeatureScore::idx).toArray();
         log.accept("\nUsing " + selIdx.length + " features for final model.");
 

src/main/java/qupath/ext/xgboost/ui/InferController.java

@@ -272,7 +272,10 @@ private void runInference() {
                 });
             }
             @Override protected void succeeded() {
-                Platform.runLater(() -> runButton.setDisable(false));
+                Platform.runLater(() -> {
+                    runButton.setDisable(false);
+                    qupath.refreshProject();
+                });
             }
         };
         new Thread(task, "xgboost-infer") {{ setDaemon(true); }}.start();

src/main/java/qupath/ext/xgboost/ui/TrainController.java

@@ -20,6 +20,8 @@
 import qupath.lib.objects.PathObjectTools;
 import qupath.lib.objects.classes.PathClass;
 import qupath.lib.projects.ProjectImageEntry;
+import qupath.lib.gui.dialogs.ProjectDialogs;
+import qupath.fx.dialogs.Dialogs;
 
 import java.awt.image.BufferedImage;
 import java.io.File;
@@ -491,6 +493,25 @@ private void runTraining() {
         float subsample = subsampleSpinner.getValue().floatValue();
         int   topN      = topNSpinner.getValue();
 
+        // Warn if any selected entry is currently open (may have unsaved changes)
+        List<ProjectImageEntry<BufferedImage>> openAndSelected = selectedEntries.stream()
+                .filter(ProjectDialogs.getCurrentImages(qupath)::contains)
+                .collect(Collectors.toList());
+
+        if (!openAndSelected.isEmpty()) {
+            String names = openAndSelected.stream()
+                    .map(ProjectImageEntry::getImageName)
+                    .collect(Collectors.joining("\n  "));
+            var result = Dialogs.builder()
+                    .title("Unsaved changes?")
+                    .content(new Label("The following images are open in a viewer and may have unsaved changes:\n\n  "
+                            + names + "\n\nSave them first (Ctrl+S) to include the latest annotations.\nContinue anyway?"))
+                    .buttons(ButtonType.YES, ButtonType.NO)
+                    .showAndWait()
+                    .orElse(ButtonType.NO);
+            if (result != ButtonType.YES) return;
+        }
+
         logArea.clear();
         trainButton.setDisable(true);