Fix issue with unsetting rating of selected images in thumbnail view #…

…237
RupertAvery · Apr 20, 2024 · 3175e78 · 3175e78
1 parent 0772a39
commit 3175e78
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/Diffusion.Toolkit/Controls/ThumbnailView.xaml.cs b/Diffusion.Toolkit/Controls/ThumbnailView.xaml.cs
@@ -399,25 +399,26 @@ private void RateSelected(int rating)
             {
                 var imageEntries = ThumbnailListView.SelectedItems.Cast<ImageEntry>().ToList();
 
+                int? effectiveRating = rating;
+
+                if (imageEntries.Count(i => i.Rating == rating) > imageEntries.Count / 2)
+                {
+                    effectiveRating = null;
+                }
+
                 foreach (var entry in imageEntries)
                 {
+                    entry.Rating = effectiveRating;
 
-                    if (entry.Rating == rating)
-                    {
-                        entry.Rating = null;
-                    }
-                    else
-                    {
-                        entry.Rating = rating;
-                    }
                     if (Model.CurrentImage != null && Model.CurrentImage.Path == entry.Path)
                     {
                         Model.CurrentImage.Rating = entry.Rating;
                     }
                 }
 
                 var ids = imageEntries.Select(x => x.Id).ToList();
-                DataStore.SetRating(ids, rating);
+
+                DataStore.SetRating(ids, effectiveRating);
             }
         }