Ionite
2 years ago
committed by
GitHub
15 changed files with 733 additions and 34 deletions
@ -0,0 +1,20 @@ |
|||||||
|
name: Remove old artifacts |
||||||
|
|
||||||
|
on: |
||||||
|
workflow_dispatch: |
||||||
|
schedule: |
||||||
|
# Every day at 1am |
||||||
|
- cron: '0 1 * * *' |
||||||
|
|
||||||
|
jobs: |
||||||
|
remove-old-artifacts: |
||||||
|
runs-on: ubuntu-latest |
||||||
|
timeout-minutes: 10 |
||||||
|
|
||||||
|
steps: |
||||||
|
- name: Remove old artifacts |
||||||
|
uses: c-hive/gha-remove-artifacts@v1 |
||||||
|
with: |
||||||
|
age: '1 day' |
||||||
|
skip-tags: true |
||||||
|
skip-recent: 1 |
||||||
@ -0,0 +1,18 @@ |
|||||||
|
using System.Threading.Tasks; |
||||||
|
using Refit; |
||||||
|
using StabilityMatrix.Models.Api; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Api; |
||||||
|
|
||||||
|
[Headers("User-Agent: StabilityMatrix")] |
||||||
|
public interface IA3WebApi |
||||||
|
{ |
||||||
|
[Get("/internal/ping")] |
||||||
|
Task<string> GetPing(); |
||||||
|
|
||||||
|
[Post("/sdapi/v1/txt2img")] |
||||||
|
Task<ImageResponse> TextToImage([Body] TextToImageRequest request); |
||||||
|
|
||||||
|
[Get("/sdapi/v1/progress")] |
||||||
|
Task<ProgressResponse> GetProgress([Body] ProgressRequest request); |
||||||
|
} |
||||||
@ -0,0 +1,50 @@ |
|||||||
|
using System; |
||||||
|
using System.Diagnostics; |
||||||
|
using System.Threading.Tasks; |
||||||
|
using System.Windows.Threading; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Helper; |
||||||
|
|
||||||
|
public class AsyncDispatcherTimer : DispatcherTimer |
||||||
|
{ |
||||||
|
public AsyncDispatcherTimer() |
||||||
|
{ |
||||||
|
Tick += AsyncDispatcherTimer_Tick; |
||||||
|
} |
||||||
|
|
||||||
|
private async void AsyncDispatcherTimer_Tick(object? sender, EventArgs e) |
||||||
|
{ |
||||||
|
if (TickTask == null) |
||||||
|
{ |
||||||
|
// no task to run |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
if (IsRunning && !IsReentrant) |
||||||
|
{ |
||||||
|
// previous task hasn't completed |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
try |
||||||
|
{ |
||||||
|
IsRunning = true; |
||||||
|
await TickTask.Invoke(); |
||||||
|
} |
||||||
|
catch (Exception) |
||||||
|
{ |
||||||
|
Debug.WriteLine("Task Failed"); |
||||||
|
throw; |
||||||
|
} |
||||||
|
finally |
||||||
|
{ |
||||||
|
// allow it to run again |
||||||
|
IsRunning = false; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
public bool IsReentrant { get; set; } |
||||||
|
public bool IsRunning { get; private set; } |
||||||
|
|
||||||
|
public Func<Task>? TickTask { get; set; } |
||||||
|
} |
||||||
@ -0,0 +1,14 @@ |
|||||||
|
using System.Collections.Generic; |
||||||
|
using System.Text.Json.Serialization; |
||||||
|
using StabilityMatrix.ViewModels; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Models.Api; |
||||||
|
|
||||||
|
public class ImageResponse |
||||||
|
{ |
||||||
|
[JsonPropertyName("images")] |
||||||
|
public string[] Images { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("info")] |
||||||
|
public string? Info { get; set; } |
||||||
|
} |
||||||
@ -0,0 +1,9 @@ |
|||||||
|
using System.Text.Json.Serialization; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Models.Api; |
||||||
|
|
||||||
|
public class ProgressRequest |
||||||
|
{ |
||||||
|
[JsonPropertyName("skip_current_image")] |
||||||
|
public bool? SkipCurrentImage { get; set; } |
||||||
|
} |
||||||
@ -0,0 +1,24 @@ |
|||||||
|
using System.Text.Json.Serialization; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Models.Api; |
||||||
|
|
||||||
|
public class ProgressResponse |
||||||
|
{ |
||||||
|
// Range from 0 to 1 |
||||||
|
[JsonPropertyName("progress")] |
||||||
|
public float Progress { get; set; } |
||||||
|
|
||||||
|
// ETA in seconds |
||||||
|
[JsonPropertyName("eta_relative")] |
||||||
|
public float EtaRelative { get; set; } |
||||||
|
|
||||||
|
// state: dict |
||||||
|
|
||||||
|
// The current image in base64 format. opts.show_progress_every_n_steps is required for this to work |
||||||
|
[JsonPropertyName("current_image")] |
||||||
|
public string? CurrentImage { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("textinfo")] |
||||||
|
public string? TextInfo { get; set; } |
||||||
|
|
||||||
|
} |
||||||
@ -0,0 +1,134 @@ |
|||||||
|
using System.Collections.Generic; |
||||||
|
using System.Text.Json.Serialization; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Models.Api; |
||||||
|
|
||||||
|
|
||||||
|
public class TextToImageRequest |
||||||
|
{ |
||||||
|
[JsonPropertyName("enable_hr")] |
||||||
|
public bool? EnableHr { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("denoising_strength")] |
||||||
|
public int? DenoisingStrength { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("firstphase_width")] |
||||||
|
public int? FirstPhaseWidth { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("firstphase_height")] |
||||||
|
public int? FirstPhaseHeight { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("hr_scale")] |
||||||
|
public int? HrScale { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("hr_upscaler")] |
||||||
|
public string? HrUpscaler { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("hr_second_pass_steps")] |
||||||
|
public int? HrSecondPassSteps { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("hr_resize_x")] |
||||||
|
public int? HrResizeX { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("hr_resize_y")] |
||||||
|
public int? HrResizeY { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("prompt")] |
||||||
|
public string Prompt { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("styles")] |
||||||
|
public string?[] Styles { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("seed")] |
||||||
|
public int? Seed { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("subseed")] |
||||||
|
public int? Subseed { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("subseed_strength")] |
||||||
|
public int? SubseedStrength { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("seed_resize_from_h")] |
||||||
|
public int? SeedResizeFromH { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("seed_resize_from_w")] |
||||||
|
public int? SeedResizeFromW { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("sampler_name")] |
||||||
|
public string? SamplerName { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("batch_size")] |
||||||
|
public int? BatchSize { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("n_iter")] |
||||||
|
public int? NIter { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("steps")] |
||||||
|
public int? Steps { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("cfg_scale")] |
||||||
|
public int? CfgScale { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("width")] |
||||||
|
public int? Width { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("height")] |
||||||
|
public int? Height { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("restore_faces")] |
||||||
|
public bool? RestoreFaces { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("tiling")] |
||||||
|
public bool? Tiling { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("do_not_save_samples")] |
||||||
|
public bool? DoNotSaveSamples { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("do_not_save_grid")] |
||||||
|
public bool? DoNotSaveGrid { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("negative_prompt")] |
||||||
|
public string? NegativePrompt { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("eta")] |
||||||
|
public int? Eta { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("s_min_uncond")] |
||||||
|
public int? SMinUncond { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("s_churn")] |
||||||
|
public int? SChurn { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("s_tmax")] |
||||||
|
public int? STmax { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("s_tmin")] |
||||||
|
public int? STmin { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("s_noise")] |
||||||
|
public int? SNoise { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("override_settings")] |
||||||
|
public Dictionary<string, string>? OverrideSettings { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("override_settings_restore_afterwards")] |
||||||
|
public bool? OverrideSettingsRestoreAfterwards { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("script_args")] |
||||||
|
public string[]? ScriptArgs { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("sampler_index")] |
||||||
|
public string? SamplerIndex { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("script_name")] |
||||||
|
public string? ScriptName { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("send_images")] |
||||||
|
public bool? SendImages { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("save_images")] |
||||||
|
public bool? SaveImages { get; set; } |
||||||
|
|
||||||
|
[JsonPropertyName("alwayson_scripts")] |
||||||
|
public Dictionary<string, string>? AlwaysOnScripts { get; set; } |
||||||
|
} |
||||||
@ -0,0 +1,157 @@ |
|||||||
|
<Page |
||||||
|
Background="{DynamicResource ApplicationBackgroundBrush}" |
||||||
|
Foreground="{DynamicResource TextFillColorPrimaryBrush}" |
||||||
|
Title="TextToImagePage" |
||||||
|
d:DataContext="{d:DesignInstance Type=viewModels:TextToImageViewModel, |
||||||
|
IsDesignTimeCreatable=True}" |
||||||
|
d:DesignHeight="700" |
||||||
|
d:DesignWidth="1100" |
||||||
|
mc:Ignorable="d" |
||||||
|
x:Class="StabilityMatrix.TextToImagePage" |
||||||
|
xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation" |
||||||
|
xmlns:d="http://schemas.microsoft.com/expression/blend/2008" |
||||||
|
xmlns:local="clr-namespace:StabilityMatrix" |
||||||
|
xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" |
||||||
|
xmlns:models="clr-namespace:StabilityMatrix.Models" |
||||||
|
xmlns:ui="http://schemas.lepo.co/wpfui/2022/xaml" |
||||||
|
xmlns:viewModels="clr-namespace:StabilityMatrix.ViewModels" |
||||||
|
xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml"> |
||||||
|
|
||||||
|
<Grid Margin="20"> |
||||||
|
<Grid.RowDefinitions> |
||||||
|
<RowDefinition Height="*" /> |
||||||
|
<RowDefinition Height="Auto" /> |
||||||
|
</Grid.RowDefinitions> |
||||||
|
|
||||||
|
<!-- Controls --> |
||||||
|
<Grid> |
||||||
|
<Grid.ColumnDefinitions> |
||||||
|
<ColumnDefinition Width="*" /> |
||||||
|
<ColumnDefinition Width="*" /> |
||||||
|
</Grid.ColumnDefinitions> |
||||||
|
|
||||||
|
<ScrollViewer |
||||||
|
HorizontalAlignment="Stretch" |
||||||
|
Margin="8,8,8,16" |
||||||
|
VerticalAlignment="Stretch"> |
||||||
|
<StackPanel> |
||||||
|
<ui:Card HorizontalAlignment="Stretch" Padding="8"> |
||||||
|
<StackPanel Margin="8"> |
||||||
|
<Label |
||||||
|
Content="Prompt" |
||||||
|
FontSize="14" |
||||||
|
Margin="2" /> |
||||||
|
<ui:AutoSuggestBox |
||||||
|
Margin="8" |
||||||
|
Name="PositivePromptBox" |
||||||
|
PlaceholderText="Prompt" |
||||||
|
QuerySubmitted="PositivePromptBox_OnQuerySubmitted" |
||||||
|
SuggestionChosen="PositivePromptBox_OnSuggestionChosen" |
||||||
|
TextChanged="PositivePromptBox_OnTextChanged" /> |
||||||
|
<Label |
||||||
|
Content="Negative Prompt" |
||||||
|
FontSize="14" |
||||||
|
Margin="2" /> |
||||||
|
<ui:AutoSuggestBox |
||||||
|
Margin="8" |
||||||
|
Name="NegativePromptBox" |
||||||
|
PlaceholderText="Negative Prompt" |
||||||
|
QuerySubmitted="NegativePromptBox_OnQuerySubmitted" |
||||||
|
SuggestionChosen="NegativePromptBox_OnSuggestionChosen" |
||||||
|
TextChanged="NegativePromptBox_OnTextChanged" /> |
||||||
|
|
||||||
|
<Grid Margin="8"> |
||||||
|
<Grid.ColumnDefinitions> |
||||||
|
<ColumnDefinition Width="*" /> |
||||||
|
<ColumnDefinition Width="Auto" /> |
||||||
|
</Grid.ColumnDefinitions> |
||||||
|
<Label |
||||||
|
Content="Steps" |
||||||
|
FontSize="14" |
||||||
|
Margin="2" |
||||||
|
VerticalAlignment="Center" /> |
||||||
|
<TextBox |
||||||
|
FontSize="14" |
||||||
|
Grid.Column="1" |
||||||
|
Margin="8" |
||||||
|
Text="{Binding GenerationSteps}" |
||||||
|
VerticalAlignment="Bottom" |
||||||
|
Width="64" /> |
||||||
|
</Grid> |
||||||
|
|
||||||
|
<Slider |
||||||
|
IsSnapToTickEnabled="True" |
||||||
|
Maximum="150" |
||||||
|
Minimum="0" |
||||||
|
TickFrequency="5" |
||||||
|
TickPlacement="BottomRight" |
||||||
|
Value="{Binding GenerationSteps}" /> |
||||||
|
</StackPanel> |
||||||
|
</ui:Card> |
||||||
|
</StackPanel> |
||||||
|
</ScrollViewer> |
||||||
|
|
||||||
|
<!-- Image Viewer --> |
||||||
|
<ui:Card |
||||||
|
Grid.Column="1" |
||||||
|
HorizontalAlignment="Stretch" |
||||||
|
Margin="8,8,8,16" |
||||||
|
Padding="8" |
||||||
|
VerticalAlignment="Top"> |
||||||
|
<Grid> |
||||||
|
<Grid.RowDefinitions> |
||||||
|
<RowDefinition Height="0.2*" /> |
||||||
|
<RowDefinition Height="0.7*" /> |
||||||
|
</Grid.RowDefinitions> |
||||||
|
<StackPanel |
||||||
|
Grid.Row="0" |
||||||
|
Orientation="Vertical" |
||||||
|
VerticalAlignment="Top"> |
||||||
|
<Label |
||||||
|
Content="Image" |
||||||
|
FontSize="14" |
||||||
|
Margin="2" /> |
||||||
|
<ProgressBar |
||||||
|
Maximum="100" |
||||||
|
Value="{Binding ProgressValue, FallbackValue=10}" |
||||||
|
Visibility="{Binding ProgressBarVisibility, FallbackValue=Visible}" |
||||||
|
Width="500" /> |
||||||
|
</StackPanel> |
||||||
|
<Grid |
||||||
|
Background="{DynamicResource ApplicationBackgroundBrush}" |
||||||
|
Grid.Row="1" |
||||||
|
MinHeight="512" |
||||||
|
VerticalAlignment="Stretch"> |
||||||
|
<ui:ProgressRing |
||||||
|
Background="DarkRed" |
||||||
|
Grid.Row="0" |
||||||
|
Height="72" |
||||||
|
HorizontalAlignment="Center" |
||||||
|
IsIndeterminate="True" |
||||||
|
Margin="8" |
||||||
|
VerticalAlignment="Center" |
||||||
|
Visibility="{Binding ProgressRingVisibility, FallbackValue=Collapsed}" |
||||||
|
Width="72" |
||||||
|
x:Name="ImageProgressRing" /> |
||||||
|
<Image |
||||||
|
HorizontalAlignment="Stretch" |
||||||
|
Margin="8" |
||||||
|
Source="{Binding ImagePreview}" |
||||||
|
Stretch="Uniform" |
||||||
|
VerticalAlignment="Stretch" |
||||||
|
Visibility="{Binding ImagePreviewVisibility, FallbackValue=Visible}" /> |
||||||
|
</Grid> |
||||||
|
</Grid> |
||||||
|
</ui:Card> |
||||||
|
</Grid> |
||||||
|
|
||||||
|
<ui:Button |
||||||
|
Appearance="Primary" |
||||||
|
Command="{Binding TextToImageGenerateCommand}" |
||||||
|
Content="Generate Image" |
||||||
|
Grid.Row="1" |
||||||
|
HorizontalAlignment="Center" |
||||||
|
Name="TextToImageButton" /> |
||||||
|
</Grid> |
||||||
|
|
||||||
|
</Page> |
||||||
@ -0,0 +1,52 @@ |
|||||||
|
using System.Windows.Controls; |
||||||
|
using StabilityMatrix.ViewModels; |
||||||
|
using Wpf.Ui.Controls.AutoSuggestBoxControl; |
||||||
|
|
||||||
|
namespace StabilityMatrix; |
||||||
|
|
||||||
|
public sealed partial class TextToImagePage : Page |
||||||
|
{ |
||||||
|
private TextToImageViewModel ViewModel => (TextToImageViewModel) DataContext; |
||||||
|
|
||||||
|
public TextToImagePage(TextToImageViewModel viewModel) |
||||||
|
{ |
||||||
|
InitializeComponent(); |
||||||
|
DataContext = viewModel; |
||||||
|
} |
||||||
|
|
||||||
|
private void PositivePromptBox_OnQuerySubmitted(AutoSuggestBox sender, AutoSuggestBoxQuerySubmittedEventArgs args) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
private void PositivePromptBox_OnSuggestionChosen(AutoSuggestBox sender, AutoSuggestBoxSuggestionChosenEventArgs args) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
private void PositivePromptBox_OnTextChanged(AutoSuggestBox sender, AutoSuggestBoxTextChangedEventArgs args) |
||||||
|
{ |
||||||
|
// Update the prompt text when the user types |
||||||
|
if (args.Reason == AutoSuggestionBoxTextChangeReason.UserInput) |
||||||
|
{ |
||||||
|
var fullText = sender.Text; |
||||||
|
ViewModel.PositivePromptText = fullText; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private void NegativePromptBox_OnQuerySubmitted(AutoSuggestBox sender, AutoSuggestBoxQuerySubmittedEventArgs args) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
private void NegativePromptBox_OnSuggestionChosen(AutoSuggestBox sender, AutoSuggestBoxSuggestionChosenEventArgs args) |
||||||
|
{ |
||||||
|
} |
||||||
|
|
||||||
|
private void NegativePromptBox_OnTextChanged(AutoSuggestBox sender, AutoSuggestBoxTextChangedEventArgs args) |
||||||
|
{ |
||||||
|
// Update the prompt text when the user types |
||||||
|
if (args.Reason == AutoSuggestionBoxTextChangeReason.UserInput) |
||||||
|
{ |
||||||
|
var fullText = sender.Text; |
||||||
|
ViewModel.NegativePromptText = fullText; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,148 @@ |
|||||||
|
using System; |
||||||
|
using System.IO; |
||||||
|
using System.Threading.Tasks; |
||||||
|
using System.Windows; |
||||||
|
using System.Windows.Media.Imaging; |
||||||
|
using CommunityToolkit.Mvvm.ComponentModel; |
||||||
|
using CommunityToolkit.Mvvm.Input; |
||||||
|
using Microsoft.Extensions.Logging; |
||||||
|
using NLog; |
||||||
|
using StabilityMatrix.Api; |
||||||
|
using StabilityMatrix.Helper; |
||||||
|
using StabilityMatrix.Models.Api; |
||||||
|
using ILogger = NLog.ILogger; |
||||||
|
|
||||||
|
namespace StabilityMatrix.ViewModels; |
||||||
|
|
||||||
|
public partial class TextToImageViewModel : ObservableObject |
||||||
|
{ |
||||||
|
private readonly ILogger<TextToImageViewModel> logger; |
||||||
|
private readonly IA3WebApi a3WebApi; |
||||||
|
private AsyncDispatcherTimer? progressQueryTimer; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private bool isGenerating; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
[NotifyPropertyChangedFor(nameof(ProgressRingVisibility))] |
||||||
|
[NotifyPropertyChangedFor(nameof(ImagePreviewVisibility))] |
||||||
|
private bool isProgressRingActive; |
||||||
|
|
||||||
|
public Visibility ProgressRingVisibility => IsProgressRingActive ? Visibility.Visible : Visibility.Collapsed; |
||||||
|
|
||||||
|
public Visibility ImagePreviewVisibility => IsProgressRingActive ? Visibility.Collapsed : Visibility.Visible; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
[NotifyPropertyChangedFor(nameof(ProgressBarVisibility))] |
||||||
|
private int progressValue; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private string positivePromptText; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private string negativePromptText; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private int generationSteps; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private BitmapImage? imagePreview; |
||||||
|
|
||||||
|
public Visibility ProgressBarVisibility => ProgressValue > 0 ? Visibility.Visible : Visibility.Collapsed; |
||||||
|
|
||||||
|
public TextToImageViewModel(IA3WebApi a3WebApi, ILogger<TextToImageViewModel> logger) |
||||||
|
{ |
||||||
|
this.logger = logger; |
||||||
|
this.a3WebApi = a3WebApi; |
||||||
|
positivePromptText = "Positive"; |
||||||
|
negativePromptText = "Negative"; |
||||||
|
generationSteps = 10; |
||||||
|
} |
||||||
|
|
||||||
|
private void StartProgressTracking(TimeSpan? interval = null) |
||||||
|
{ |
||||||
|
progressQueryTimer = new AsyncDispatcherTimer |
||||||
|
{ |
||||||
|
Interval = interval ?? TimeSpan.FromMilliseconds(150), |
||||||
|
IsReentrant = false, |
||||||
|
TickTask = OnProgressTrackingTick, |
||||||
|
}; |
||||||
|
progressQueryTimer.Start(); |
||||||
|
} |
||||||
|
|
||||||
|
private void StopProgressTracking() |
||||||
|
{ |
||||||
|
IsProgressRingActive = false; |
||||||
|
ProgressValue = 0; |
||||||
|
progressQueryTimer?.Stop(); |
||||||
|
} |
||||||
|
|
||||||
|
private async Task OnProgressTrackingTick() |
||||||
|
{ |
||||||
|
var request = new ProgressRequest(); |
||||||
|
var response = await a3WebApi.GetProgress(request); |
||||||
|
var progress = response.Progress; |
||||||
|
logger.LogInformation("Image Progress: {ResponseProgress}, ETA: {ResponseEtaRelative} s", response.Progress, response.EtaRelative); |
||||||
|
if (Math.Abs(progress - 1.0) < 0.01) |
||||||
|
{ |
||||||
|
ProgressValue = 100; |
||||||
|
progressQueryTimer?.Stop(); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
// Update progress |
||||||
|
ProgressValue = (int) Math.Clamp(Math.Ceiling(progress * 100), 0, 100); |
||||||
|
// Update preview image |
||||||
|
var result = response.CurrentImage; |
||||||
|
if (result != null) |
||||||
|
{ |
||||||
|
// Stop indeterminate progress ring |
||||||
|
IsProgressRingActive = false; |
||||||
|
// Set preview image |
||||||
|
var bitmap = Base64ToBitmap(result); |
||||||
|
ImagePreview = bitmap; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private static BitmapImage Base64ToBitmap(string base64String) |
||||||
|
{ |
||||||
|
var imageBytes = Convert.FromBase64String(base64String); |
||||||
|
|
||||||
|
var bitmapImage = new BitmapImage(); |
||||||
|
bitmapImage.BeginInit(); |
||||||
|
|
||||||
|
using var ms = new MemoryStream(imageBytes, 0, imageBytes.Length); |
||||||
|
bitmapImage.StreamSource = ms; |
||||||
|
bitmapImage.CacheOption = BitmapCacheOption.OnLoad; |
||||||
|
bitmapImage.EndInit(); |
||||||
|
return bitmapImage; |
||||||
|
} |
||||||
|
|
||||||
|
[RelayCommand] |
||||||
|
private async void TextToImageGenerate() |
||||||
|
{ |
||||||
|
// Start indeterminate progress ring |
||||||
|
IsProgressRingActive = true; |
||||||
|
|
||||||
|
var request = new TextToImageRequest |
||||||
|
{ |
||||||
|
Prompt = PositivePromptText, |
||||||
|
NegativePrompt = NegativePromptText, |
||||||
|
Steps = GenerationSteps, |
||||||
|
}; |
||||||
|
var task = a3WebApi.TextToImage(request); |
||||||
|
|
||||||
|
// Progress track while waiting for response |
||||||
|
StartProgressTracking(); |
||||||
|
var response = await task; |
||||||
|
StopProgressTracking(); |
||||||
|
|
||||||
|
// Decode base64 image |
||||||
|
var result = response.Images[0]; |
||||||
|
var bitmap = Base64ToBitmap(result); |
||||||
|
|
||||||
|
ImagePreview = bitmap; |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue