From e4602387a66e882b49e97569edca8db64af05e91 Mon Sep 17 00:00:00 2001
From: Ionite <dev@ionite.io>
Date: Mon, 31 Jul 2023 14:27:20 -0400
Subject: [PATCH] Add width and height to SamplerCard

---
 .../Controls/SamplerCard.axaml                | 53 +++++++++++++------
 .../Inference/SamplerCardViewModel.cs         |  2 +
 2 files changed, 40 insertions(+), 15 deletions(-)

diff --git a/StabilityMatrix.Avalonia/Controls/SamplerCard.axaml b/StabilityMatrix.Avalonia/Controls/SamplerCard.axaml
index 6bf2c4d1..82dc4989 100644
--- a/StabilityMatrix.Avalonia/Controls/SamplerCard.axaml
+++ b/StabilityMatrix.Avalonia/Controls/SamplerCard.axaml
@@ -60,21 +60,44 @@
                                 SpinButtonPlacementMode="Inline"/>
                         </Grid>
                         
-                        <!--<Grid ColumnDefinitions="Auto,*,Auto,*">
-                            <TextBlock
-                                VerticalAlignment="Center"
-                                Text="Steps" />
-                            <ui:NumberBox
-                                Grid.Column="1"
-                                SpinButtonPlacementMode="Compact"/>
-                            <TextBlock
-                                Grid.Column="2"
-                                VerticalAlignment="Center"
-                                Text="CFG Scale" />
-                            <ui:NumberBox
-                                Grid.Column="3"
-                                SpinButtonPlacementMode="Compact"/>
-                        </Grid>-->
+                        <StackPanel>
+                            <Grid ColumnDefinitions="*,*,*" RowDefinitions="Auto,*">
+                                <TextBlock
+                                    Text="Width" />
+                                <Slider 
+                                    Grid.Row="1" 
+                                    Grid.Column="0" 
+                                    Grid.ColumnSpan="2"
+                                    Minimum="0"
+                                    Maximum="2048"
+                                    Value="{Binding Width}"/>
+                                <ui:NumberBox
+                                    Grid.Row="1"
+                                    Grid.Column="2"
+                                    Value="{Binding Width}"
+                                    HorizontalAlignment="Right"
+                                    VerticalAlignment="Center"
+                                    SpinButtonPlacementMode="Compact"/>
+                            </Grid>
+                            <Grid ColumnDefinitions="*,*,*" RowDefinitions="Auto,*">
+                                <TextBlock
+                                    Text="Height" />
+                                <Slider 
+                                    Grid.Row="1" 
+                                    Grid.Column="0" 
+                                    Grid.ColumnSpan="2"
+                                    Minimum="0"
+                                    Maximum="2048"
+                                    Value="{Binding Height}"/>
+                                <ui:NumberBox
+                                    Grid.Row="1"
+                                    Grid.Column="2"
+                                    Value="{Binding Height}"
+                                    HorizontalAlignment="Right"
+                                    VerticalAlignment="Center"
+                                    SpinButtonPlacementMode="Compact"/>
+                            </Grid>
+                        </StackPanel>
                     </StackPanel>
                 </controls:Card>
             </ControlTemplate>
diff --git a/StabilityMatrix.Avalonia/ViewModels/Inference/SamplerCardViewModel.cs b/StabilityMatrix.Avalonia/ViewModels/Inference/SamplerCardViewModel.cs
index 1f57e804..caedc169 100644
--- a/StabilityMatrix.Avalonia/ViewModels/Inference/SamplerCardViewModel.cs
+++ b/StabilityMatrix.Avalonia/ViewModels/Inference/SamplerCardViewModel.cs
@@ -10,6 +10,8 @@ public partial class SamplerCardViewModel : ViewModelBase
 {
     [ObservableProperty] private int steps;
     [ObservableProperty] private double cfgScale;
+    [ObservableProperty] private int width;
+    [ObservableProperty] private int height;
     
     [ObservableProperty] private string? selectedSampler;
     [ObservableProperty] private IReadOnlyList<string> samplers = new List<string>();