Polish inference workflow selector UX

Author: mohnjiles

CHANGELOG.md

@@ -12,6 +12,7 @@ and this project adheres to [Semantic Versioning 2.0](https://semver.org/spec/v2
 - Added an Inference workflow profile selector with recommended defaults for default checkpoints, Flux, Flux.2, Z-Image Base/Turbo, and Anima
 ### Changed
 - The CivitAI base model type filter now uses CivitAI's official `/api/v1/enums` endpoint, with fallbacks to the previous technique and a built-in list, so the filter stays populated even if the CivitAI response format changes or the service is unreachable
+- Improved the Inference workflow selector UX with Auto detection feedback, workflow-aware model lists, and recommended-default tooltips
 ### Fixed
 - Fixed CivitAI model browsing breaking during Discovery API outages — the browser now falls back to the direct CivitAI API when Discovery returns a server error, authentication failure, or times out
 - Fixed UNet-only model selection sometimes clearing during model-list refreshes, which could hide text encoder slots after generating, cancelling, or reconnecting to ComfyUI

StabilityMatrix.Avalonia/Controls/Inference/ModelCard.axaml

@@ -56,7 +56,7 @@
                                 Grid.Column="1"
                                 Padding="8,6,4,6"
                                 HorizontalAlignment="Stretch"
-                                ItemsSource="{Binding ClientManager.AllModels}"
+                                ItemsSource="{Binding WorkflowFilteredModels}"
                                 SelectedItem="{Binding SelectedUnifiedModel}"
                                 Theme="{StaticResource BetterComboBoxHybridModelTheme}" />
 
@@ -144,10 +144,16 @@
                                 VerticalAlignment="Stretch"
                                 Command="{Binding ApplyRecommendedDefaultsCommand}"
                                 IsEnabled="{Binding HasRecommendedDefaults}"
-                                ToolTip.Tip="Apply recommended sampler defaults">
+                                ToolTip.Tip="{Binding RecommendedDefaultsToolTip}">
                                 <fluent:SymbolIcon FontSize="16" Symbol="Sparkle" />
                             </Button>
                         </Grid>
+                        <TextBlock
+                            Margin="90,4,0,0"
+                            FontSize="11"
+                            Foreground="{DynamicResource TextFillColorSecondaryBrush}"
+                            IsVisible="{Binding ShowWorkflowProfileStatus}"
+                            Text="{Binding WorkflowProfileStatusText}" />
 
                         <!--  Refiner (below Model, when enabled)  -->
                         <Grid

StabilityMatrix.Avalonia/ViewModels/Dialogs/ModelPickerDialogViewModel.cs

@@ -12,6 +12,7 @@
 using FuzzySharp;
 using Injectio.Attributes;
 using StabilityMatrix.Avalonia.Controls;
+using StabilityMatrix.Avalonia.Models.Inference;
 using StabilityMatrix.Avalonia.Services;
 using StabilityMatrix.Avalonia.ViewModels.Base;
 using StabilityMatrix.Avalonia.ViewModels.CheckpointManager;
@@ -67,6 +68,11 @@ public partial class ModelPickerDialogViewModel : ContentDialogViewModelBase
     /// </summary>
     public ModelPickerSource Source { get; set; } = ModelPickerSource.CheckpointAndUnet;
 
+    /// <summary>
+    /// Optional workflow hint from Inference model cards. Used to open near compatible models.
+    /// </summary>
+    public InferenceWorkflowProfile PreferredWorkflowProfile { get; set; } = InferenceWorkflowProfile.Auto;
+
     [ObservableProperty]
     private string title = "Select Model";
 
@@ -215,6 +221,8 @@ public override void OnLoaded()
         if (preShowCheckpoints)
             ShowCheckpointsOnly = true;
 
+        ApplyPreferredWorkflowProfileFilters();
+
         // Populate models in background after dialog appears to reduce opening hitch.
         Dispatcher.UIThread.Post(RefreshAllModels, DispatcherPriority.Background);
 
@@ -434,7 +442,7 @@ private void LoadFilterStateForCurrentSource()
 
     private void SaveFilterStateForCurrentSource()
     {
-        if (isApplyingSavedFilterState)
+        if (isApplyingSavedFilterState || PreferredWorkflowProfile is not InferenceWorkflowProfile.Auto)
             return;
 
         var selectedBaseModels =
@@ -457,6 +465,41 @@ private void SaveFilterStateForCurrentSource()
         });
     }
 
+    private void ApplyPreferredWorkflowProfileFilters()
+    {
+        if (
+            Source is not ModelPickerSource.CheckpointAndUnet
+            || PreferredWorkflowProfile is InferenceWorkflowProfile.Auto or InferenceWorkflowProfile.Custom
+        )
+        {
+            return;
+        }
+
+        if (PreferredWorkflowProfile is InferenceWorkflowProfile.DefaultCheckpoint)
+        {
+            ShowCheckpointsOnly = true;
+            return;
+        }
+
+        ShowUnetsOnly = true;
+        pendingSelectedBaseModels = GetPreferredBaseModels(PreferredWorkflowProfile)
+            .ToHashSet(StringComparer.OrdinalIgnoreCase);
+    }
+
+    private static IEnumerable<string> GetPreferredBaseModels(InferenceWorkflowProfile profile)
+    {
+        return profile switch
+        {
+            InferenceWorkflowProfile.Flux => ["Flux.1"],
+            InferenceWorkflowProfile.Flux2 => ["Flux.2"],
+            InferenceWorkflowProfile.ZImageBase => ["ZImageBase"],
+            InferenceWorkflowProfile.ZImageTurbo => ["ZImageTurbo"],
+            InferenceWorkflowProfile.Anima => ["Anima"],
+            InferenceWorkflowProfile.HiDream => ["HiDream"],
+            _ => [],
+        };
+    }
+
     private void UpdateFilteredModels()
     {
         var models = allModels.AsEnumerable();

StabilityMatrix.Avalonia/ViewModels/Inference/ModelCardViewModel.cs

@@ -1,4 +1,5 @@
 using System.Collections.ObjectModel;
+using System.Collections.Specialized;
 using System.ComponentModel.DataAnnotations;
 using System.Text.Json.Nodes;
 using CommunityToolkit.Mvvm.ComponentModel;
@@ -14,6 +15,7 @@
 using StabilityMatrix.Avalonia.ViewModels.Dialogs;
 using StabilityMatrix.Avalonia.ViewModels.Inference.Modules;
 using StabilityMatrix.Core.Attributes;
+using StabilityMatrix.Core.Extensions;
 using StabilityMatrix.Core.Models;
 using StabilityMatrix.Core.Models.Api.Comfy.Nodes;
 using StabilityMatrix.Core.Models.Api.Comfy.NodeTypes;
@@ -32,6 +34,10 @@ TabContext tabContext
 {
     [ObservableProperty]
     [NotifyPropertyChangedFor(nameof(SelectedUnifiedModel))]
+    [NotifyPropertyChangedFor(nameof(WorkflowFilteredModels))]
+    [NotifyPropertyChangedFor(nameof(WorkflowProfileStatusText))]
+    [NotifyPropertyChangedFor(nameof(ShowWorkflowProfileStatus))]
+    [NotifyPropertyChangedFor(nameof(RecommendedDefaultsToolTip))]
     private HybridModelFile? selectedModel;
 
     [ObservableProperty]
@@ -40,7 +46,11 @@ TabContext tabContext
         nameof(ShowPrecisionSelection),
         nameof(SelectedUnifiedModel),
         nameof(HasActiveAdvancedOptions),
-        nameof(AdvancedOptionsHeader)
+        nameof(AdvancedOptionsHeader),
+        nameof(WorkflowFilteredModels),
+        nameof(WorkflowProfileStatusText),
+        nameof(ShowWorkflowProfileStatus),
+        nameof(RecommendedDefaultsToolTip)
     )]
     private HybridModelFile? selectedUnetModel;
 
@@ -118,7 +128,11 @@ public HybridModelFile? SelectedUnifiedModel
         nameof(ShowPrecisionSelection),
         nameof(ShowEncoderSection),
         nameof(HasActiveAdvancedOptions),
-        nameof(AdvancedOptionsHeader)
+        nameof(AdvancedOptionsHeader),
+        nameof(WorkflowFilteredModels),
+        nameof(WorkflowProfileStatusText),
+        nameof(ShowWorkflowProfileStatus),
+        nameof(RecommendedDefaultsToolTip)
     )]
     private ModelLoader selectedModelLoader;
 
@@ -160,7 +174,11 @@ public HybridModelFile? SelectedUnifiedModel
         nameof(IsHiDreamWorkflow),
         nameof(ShowShift),
         nameof(ShowEncoderTypeSelection),
-        nameof(HasRecommendedDefaults)
+        nameof(HasRecommendedDefaults),
+        nameof(WorkflowFilteredModels),
+        nameof(WorkflowProfileStatusText),
+        nameof(ShowWorkflowProfileStatus),
+        nameof(RecommendedDefaultsToolTip)
     )]
     private string? selectedClipType;
 
@@ -170,7 +188,11 @@ public HybridModelFile? SelectedUnifiedModel
         nameof(IsHiDreamWorkflow),
         nameof(ShowShift),
         nameof(ShowEncoderTypeSelection),
-        nameof(HasRecommendedDefaults)
+        nameof(HasRecommendedDefaults),
+        nameof(WorkflowFilteredModels),
+        nameof(WorkflowProfileStatusText),
+        nameof(ShowWorkflowProfileStatus),
+        nameof(RecommendedDefaultsToolTip)
     )]
     private InferenceWorkflowProfile selectedWorkflowProfile = InferenceWorkflowProfile.Auto;
 
@@ -273,6 +295,29 @@ or InferenceWorkflowProfile.Flux2
                 or InferenceWorkflowProfile.ZImageBase
                 or InferenceWorkflowProfile.ZImageTurbo
                 or InferenceWorkflowProfile.Anima;
+    public bool ShowWorkflowProfileStatus =>
+        SelectedWorkflowProfile is InferenceWorkflowProfile.Auto
+        && SelectedUnifiedModel is not null
+        && ResolvedWorkflowProfile is not InferenceWorkflowProfile.Custom;
+    public string WorkflowProfileStatusText => $"Detected: {ResolvedWorkflowProfile.GetStringValue()}";
+    public string RecommendedDefaultsToolTip =>
+        ResolvedWorkflowProfile switch
+        {
+            InferenceWorkflowProfile.DefaultCheckpoint =>
+                "Apply recommended sampler defaults: Euler Ancestral / Normal / 30 steps / CFG 5",
+            InferenceWorkflowProfile.Flux =>
+                "Apply recommended sampler defaults: Euler / Simple / 20 steps / CFG 3.5",
+            InferenceWorkflowProfile.Flux2 =>
+                "Apply recommended sampler defaults: Euler / Flux2Scheduler / 20 steps / CFG 5",
+            InferenceWorkflowProfile.ZImageBase =>
+                "Apply recommended sampler defaults: Res Multistep / Simple / 30 steps / CFG 4",
+            InferenceWorkflowProfile.ZImageTurbo =>
+                "Apply recommended sampler defaults: Res Multistep / Simple / 8 steps / CFG 1",
+            InferenceWorkflowProfile.Anima =>
+                "Apply recommended sampler defaults: ER SDE / Simple / 30 steps / CFG 4",
+            _ => "No recommended sampler defaults for this workflow",
+        };
+    public IReadOnlyList<HybridModelFile> WorkflowFilteredModels => GetWorkflowFilteredModels();
 
     public event Action<InferenceWorkflowProfile>? RecommendedDefaultsRequested;
 
@@ -286,19 +331,27 @@ protected override void OnInitialLoaded()
         {
             SetDefaultEncoderCount();
         }
+
+        ClientManager.AllModels.CollectionChanged += AllModelsOnCollectionChanged;
     }
 
     public override void OnUnloaded()
     {
         base.OnUnloaded();
         ExtraNetworksStackCardViewModel.CardAdded -= ExtraNetworksStackCardViewModelOnCardAdded;
+        ClientManager.AllModels.CollectionChanged -= AllModelsOnCollectionChanged;
     }
 
     private void ExtraNetworksStackCardViewModelOnCardAdded(object? sender, LoadableViewModelBase e)
     {
         OnSelectedModelChanged(SelectedModel);
     }
 
+    private void AllModelsOnCollectionChanged(object? sender, NotifyCollectionChangedEventArgs e)
+    {
+        OnPropertyChanged(nameof(WorkflowFilteredModels));
+    }
+
     [RelayCommand]
     private static async Task OnConfigClickAsync()
     {
@@ -327,6 +380,7 @@ private async Task OpenModelPickerAsync()
         using var pickerScope = vmFactory.CreateScope();
         var pickerVm = pickerScope.ServiceManager.Get<ModelPickerDialogViewModel>();
         pickerVm.Title = "Select Model";
+        pickerVm.PreferredWorkflowProfile = SelectedWorkflowProfile;
 
         if (await pickerVm.GetDialog().ShowAsync() == ContentDialogResult.Primary)
         {
@@ -596,10 +650,18 @@ public override void LoadStateFromJsonObject(JsonObject state)
 
     private InferenceWorkflowProfile InferWorkflowProfile()
     {
-        if (SelectedModelLoader is not ModelLoader.Unet)
+        return InferWorkflowProfile(
+            SelectedUnifiedModel,
+            SelectedModelLoader is ModelLoader.Unet
+                || SelectedUnifiedModel?.Local?.SharedFolderType is SharedFolderType.DiffusionModels
+        );
+    }
+
+    private static InferenceWorkflowProfile InferWorkflowProfile(HybridModelFile? model, bool isUnetModel)
+    {
+        if (!isUnetModel)
             return InferenceWorkflowProfile.DefaultCheckpoint;
 
-        var model = SelectedUnetModel;
         var baseModel = model?.Local?.ConnectedModelInfo?.BaseModel;
 
         if (!string.IsNullOrWhiteSpace(baseModel))
@@ -636,7 +698,11 @@ private InferenceWorkflowProfile InferWorkflowProfile()
                 : InferenceWorkflowProfile.ZImageBase;
         }
 
-        if (name.Contains("flux2", StringComparison.OrdinalIgnoreCase))
+        if (
+            name.Contains("flux2", StringComparison.OrdinalIgnoreCase)
+            || name.Contains("flux-2", StringComparison.OrdinalIgnoreCase)
+            || name.Contains("flux_2", StringComparison.OrdinalIgnoreCase)
+        )
             return InferenceWorkflowProfile.Flux2;
 
         if (name.Contains("flux", StringComparison.OrdinalIgnoreCase))
@@ -651,6 +717,44 @@ private InferenceWorkflowProfile InferWorkflowProfile()
         return InferenceWorkflowProfile.DefaultCheckpoint;
     }
 
+    private IReadOnlyList<HybridModelFile> GetWorkflowFilteredModels()
+    {
+        var allModels = ClientManager.AllModels.ToList();
+
+        if (SelectedWorkflowProfile is InferenceWorkflowProfile.Auto or InferenceWorkflowProfile.Custom)
+            return allModels;
+
+        var filteredModels = allModels
+            .Where(model => IsModelCompatibleWithWorkflow(model, SelectedWorkflowProfile))
+            .ToList();
+
+        if (filteredModels.Count == 0)
+            return allModels;
+
+        if (
+            SelectedUnifiedModel is { } selected
+            && filteredModels.All(model => !HybridModelFile.Comparer.Equals(model, selected))
+        )
+        {
+            filteredModels.Insert(0, selected);
+        }
+
+        return filteredModels;
+    }
+
+    private static bool IsModelCompatibleWithWorkflow(HybridModelFile model, InferenceWorkflowProfile profile)
+    {
+        var isUnetModel = model.Local?.SharedFolderType is SharedFolderType.DiffusionModels;
+
+        if (profile is InferenceWorkflowProfile.DefaultCheckpoint)
+            return !isUnetModel;
+
+        if (!isUnetModel)
+            return false;
+
+        return InferWorkflowProfile(model, true) == profile;
+    }
+
     /// <summary>
     /// Loads text encoders from the saved model state, supporting both new and legacy formats.
     /// </summary>
@@ -870,6 +974,10 @@ private void RefreshWorkflowProfileState()
         OnPropertyChanged(nameof(ShowShift));
         OnPropertyChanged(nameof(ShowEncoderTypeSelection));
         OnPropertyChanged(nameof(HasRecommendedDefaults));
+        OnPropertyChanged(nameof(WorkflowFilteredModels));
+        OnPropertyChanged(nameof(WorkflowProfileStatusText));
+        OnPropertyChanged(nameof(ShowWorkflowProfileStatus));
+        OnPropertyChanged(nameof(RecommendedDefaultsToolTip));
 
         if (!isLoadingState)
         {