From d8a629c8d1d23e0c9c4052103d541d5d97199363 Mon Sep 17 00:00:00 2001
From: Darth Affe <darthaffe@wyrez.org>
Date: Sun, 25 Aug 2024 16:42:24 +0200
Subject: [PATCH] Updated example for 3.0.0

---
 .../ImageCreationUI/ImageCreationUI.csproj    |  11 +-
 Examples/ImageCreationUI/MainWindow.xaml      | 165 +++++++++++-------
 .../ImageCreationUI/MainWindowViewModel.cs    | 146 +++++++++++++---
 3 files changed, 223 insertions(+), 99 deletions(-)

diff --git a/Examples/ImageCreationUI/ImageCreationUI.csproj b/Examples/ImageCreationUI/ImageCreationUI.csproj
index 66e1258..ab86463 100644
--- a/Examples/ImageCreationUI/ImageCreationUI.csproj
+++ b/Examples/ImageCreationUI/ImageCreationUI.csproj
@@ -11,10 +11,13 @@
 
   <ItemGroup>
     <PackageReference Include="HPPH.System.Drawing" Version="1.0.0" />
-    <PackageReference Include="StableDiffusion.NET" Version="2.0.0" />
-    <PackageReference Include="StableDiffusion.NET.Backend.Cpu" Version="2.0.0" />
-    <PackageReference Include="StableDiffusion.NET.Backend.Cuda" Version="2.0.0" />
-    <PackageReference Include="StableDiffusion.NET.Backend.Rocm" Version="2.0.0" />
+    <PackageReference Include="StableDiffusion.NET.Backend.Cpu" Version="3.0.0" />
+    <PackageReference Include="StableDiffusion.NET.Backend.Cuda" Version="3.0.0" />
+    <PackageReference Include="StableDiffusion.NET.Backend.Rocm" Version="3.0.0" />
+  </ItemGroup>
+
+  <ItemGroup>
+    <ProjectReference Include="..\..\StableDiffusion.NET\StableDiffusion.NET.csproj" />
   </ItemGroup>
 
 </Project>
diff --git a/Examples/ImageCreationUI/MainWindow.xaml b/Examples/ImageCreationUI/MainWindow.xaml
index b5b234f..13aa12f 100644
--- a/Examples/ImageCreationUI/MainWindow.xaml
+++ b/Examples/ImageCreationUI/MainWindow.xaml
@@ -8,7 +8,7 @@
         xmlns:sys="clr-namespace:System;assembly=System.Runtime"
         xmlns:sd="clr-namespace:StableDiffusion.NET;assembly=StableDiffusion.NET"
         mc:Ignorable="d"
-        Title="StableDiffusion.NET" Width="1280" Height="800">
+        Title="StableDiffusion.NET" Width="1706" Height="960">
     <Window.DataContext>
         <local:MainWindowViewModel />
     </Window.DataContext>
@@ -42,81 +42,112 @@
             <ColumnDefinition Width="400" />
         </Grid.ColumnDefinitions>
 
-        <StackPanel Margin="4,0" Orientation="Vertical">
-            <TextBlock Margin="8" Text="This is just an example - inputs are not validated!&#10;Make sure everything is correct before pressing 'Create Image'." />
+        <ScrollViewer VerticalScrollBarVisibility="Auto" HorizontalScrollBarVisibility="Disabled">
+            <StackPanel Margin="4,0" Orientation="Vertical">
+                <TextBlock Margin="8" Text="This is just an example - inputs are not validated!&#10;Make sure everything is correct before pressing 'Create Image'." />
 
-            <Separator />
+                <Separator />
 
-            <Label Content="Model-Path" />
-            <DockPanel>
-                <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectModelCommand}" IsEnabled="{Binding IsReady}" />
-                <TextBox Text="{Binding ModelPath}" />
-            </DockPanel>
-
-            <Label Content="Vae-Path (Optional)" />
-            <DockPanel>
-                <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectVaeCommand}" IsEnabled="{Binding IsReady}" />
-                <TextBox Text="{Binding VaePath}" />
-            </DockPanel>
-
-            <Label Content="Schedule" />
-            <ComboBox ItemsSource="{Binding Source={StaticResource ScheduleDataSource}}" SelectedItem="{Binding Schedule}" />
-
-            <Button Margin="0,8"  Content="Load Model" Command="{Binding LoadModelCommand}" IsEnabled="{Binding IsReady}" />
-
-            <Separator />
-
-            <Label Margin="0,8,0,0" Content="Prompt" />
-            <TextBox Height="80" TextWrapping="Wrap" Text="{Binding Prompt}" />
-
-            <Label Content="AntiPrompt" />
-            <TextBox Height="80" TextWrapping="Wrap" Text="{Binding AntiPrompt}" />
-
-            <DockPanel LastChildFill="True">
-                <StackPanel DockPanel.Dock="Left">
-                    <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
-                        <Label Width="50" Content="Width" />
-                        <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Width}" />
-                    </StackPanel>
-
-                    <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
-                        <Label Width="50" Content="Height" />
-                        <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Height}" />
-                    </StackPanel>
-
-                    <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
-                        <Label Width="50" Content="Cfg" />
-                        <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Cfg}" />
-                    </StackPanel>
-
-                    <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
-                        <Label Width="50" Content="Steps" />
-                        <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Steps}" />
-                    </StackPanel>
-
-                    <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
-                        <Label Width="50" Content="Seed" />
-                        <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Seed}" />
-                    </StackPanel>
+                <Label Content="Model-Type" />
+                <StackPanel Orientation="Horizontal" Margin="4,0,0,0">
+                    <RadioButton Content="Stable Diffusion" IsChecked="{Binding IsStableDiffusionSelected}" GroupName="ModelType" />
+                    <RadioButton Margin="16,0,0,0" Content="Flux" IsChecked="{Binding IsFluxSelected}" GroupName="ModelType" />
                 </StackPanel>
 
-                <DockPanel Margin="16,0,0,0" LastChildFill="True">
-                    <Label DockPanel.Dock="Top" Content="Image2Image Source" />
-                    <DockPanel DockPanel.Dock="Top" >
-                        <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectImage2ImageSourceCommand}" IsEnabled="{Binding IsReady}" />
-                        <TextBox Text="{Binding Image2ImageSourcePath}" />
+                <StackPanel Orientation="Vertical" IsEnabled="{Binding IsStableDiffusionSelected}">
+                    <Label Content="Model-Path" />
+                    <DockPanel>
+                        <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectModelCommand}" IsEnabled="{Binding IsReady}" />
+                        <TextBox Text="{Binding ModelPath}" />
+                    </DockPanel>
+                </StackPanel>
+
+                <StackPanel Orientation="Vertical" IsEnabled="{Binding IsFluxSelected}">
+                    <Label Content="DiffusionModel-Path" />
+                    <DockPanel>
+                        <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectDiffusionModelCommand}" IsEnabled="{Binding IsReady}" />
+                        <TextBox Text="{Binding DiffusionModelPath}" />
                     </DockPanel>
 
-                    <Image Source="{Binding Image2ImageSource, Converter={StaticResource ImageToImageSourceConverter}}" />
+                    <Label Content="ClipL-Path" />
+                    <DockPanel>
+                        <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectClipLCommand}" IsEnabled="{Binding IsReady}" />
+                        <TextBox Text="{Binding ClipLPath}" />
+                    </DockPanel>
+
+                    <Label Content="T5xxl-Path" />
+                    <DockPanel>
+                        <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectT5xxlCommand}" IsEnabled="{Binding IsReady}" />
+                        <TextBox Text="{Binding T5xxlPath}" />
+                    </DockPanel>
+                </StackPanel>
+
+                <Label Content="Vae-Path (Optional)" />
+                <DockPanel>
+                    <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectVaeCommand}" IsEnabled="{Binding IsReady}" />
+                    <TextBox Text="{Binding VaePath}" />
                 </DockPanel>
-            </DockPanel>
 
-            <Label Content="Sample-Method" />
-            <ComboBox ItemsSource="{Binding Source={StaticResource SamplerDataSource}}" SelectedItem="{Binding SampleMethod}" />
+                <Label Content="Schedule" />
+                <ComboBox ItemsSource="{Binding Source={StaticResource ScheduleDataSource}}" SelectedItem="{Binding Schedule}" />
 
-            <Button Margin="0,16,0,0" Content="Create Image" Command="{Binding CreateImageCommand}" IsEnabled="{Binding IsReady}" />
-            <Button Margin="0,16,0,0" Content="Save Image" Command="{Binding SaveImageCommand}" IsEnabled="{Binding IsReady}" />
-        </StackPanel>
+                <Button Margin="0,8"  Content="Load Model" Command="{Binding LoadModelCommand}" IsEnabled="{Binding IsReady}" />
+
+                <Separator />
+
+                <Label Margin="0,8,0,0" Content="Prompt" />
+                <TextBox Height="80" TextWrapping="Wrap" Text="{Binding Prompt}" />
+
+                <Label Content="AntiPrompt" />
+                <TextBox Height="60" TextWrapping="Wrap" Text="{Binding AntiPrompt}" />
+
+                <DockPanel LastChildFill="True">
+                    <StackPanel DockPanel.Dock="Left">
+                        <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
+                            <Label Width="50" Content="Width" />
+                            <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Width}" />
+                        </StackPanel>
+
+                        <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
+                            <Label Width="50" Content="Height" />
+                            <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Height}" />
+                        </StackPanel>
+
+                        <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
+                            <Label Width="50" Content="Cfg" />
+                            <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Cfg}" />
+                        </StackPanel>
+
+                        <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
+                            <Label Width="50" Content="Steps" />
+                            <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Steps}" />
+                        </StackPanel>
+
+                        <StackPanel DockPanel.Dock="Top" Orientation="Horizontal" Margin="0,8,0,0">
+                            <Label Width="50" Content="Seed" />
+                            <TextBox HorizontalAlignment="Left" Width="60" Text="{Binding Seed}" />
+                        </StackPanel>
+                    </StackPanel>
+
+                    <DockPanel Margin="16,0,0,0" LastChildFill="True">
+                        <Label DockPanel.Dock="Top" Content="Image2Image Source" />
+                        <DockPanel DockPanel.Dock="Top" >
+                            <Button DockPanel.Dock="Right" Width="24" Margin="2,0,0,0" Content="..." Command="{Binding SelectImage2ImageSourceCommand}" IsEnabled="{Binding IsReady}" />
+                            <TextBox Text="{Binding Image2ImageSourcePath}" />
+                        </DockPanel>
+
+                        <Image Source="{Binding Image2ImageSource, Converter={StaticResource ImageToImageSourceConverter}}" />
+                    </DockPanel>
+                </DockPanel>
+
+                <Label Content="Sample-Method" />
+                <ComboBox ItemsSource="{Binding Source={StaticResource SamplerDataSource}}" SelectedItem="{Binding SampleMethod}" />
+
+                <Button Margin="0,16,0,0" Content="Create Image" Command="{Binding CreateImageCommand}" IsEnabled="{Binding IsReady}" />
+                <Button Margin="0,16,0,0" Content="Save Image" Command="{Binding SaveImageCommand}" IsEnabled="{Binding IsReady}" />
+            </StackPanel>
+
+        </ScrollViewer>
 
         <GridSplitter Grid.Column="1" Margin="2,0" Width="2" Background="DimGray" VerticalAlignment="Stretch" HorizontalAlignment="Center" />
 
diff --git a/Examples/ImageCreationUI/MainWindowViewModel.cs b/Examples/ImageCreationUI/MainWindowViewModel.cs
index be6fae4..c9922c0 100644
--- a/Examples/ImageCreationUI/MainWindowViewModel.cs
+++ b/Examples/ImageCreationUI/MainWindowViewModel.cs
@@ -13,7 +13,21 @@ public class MainWindowViewModel : INotifyPropertyChanged
 {
     #region Properties & Fields
 
-    private StableDiffusionModel? _model;
+    private DiffusionModel? _model;
+
+    private bool _isStableDiffusionSelected = true;
+    public bool IsStableDiffusionSelected
+    {
+        get => _isStableDiffusionSelected;
+        set => SetProperty(ref _isStableDiffusionSelected, value);
+    }
+
+    private bool _isFluxSelected;
+    public bool IsFluxSelected
+    {
+        get => _isFluxSelected;
+        set => SetProperty(ref _isFluxSelected, value);
+    }
 
     private string _modelPath = string.Empty;
     public string ModelPath
@@ -22,6 +36,27 @@ public class MainWindowViewModel : INotifyPropertyChanged
         set => SetProperty(ref _modelPath, value);
     }
 
+    private string _clipLPath;
+    public string ClipLPath
+    {
+        get => _clipLPath;
+        set => SetProperty(ref _clipLPath, value);
+    }
+
+    private string _t5xxlPath;
+    public string T5xxlPath
+    {
+        get => _t5xxlPath;
+        set => SetProperty(ref _t5xxlPath, value);
+    }
+
+    private string _diffusionModelPath;
+    public string DiffusionModelPath
+    {
+        get => _diffusionModelPath;
+        set => SetProperty(ref _diffusionModelPath, value);
+    }
+
     private string _vaePath = string.Empty;
     public string VaePath
     {
@@ -78,8 +113,8 @@ public class MainWindowViewModel : INotifyPropertyChanged
         set => SetProperty(ref _steps, value);
     }
 
-    private int _seed = -1;
-    public int Seed
+    private long _seed = -1;
+    public long Seed
     {
         get => _seed;
         set => SetProperty(ref _seed, value);
@@ -156,6 +191,15 @@ public class MainWindowViewModel : INotifyPropertyChanged
     private ActionCommand? _selectModelCommand;
     public ActionCommand SelectModelCommand => _selectModelCommand ??= new ActionCommand(SelectModel);
 
+    private ActionCommand? _selectDiffusionModelCommand;
+    public ActionCommand SelectDiffusionModelCommand => _selectDiffusionModelCommand ??= new ActionCommand(SelectDiffusionModel);
+
+    private ActionCommand? _selectClipLCommand;
+    public ActionCommand SelectClipLCommand => _selectClipLCommand ??= new ActionCommand(SelectClipL);
+
+    private ActionCommand? _selectT5xxlCommand;
+    public ActionCommand SelectT5xxlCommand => _selectT5xxlCommand ??= new ActionCommand(SelectT5xxl);
+
     private ActionCommand? _selectVaeCommand;
     public ActionCommand SelectVaeCommand => _selectVaeCommand ??= new ActionCommand(SelectVae);
 
@@ -170,8 +214,8 @@ public class MainWindowViewModel : INotifyPropertyChanged
     {
         try
         {
-            StableDiffusionModel.Log += (_, args) => LogLine($"LOG [{args.Level}]: {args.Text}", false);
-            StableDiffusionModel.Progress += (_, args) => LogLine($"PROGRESS {args.Step} / {args.Steps} ({(args.Progress * 100):N2} %) {args.IterationsPerSecond:N2} it/s ({args.Time})");
+            StableDiffusionCpp.Log += (_, args) => LogLine($"LOG [{args.Level}]: {args.Text}", false);
+            StableDiffusionCpp.Progress += (_, args) => LogLine($"PROGRESS {args.Step} / {args.Steps} ({(args.Progress * 100):N2} %) {args.IterationsPerSecond:N2} it/s ({args.Time})");
         }
         catch (Exception ex)
         {
@@ -191,8 +235,38 @@ public class MainWindowViewModel : INotifyPropertyChanged
 
             _model?.Dispose();
 
-            LogLine($"Loading model '{ModelPath}'");
-            _model = await Task.Run(() => new StableDiffusionModel(ModelPath, new ModelParameter { VaePath = VaePath, Schedule = Schedule }));
+            bool restoreDefaultParameters = false;
+
+            if (IsStableDiffusionSelected)
+            {
+                restoreDefaultParameters = _model?.ModelParameter.DiffusionModelType != DiffusionModelType.StableDiffusion;
+
+                LogLine($"Loading stable diffusion-model '{ModelPath}'");
+                _model = await Task.Run(() => ModelBuilder.StableDiffusion(ModelPath).WithMultithreading().WithVae(VaePath).WithSchedule(Schedule).Build());
+            }
+            else if (IsFluxSelected)
+            {
+                restoreDefaultParameters = _model?.ModelParameter.DiffusionModelType != DiffusionModelType.Flux;
+
+                LogLine($"Loading flux-model '{DiffusionModelPath}'");
+                _model = await Task.Run(() => ModelBuilder.Flux(DiffusionModelPath, ClipLPath, T5xxlPath, VaePath).WithMultithreading().WithSchedule(Schedule).Build());
+            }
+            else
+            {
+                LogLine("No model-type selected :(");
+            }
+
+            if (restoreDefaultParameters && (_model != null))
+            {
+                DiffusionParameter parameter = _model.GetDefaultParameter();
+                AntiPrompt = parameter.NegativePrompt;
+                Width = parameter.Width;
+                Height = parameter.Height;
+                Steps = parameter.SampleSteps;
+                Seed = parameter.Seed;
+                SampleMethod = parameter.SampleMethod;
+                Cfg = _model.ModelParameter.DiffusionModelType == DiffusionModelType.Flux ? parameter.Guidance : parameter.CfgScale;
+            }
         }
         catch (Exception ex)
         {
@@ -206,37 +280,32 @@ public class MainWindowViewModel : INotifyPropertyChanged
 
     private async void CreateImage()
     {
+        if (_model == null) return;
+
         try
         {
             IsReady = false;
 
+            DiffusionParameter parameter = _model.GetDefaultParameter()
+                                                 .WithNegativePrompt(AntiPrompt)
+                                                 .WithSize(Width, Height)
+                                                 .WithSteps(Steps)
+                                                 .WithSeed(Seed)
+                                                 .WithSampler(SampleMethod);
+            if (_model.ModelParameter.DiffusionModelType == DiffusionModelType.StableDiffusion)
+                parameter = parameter.WithCfg(Cfg);
+            else if (_model.ModelParameter.DiffusionModelType == DiffusionModelType.Flux)
+                parameter = parameter.WithGuidance(Cfg);
+
             if (Image2ImageSource == null)
             {
                 LogLine("Creating image ...");
-                Image = await Task.Run(() => _model?.TextToImage(Prompt, new StableDiffusionParameter
-                {
-                    NegativePrompt = AntiPrompt,
-                    Width = Width,
-                    Height = Height,
-                    CfgScale = Cfg,
-                    SampleSteps = Steps,
-                    Seed = Seed,
-                    SampleMethod = SampleMethod
-                }));
+                Image = await Task.Run(() => _model?.TextToImage(Prompt, parameter));
             }
             else
             {
                 LogLine("Manipulating image ...");
-                Image = await Task.Run(() => _model?.ImageToImage(Prompt, Image2ImageSource, new StableDiffusionParameter
-                {
-                    NegativePrompt = AntiPrompt,
-                    Width = Width,
-                    Height = Height,
-                    CfgScale = Cfg,
-                    SampleSteps = Steps,
-                    Seed = Seed,
-                    SampleMethod = SampleMethod
-                }));
+                Image = await Task.Run(() => _model?.ImageToImage(Prompt, Image2ImageSource, parameter));
             }
 
             LogLine("done!");
@@ -277,9 +346,30 @@ public class MainWindowViewModel : INotifyPropertyChanged
             ModelPath = openFileDialog.FileName;
     }
 
+    private void SelectDiffusionModel()
+    {
+        OpenFileDialog openFileDialog = new() { Filter = "Diffusion Model|*.*" };
+        if (openFileDialog.ShowDialog() == true)
+            DiffusionModelPath = openFileDialog.FileName;
+    }
+
+    private void SelectClipL()
+    {
+        OpenFileDialog openFileDialog = new() { Filter = "ClipL|*.*" };
+        if (openFileDialog.ShowDialog() == true)
+            ClipLPath = openFileDialog.FileName;
+    }
+
+    private void SelectT5xxl()
+    {
+        OpenFileDialog openFileDialog = new() { Filter = "T5xxl|*.*" };
+        if (openFileDialog.ShowDialog() == true)
+            T5xxlPath = openFileDialog.FileName;
+    }
+
     private void SelectVae()
     {
-        OpenFileDialog openFileDialog = new() { Filter = "Stable Diffusion VAE|*.*" };
+        OpenFileDialog openFileDialog = new() { Filter = "VAE|*.*" };
         if (openFileDialog.ShowDialog() == true)
             VaePath = openFileDialog.FileName;
     }