Filter ModelPicker by supported diffusers

Author: saddam213

OnnxStack.UI/UserControls/ModelPickerControl.xaml

@@ -52,8 +52,9 @@
             <Grid>
                 <ComboBox 
                      DisplayMemberPath="Name" 
-                     ItemsSource="{Binding UISettings.StableDiffusionModelSets}" 
+                     ItemsSource="{Binding ModelCollectionView}" 
                      SelectedItem="{Binding SelectedModel}" 
+                     IsSynchronizedWithCurrentItem="False"
                      IsEnabled="{Binding SelectedModel.IsLoading, FallbackValue=True, TargetNullValue=True, Converter={StaticResource InverseBoolConverter}}" > 
                 </ComboBox>
                 <TextBlock Text="-- Select Model --" Visibility="{Binding SelectedModel, Converter={StaticResource InverseNullVisibilityConverter}}" IsHitTestVisible="False" Margin="3,2" FontStyle="Italic" Opacity=".7" />

OnnxStack.UI/UserControls/ModelPickerControl.xaml.cs

@@ -12,6 +12,7 @@
 using System.Threading.Tasks;
 using System.Windows;
 using System.Windows.Controls;
+using System.Windows.Data;
 
 namespace OnnxStack.UI.UserControls
 {
@@ -22,6 +23,7 @@ public partial class ModelPickerControl : UserControl, INotifyPropertyChanged
     {
         private readonly ILogger<ModelPickerControl> _logger;
         private readonly IStableDiffusionService _stableDiffusionService;
+        private ICollectionView _modelCollectionView;
 
         /// <summary>Initializes a new instance of the <see cref="ModelPickerControl" /> class.</summary>
         public ModelPickerControl()
@@ -46,8 +48,11 @@ public OnnxStackUIConfig UISettings
             set { SetValue(UISettingsProperty, value); }
         }
         public static readonly DependencyProperty UISettingsProperty =
-            DependencyProperty.Register("UISettings", typeof(OnnxStackUIConfig), typeof(ModelPickerControl));
-
+            DependencyProperty.Register("UISettings", typeof(OnnxStackUIConfig), typeof(ModelPickerControl), new PropertyMetadata((d, e) =>
+            {
+                if (d is ModelPickerControl control && e.NewValue is OnnxStackUIConfig settings)
+                    control.InitializeModels();
+            }));
 
         /// <summary>
         /// Gets or sets the supported diffusers.
@@ -58,8 +63,11 @@ public List<DiffuserType> SupportedDiffusers
             set { SetValue(SupportedDiffusersProperty, value); }
         }
         public static readonly DependencyProperty SupportedDiffusersProperty =
-            DependencyProperty.Register("SupportedDiffusers", typeof(List<DiffuserType>), typeof(ModelPickerControl));
-
+            DependencyProperty.Register("SupportedDiffusers", typeof(List<DiffuserType>), typeof(ModelPickerControl), new PropertyMetadata((d, e) =>
+            {
+                if (d is ModelPickerControl control && e.NewValue is List<DiffuserType> diffusers)
+                    control.ModelCollectionView?.Refresh();
+            }));
 
         /// <summary>
         /// Gets or sets the selected model.
@@ -73,6 +81,25 @@ public StableDiffusionModelSetViewModel SelectedModel
             DependencyProperty.Register("SelectedModel", typeof(StableDiffusionModelSetViewModel), typeof(ModelPickerControl));
 
 
+        public ICollectionView ModelCollectionView
+        {
+            get { return _modelCollectionView; }
+            set { _modelCollectionView = value; NotifyPropertyChanged(); }
+        }
+
+
+        private void InitializeModels()
+        {
+            ModelCollectionView = new ListCollectionView(UISettings.StableDiffusionModelSets);
+            ModelCollectionView.Filter = (obj) =>
+            {
+                if (obj is not StableDiffusionModelSetViewModel viewModel)
+                    return false;
+
+                return viewModel.ModelSet.Diffusers.Intersect(SupportedDiffusers).Any();
+            };
+        }
+
 
         /// <summary>
         /// Loads the model.