JT
11 months ago
28 changed files with 1281 additions and 94 deletions
@ -0,0 +1,88 @@ |
|||||||
|
using System; |
||||||
|
using System.Collections.Generic; |
||||||
|
using System.Linq; |
||||||
|
using Avalonia.Threading; |
||||||
|
using CommunityToolkit.Mvvm.ComponentModel; |
||||||
|
using FluentAvalonia.UI.Controls; |
||||||
|
using Microsoft.Extensions.Logging; |
||||||
|
using StabilityMatrix.Avalonia.Animations; |
||||||
|
using StabilityMatrix.Avalonia.Services; |
||||||
|
using StabilityMatrix.Avalonia.ViewModels.Base; |
||||||
|
using StabilityMatrix.Avalonia.Views.Dialogs; |
||||||
|
using StabilityMatrix.Core.Attributes; |
||||||
|
using StabilityMatrix.Core.Helper; |
||||||
|
using StabilityMatrix.Core.Helper.Factory; |
||||||
|
using StabilityMatrix.Core.Models; |
||||||
|
using StabilityMatrix.Core.Models.Packages; |
||||||
|
using StabilityMatrix.Core.Python; |
||||||
|
using StabilityMatrix.Core.Services; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Avalonia.ViewModels.Dialogs; |
||||||
|
|
||||||
|
[View(typeof(NewInstallerDialog))] |
||||||
|
[Transient, ManagedService] |
||||||
|
public partial class NewInstallerDialogViewModel : PageViewModelBase |
||||||
|
{ |
||||||
|
private readonly INavigationService<NewPackageManagerViewModel> packageNavigationService; |
||||||
|
private readonly ISettingsManager settingsManager; |
||||||
|
private readonly INotificationService notificationService; |
||||||
|
private readonly ILogger<PackageInstallDetailViewModel> logger; |
||||||
|
private readonly IPyRunner pyRunner; |
||||||
|
private readonly IPrerequisiteHelper prerequisiteHelper; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private IEnumerable<BasePackage> inferencePackages; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private IEnumerable<BasePackage> trainingPackages; |
||||||
|
|
||||||
|
public NewInstallerDialogViewModel( |
||||||
|
IPackageFactory packageFactory, |
||||||
|
INavigationService<NewPackageManagerViewModel> packageNavigationService, |
||||||
|
ISettingsManager settingsManager, |
||||||
|
INotificationService notificationService, |
||||||
|
ILogger<PackageInstallDetailViewModel> logger, |
||||||
|
IPyRunner pyRunner, |
||||||
|
IPrerequisiteHelper prerequisiteHelper |
||||||
|
) |
||||||
|
{ |
||||||
|
this.packageNavigationService = packageNavigationService; |
||||||
|
this.settingsManager = settingsManager; |
||||||
|
this.notificationService = notificationService; |
||||||
|
this.logger = logger; |
||||||
|
this.pyRunner = pyRunner; |
||||||
|
this.prerequisiteHelper = prerequisiteHelper; |
||||||
|
inferencePackages = packageFactory |
||||||
|
.GetPackagesByType(PackageType.SdInference) |
||||||
|
.OrderBy(p => p.InstallerSortOrder); |
||||||
|
trainingPackages = packageFactory |
||||||
|
.GetPackagesByType(PackageType.SdTraining) |
||||||
|
.OrderBy(p => p.InstallerSortOrder); |
||||||
|
} |
||||||
|
|
||||||
|
public override string Title => "Add Package"; |
||||||
|
public override IconSource IconSource => new SymbolIconSource { Symbol = Symbol.Add }; |
||||||
|
|
||||||
|
public void OnPackageSelected(BaseGitPackage package) |
||||||
|
{ |
||||||
|
if (package is null) |
||||||
|
{ |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
var vm = new PackageInstallDetailViewModel( |
||||||
|
package, |
||||||
|
settingsManager, |
||||||
|
notificationService, |
||||||
|
logger, |
||||||
|
pyRunner, |
||||||
|
prerequisiteHelper, |
||||||
|
packageNavigationService |
||||||
|
); |
||||||
|
|
||||||
|
Dispatcher.UIThread.Post( |
||||||
|
() => packageNavigationService.NavigateTo(vm, BetterSlideNavigationTransition.PageSlideFromRight), |
||||||
|
DispatcherPriority.Send |
||||||
|
); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,13 @@ |
|||||||
|
using CommunityToolkit.Mvvm.ComponentModel; |
||||||
|
using StabilityMatrix.Avalonia.ViewModels.Base; |
||||||
|
using StabilityMatrix.Core.Models.Packages.Extensions; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Avalonia.ViewModels; |
||||||
|
|
||||||
|
public partial class ExtensionViewModel() : ViewModelBase |
||||||
|
{ |
||||||
|
[ObservableProperty] |
||||||
|
private bool isSelected; |
||||||
|
|
||||||
|
public ExtensionBase Extension { get; init; } |
||||||
|
} |
||||||
@ -0,0 +1,66 @@ |
|||||||
|
using System.Collections.Generic; |
||||||
|
using System.Collections.ObjectModel; |
||||||
|
using CommunityToolkit.Mvvm.ComponentModel; |
||||||
|
using DynamicData; |
||||||
|
using FluentAvalonia.UI.Controls; |
||||||
|
using StabilityMatrix.Avalonia.Services; |
||||||
|
using StabilityMatrix.Avalonia.ViewModels.Base; |
||||||
|
using StabilityMatrix.Avalonia.ViewModels.Dialogs; |
||||||
|
using StabilityMatrix.Avalonia.Views; |
||||||
|
using StabilityMatrix.Core.Attributes; |
||||||
|
using Symbol = FluentIcons.Common.Symbol; |
||||||
|
using SymbolIconSource = FluentIcons.FluentAvalonia.SymbolIconSource; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Avalonia.ViewModels; |
||||||
|
|
||||||
|
[View(typeof(NewPackageManagerPage))] |
||||||
|
[Singleton] |
||||||
|
public partial class NewPackageManagerViewModel : PageViewModelBase |
||||||
|
{ |
||||||
|
public override string Title => "Packages"; |
||||||
|
public override IconSource IconSource => new SymbolIconSource { Symbol = Symbol.Box, IsFilled = true }; |
||||||
|
|
||||||
|
public IReadOnlyList<PageViewModelBase> SubPages { get; } |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private ObservableCollection<PageViewModelBase> currentPagePath = []; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private PageViewModelBase? currentPage; |
||||||
|
|
||||||
|
public NewPackageManagerViewModel(ServiceManager<ViewModelBase> vmFactory) |
||||||
|
{ |
||||||
|
SubPages = new PageViewModelBase[] |
||||||
|
{ |
||||||
|
vmFactory.Get<PackageManagerViewModel>(), |
||||||
|
vmFactory.Get<NewInstallerDialogViewModel>(), |
||||||
|
}; |
||||||
|
|
||||||
|
CurrentPagePath.AddRange(SubPages); |
||||||
|
|
||||||
|
CurrentPage = SubPages[0]; |
||||||
|
} |
||||||
|
|
||||||
|
partial void OnCurrentPageChanged(PageViewModelBase? value) |
||||||
|
{ |
||||||
|
if (value is null) |
||||||
|
{ |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
if (value is PackageManagerViewModel) |
||||||
|
{ |
||||||
|
CurrentPagePath.Clear(); |
||||||
|
CurrentPagePath.Add(value); |
||||||
|
} |
||||||
|
else if (value is PackageInstallDetailViewModel) |
||||||
|
{ |
||||||
|
CurrentPagePath.Add(value); |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
CurrentPagePath.Clear(); |
||||||
|
CurrentPagePath.AddRange(new[] { SubPages[0], value }); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,276 @@ |
|||||||
|
using System; |
||||||
|
using System.Collections.Generic; |
||||||
|
using System.Collections.ObjectModel; |
||||||
|
using System.IO; |
||||||
|
using System.Linq; |
||||||
|
using System.Threading.Tasks; |
||||||
|
using AsyncAwaitBestPractices; |
||||||
|
using Avalonia.Controls; |
||||||
|
using Avalonia.Controls.Notifications; |
||||||
|
using CommunityToolkit.Mvvm.ComponentModel; |
||||||
|
using CommunityToolkit.Mvvm.Input; |
||||||
|
using FluentAvalonia.UI.Controls; |
||||||
|
using Microsoft.Extensions.Logging; |
||||||
|
using StabilityMatrix.Avalonia.Extensions; |
||||||
|
using StabilityMatrix.Avalonia.Languages; |
||||||
|
using StabilityMatrix.Avalonia.Services; |
||||||
|
using StabilityMatrix.Avalonia.ViewModels.Base; |
||||||
|
using StabilityMatrix.Avalonia.Views; |
||||||
|
using StabilityMatrix.Core.Attributes; |
||||||
|
using StabilityMatrix.Core.Helper; |
||||||
|
using StabilityMatrix.Core.Models; |
||||||
|
using StabilityMatrix.Core.Models.Database; |
||||||
|
using StabilityMatrix.Core.Models.FileInterfaces; |
||||||
|
using StabilityMatrix.Core.Models.PackageModification; |
||||||
|
using StabilityMatrix.Core.Models.Packages; |
||||||
|
using StabilityMatrix.Core.Python; |
||||||
|
using StabilityMatrix.Core.Services; |
||||||
|
using SymbolIconSource = FluentIcons.FluentAvalonia.SymbolIconSource; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Avalonia.ViewModels; |
||||||
|
|
||||||
|
[View(typeof(PackageInstallDetailView))] |
||||||
|
public partial class PackageInstallDetailViewModel( |
||||||
|
BaseGitPackage package, |
||||||
|
ISettingsManager settingsManager, |
||||||
|
INotificationService notificationService, |
||||||
|
ILogger<PackageInstallDetailViewModel> logger, |
||||||
|
IPyRunner pyRunner, |
||||||
|
IPrerequisiteHelper prerequisiteHelper, |
||||||
|
INavigationService<NewPackageManagerViewModel> packageNavigationService |
||||||
|
) : PageViewModelBase |
||||||
|
{ |
||||||
|
public BaseGitPackage SelectedPackage { get; } = package; |
||||||
|
public override string Title { get; } = package.DisplayName; |
||||||
|
public override IconSource IconSource => new SymbolIconSource(); |
||||||
|
|
||||||
|
public string FullInstallPath => Path.Combine(settingsManager.LibraryDir, "Packages", InstallName); |
||||||
|
public bool ShowReleaseMode => SelectedPackage.ShouldIgnoreReleases == false; |
||||||
|
|
||||||
|
public string ReleaseLabelText => IsReleaseMode ? Resources.Label_Version : Resources.Label_Branch; |
||||||
|
|
||||||
|
public bool ShowTorchVersionOptions => SelectedTorchVersion != TorchVersion.None; |
||||||
|
public bool ShowExtensions => SelectedPackage.AvailableExtensions.Any(); |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
[NotifyPropertyChangedFor(nameof(FullInstallPath))] |
||||||
|
private string installName = package.DisplayName; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private bool showDuplicateWarning; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
[NotifyPropertyChangedFor(nameof(ReleaseLabelText))] |
||||||
|
private bool isReleaseMode; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private IEnumerable<PackageVersion> availableVersions = new List<PackageVersion>(); |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private PackageVersion? selectedVersion; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private SharedFolderMethod selectedSharedFolderMethod; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
[NotifyPropertyChangedFor(nameof(ShowTorchVersionOptions))] |
||||||
|
private TorchVersion selectedTorchVersion; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private ObservableCollection<GitCommit>? availableCommits; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private GitCommit? selectedCommit; |
||||||
|
|
||||||
|
[ObservableProperty] |
||||||
|
private ObservableCollection<ExtensionViewModel> availableExtensions = []; |
||||||
|
|
||||||
|
private PackageVersionOptions? allOptions; |
||||||
|
|
||||||
|
public override async Task OnLoadedAsync() |
||||||
|
{ |
||||||
|
if (Design.IsDesignMode) |
||||||
|
return; |
||||||
|
|
||||||
|
OnInstallNameChanged(InstallName); |
||||||
|
AvailableExtensions = new ObservableCollection<ExtensionViewModel>( |
||||||
|
SelectedPackage.AvailableExtensions.Select(p => new ExtensionViewModel { Extension = p }) |
||||||
|
); |
||||||
|
|
||||||
|
allOptions = await SelectedPackage.GetAllVersionOptions(); |
||||||
|
if (ShowReleaseMode) |
||||||
|
{ |
||||||
|
IsReleaseMode = true; |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
UpdateVersions(); |
||||||
|
await UpdateCommits(SelectedPackage.MainBranch); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
[RelayCommand] |
||||||
|
private async Task Install() |
||||||
|
{ |
||||||
|
if (string.IsNullOrWhiteSpace(InstallName)) |
||||||
|
{ |
||||||
|
notificationService.Show( |
||||||
|
new Notification( |
||||||
|
"Package name is empty", |
||||||
|
"Please enter a name for the package", |
||||||
|
NotificationType.Error |
||||||
|
) |
||||||
|
); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
var setPackageInstallingStep = new SetPackageInstallingStep(settingsManager, InstallName); |
||||||
|
|
||||||
|
var installLocation = Path.Combine(settingsManager.LibraryDir, "Packages", InstallName); |
||||||
|
if (Directory.Exists(installLocation)) |
||||||
|
{ |
||||||
|
var installPath = new DirectoryPath(installLocation); |
||||||
|
await installPath.DeleteVerboseAsync(logger); |
||||||
|
} |
||||||
|
|
||||||
|
var prereqStep = new SetupPrerequisitesStep(prerequisiteHelper, pyRunner); |
||||||
|
|
||||||
|
var downloadOptions = new DownloadPackageVersionOptions(); |
||||||
|
var installedVersion = new InstalledPackageVersion(); |
||||||
|
if (IsReleaseMode) |
||||||
|
{ |
||||||
|
downloadOptions.VersionTag = |
||||||
|
SelectedVersion?.TagName ?? throw new NullReferenceException("Selected version is null"); |
||||||
|
downloadOptions.IsLatest = AvailableVersions?.First().TagName == downloadOptions.VersionTag; |
||||||
|
downloadOptions.IsPrerelease = SelectedVersion.IsPrerelease; |
||||||
|
|
||||||
|
installedVersion.InstalledReleaseVersion = downloadOptions.VersionTag; |
||||||
|
installedVersion.IsPrerelease = SelectedVersion.IsPrerelease; |
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
downloadOptions.CommitHash = |
||||||
|
SelectedCommit?.Sha ?? throw new NullReferenceException("Selected commit is null"); |
||||||
|
downloadOptions.BranchName = |
||||||
|
SelectedVersion?.TagName ?? throw new NullReferenceException("Selected version is null"); |
||||||
|
downloadOptions.IsLatest = AvailableCommits?.First().Sha == SelectedCommit.Sha; |
||||||
|
|
||||||
|
installedVersion.InstalledBranch = |
||||||
|
SelectedVersion?.TagName ?? throw new NullReferenceException("Selected version is null"); |
||||||
|
installedVersion.InstalledCommitSha = downloadOptions.CommitHash; |
||||||
|
} |
||||||
|
|
||||||
|
var downloadStep = new DownloadPackageVersionStep(SelectedPackage, installLocation, downloadOptions); |
||||||
|
var installStep = new InstallPackageStep( |
||||||
|
SelectedPackage, |
||||||
|
SelectedTorchVersion, |
||||||
|
SelectedSharedFolderMethod, |
||||||
|
downloadOptions, |
||||||
|
installLocation |
||||||
|
); |
||||||
|
var installExtensionSteps = AvailableExtensions |
||||||
|
.Where(e => e.IsSelected) |
||||||
|
.Select( |
||||||
|
e => |
||||||
|
new InstallExtensionStep( |
||||||
|
e.Extension, |
||||||
|
Path.Combine(installLocation, SelectedPackage.ExtensionsFolderName) |
||||||
|
) |
||||||
|
) |
||||||
|
.ToList(); |
||||||
|
|
||||||
|
var setupModelFoldersStep = new SetupModelFoldersStep( |
||||||
|
SelectedPackage, |
||||||
|
SelectedSharedFolderMethod, |
||||||
|
installLocation |
||||||
|
); |
||||||
|
|
||||||
|
var package = new InstalledPackage |
||||||
|
{ |
||||||
|
DisplayName = InstallName, |
||||||
|
LibraryPath = Path.Combine("Packages", InstallName), |
||||||
|
Id = Guid.NewGuid(), |
||||||
|
PackageName = SelectedPackage.Name, |
||||||
|
Version = installedVersion, |
||||||
|
LaunchCommand = SelectedPackage.LaunchCommand, |
||||||
|
LastUpdateCheck = DateTimeOffset.Now, |
||||||
|
PreferredTorchVersion = SelectedTorchVersion, |
||||||
|
PreferredSharedFolderMethod = SelectedSharedFolderMethod |
||||||
|
}; |
||||||
|
|
||||||
|
var addInstalledPackageStep = new AddInstalledPackageStep(settingsManager, package); |
||||||
|
|
||||||
|
var steps = new List<IPackageStep> |
||||||
|
{ |
||||||
|
setPackageInstallingStep, |
||||||
|
prereqStep, |
||||||
|
downloadStep, |
||||||
|
installStep, |
||||||
|
}; |
||||||
|
|
||||||
|
if (installExtensionSteps.Count > 0) |
||||||
|
{ |
||||||
|
steps.AddRange(installExtensionSteps); |
||||||
|
} |
||||||
|
|
||||||
|
steps.Add(setupModelFoldersStep); |
||||||
|
steps.Add(addInstalledPackageStep); |
||||||
|
|
||||||
|
var runner = new PackageModificationRunner { ShowDialogOnStart = true }; |
||||||
|
EventManager.Instance.OnPackageInstallProgressAdded(runner); |
||||||
|
await runner.ExecuteSteps(steps.ToList()); |
||||||
|
|
||||||
|
if (!runner.Failed) |
||||||
|
{ |
||||||
|
EventManager.Instance.OnInstalledPackagesChanged(); |
||||||
|
notificationService.Show( |
||||||
|
"Package Install Complete", |
||||||
|
$"{InstallName} installed successfully", |
||||||
|
NotificationType.Success |
||||||
|
); |
||||||
|
packageNavigationService.GoBack(); |
||||||
|
packageNavigationService.GoBack(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private void UpdateVersions() |
||||||
|
{ |
||||||
|
AvailableVersions = |
||||||
|
IsReleaseMode && ShowReleaseMode ? allOptions.AvailableVersions : allOptions.AvailableBranches; |
||||||
|
|
||||||
|
SelectedVersion = !IsReleaseMode |
||||||
|
? AvailableVersions?.FirstOrDefault(x => x.TagName == SelectedPackage.MainBranch) |
||||||
|
?? AvailableVersions?.FirstOrDefault() |
||||||
|
: AvailableVersions?.FirstOrDefault(); |
||||||
|
} |
||||||
|
|
||||||
|
private async Task UpdateCommits(string branchName) |
||||||
|
{ |
||||||
|
var commits = await SelectedPackage.GetAllCommits(branchName); |
||||||
|
if (commits != null) |
||||||
|
{ |
||||||
|
AvailableCommits = new ObservableCollection<GitCommit>(commits); |
||||||
|
SelectedCommit = AvailableCommits.FirstOrDefault(); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
partial void OnInstallNameChanged(string? value) |
||||||
|
{ |
||||||
|
ShowDuplicateWarning = settingsManager.Settings.InstalledPackages.Any( |
||||||
|
p => p.LibraryPath == $"Packages{Path.DirectorySeparatorChar}{value}" |
||||||
|
); |
||||||
|
} |
||||||
|
|
||||||
|
partial void OnIsReleaseModeChanged(bool value) |
||||||
|
{ |
||||||
|
UpdateVersions(); |
||||||
|
} |
||||||
|
|
||||||
|
partial void OnSelectedVersionChanged(PackageVersion? value) |
||||||
|
{ |
||||||
|
if (IsReleaseMode) |
||||||
|
return; |
||||||
|
|
||||||
|
UpdateCommits(value?.TagName ?? SelectedPackage.MainBranch).SafeFireAndForget(); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,233 @@ |
|||||||
|
<controls:UserControlBase xmlns="https://github.com/avaloniaui" |
||||||
|
xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml" |
||||||
|
xmlns:d="http://schemas.microsoft.com/expression/blend/2008" |
||||||
|
xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" |
||||||
|
xmlns:controls="clr-namespace:StabilityMatrix.Avalonia.Controls" |
||||||
|
xmlns:dialogs="clr-namespace:StabilityMatrix.Avalonia.ViewModels.Dialogs" |
||||||
|
xmlns:mocks="clr-namespace:StabilityMatrix.Avalonia.DesignData" |
||||||
|
xmlns:packages="clr-namespace:StabilityMatrix.Core.Models.Packages;assembly=StabilityMatrix.Core" |
||||||
|
xmlns:models="clr-namespace:StabilityMatrix.Core.Models;assembly=StabilityMatrix.Core" |
||||||
|
xmlns:controls1="clr-namespace:FluentAvalonia.UI.Controls;assembly=FluentAvalonia" |
||||||
|
xmlns:input="clr-namespace:FluentAvalonia.UI.Input;assembly=FluentAvalonia" |
||||||
|
mc:Ignorable="d" d:DesignWidth="800" d:DesignHeight="450" |
||||||
|
x:DataType="dialogs:NewInstallerDialogViewModel" |
||||||
|
d:DataContext="{x:Static mocks:DesignData.NewInstallerDialogViewModel}" |
||||||
|
x:Class="StabilityMatrix.Avalonia.Views.Dialogs.NewInstallerDialog"> |
||||||
|
|
||||||
|
<controls:UserControlBase.Resources> |
||||||
|
<input:StandardUICommand x:Key="PackageSelected" |
||||||
|
Command="{Binding OnPackageSelected}"/> |
||||||
|
|
||||||
|
<DataTemplate x:DataType="packages:BaseGitPackage" x:Key="PackageTemplate"> |
||||||
|
<Button Padding="0" |
||||||
|
HorizontalAlignment="Stretch" |
||||||
|
HorizontalContentAlignment="Stretch" |
||||||
|
Margin="4" |
||||||
|
Classes="transparent-full" |
||||||
|
CornerRadius="8" |
||||||
|
Command="{StaticResource PackageSelected}" |
||||||
|
CommandParameter="{Binding }"> |
||||||
|
<controls:Card> |
||||||
|
|
||||||
|
<Grid RowDefinitions="Auto, Auto, Auto, Auto" |
||||||
|
ColumnDefinitions="Auto, *, Auto"> |
||||||
|
<controls:BetterAdvancedImage |
||||||
|
Grid.Row="0" |
||||||
|
Grid.RowSpan="4" |
||||||
|
Grid.Column="0" |
||||||
|
MaxHeight="128" |
||||||
|
MaxWidth="128" |
||||||
|
Stretch="UniformToFill" |
||||||
|
CornerRadius="4" |
||||||
|
Source="{Binding PreviewImageUri}" /> |
||||||
|
<TextBlock Grid.Row="0" |
||||||
|
Grid.Column="1" |
||||||
|
Text="{Binding DisplayName}" |
||||||
|
FontWeight="Light" |
||||||
|
Margin="16,0,0,0" |
||||||
|
FontSize="20" /> |
||||||
|
<TextBlock Grid.Row="1" |
||||||
|
Grid.Column="1" |
||||||
|
Text="{Binding ByAuthor}" |
||||||
|
FontWeight="Light" |
||||||
|
Margin="16,-4,0,0" |
||||||
|
FontSize="13" /> |
||||||
|
|
||||||
|
<TextBlock Grid.Row="0" |
||||||
|
Grid.RowSpan="4" |
||||||
|
Grid.Column="1" |
||||||
|
TextWrapping="Wrap" |
||||||
|
Text="{Binding Blurb}" |
||||||
|
FontWeight="Light" |
||||||
|
Margin="16,0,0,0" |
||||||
|
VerticalAlignment="Center" |
||||||
|
FontSize="16" /> |
||||||
|
|
||||||
|
<ItemsRepeater Grid.Column="1" |
||||||
|
Grid.Row="3" |
||||||
|
Margin="16, 8, 0, 8" |
||||||
|
ItemsSource="{Binding AvailableTorchVersions}"> |
||||||
|
<ItemsRepeater.Layout> |
||||||
|
<StackLayout Orientation="Horizontal" /> |
||||||
|
</ItemsRepeater.Layout> |
||||||
|
<ItemsRepeater.ItemTemplate> |
||||||
|
<DataTemplate DataType="{x:Type models:TorchVersion}"> |
||||||
|
<controls:Card |
||||||
|
Tag="{Binding }" |
||||||
|
HorizontalAlignment="Left" |
||||||
|
Padding="4" |
||||||
|
Margin="0,0,8,0" |
||||||
|
VerticalAlignment="Top"> |
||||||
|
|
||||||
|
<controls:Card.Styles> |
||||||
|
<Style Selector="controls|Card[Tag=Cuda]"> |
||||||
|
<Setter Property="Background" |
||||||
|
Value="{DynamicResource ThemeGreenColorTransparent}" /> |
||||||
|
<Setter Property="BorderBrush" |
||||||
|
Value="{DynamicResource ThemeGreenColorTransparent}" /> |
||||||
|
<Style |
||||||
|
Selector="^ /template/ ContentPresenter#PART_ContentPresenter"> |
||||||
|
<Setter Property="Foreground" |
||||||
|
Value="{DynamicResource ButtonForeground}" /> |
||||||
|
</Style> |
||||||
|
<Setter Property="Content"> |
||||||
|
<Template> |
||||||
|
<TextBlock |
||||||
|
FontWeight="Medium" |
||||||
|
HorizontalAlignment="Center" |
||||||
|
Text="NVIDIA" |
||||||
|
TextAlignment="Center" |
||||||
|
VerticalAlignment="Center" /> |
||||||
|
</Template> |
||||||
|
</Setter> |
||||||
|
</Style> |
||||||
|
<Style Selector="controls|Card[Tag=Rocm]"> |
||||||
|
<Setter Property="Background" |
||||||
|
Value="{DynamicResource ThemeDarkRedColor}" /> |
||||||
|
<Setter Property="BorderBrush" |
||||||
|
Value="{DynamicResource ThemeDarkRedColor}" /> |
||||||
|
<Style |
||||||
|
Selector="^ /template/ ContentPresenter#PART_ContentPresenter"> |
||||||
|
<Setter Property="Foreground" |
||||||
|
Value="{DynamicResource ButtonForeground}" /> |
||||||
|
</Style> |
||||||
|
<Setter Property="Content"> |
||||||
|
<Template> |
||||||
|
<TextBlock |
||||||
|
FontWeight="Medium" |
||||||
|
HorizontalAlignment="Center" |
||||||
|
Text="AMD (Linux)" |
||||||
|
TextAlignment="Center" |
||||||
|
VerticalAlignment="Center" |
||||||
|
ToolTip.Tip="For AMD GPUs that support ROCm on Linux" /> |
||||||
|
</Template> |
||||||
|
</Setter> |
||||||
|
</Style> |
||||||
|
<Style Selector="controls|Card[Tag=DirectMl]"> |
||||||
|
<Setter Property="Background" |
||||||
|
Value="{DynamicResource ThemeDarkBlueColor}" /> |
||||||
|
<Setter Property="BorderBrush" |
||||||
|
Value="{DynamicResource ThemeDarkBlueColor}" /> |
||||||
|
<Setter Property="Content"> |
||||||
|
<Template> |
||||||
|
<TextBlock |
||||||
|
FontWeight="Medium" |
||||||
|
HorizontalAlignment="Center" |
||||||
|
Text="DirectML" |
||||||
|
TextAlignment="Center" |
||||||
|
VerticalAlignment="Center" |
||||||
|
ToolTip.Tip="For any DirectX compatible GPU on Windows" /> |
||||||
|
</Template> |
||||||
|
</Setter> |
||||||
|
</Style> |
||||||
|
<Style Selector="controls|Card[Tag=Mps]"> |
||||||
|
<Setter Property="Background" Value="White" /> |
||||||
|
<Setter Property="BorderBrush" Value="White" /> |
||||||
|
<Style |
||||||
|
Selector="^ /template/ ContentPresenter#PART_ContentPresenter"> |
||||||
|
<Setter Property="Foreground" |
||||||
|
Value="{DynamicResource ButtonForeground}" /> |
||||||
|
</Style> |
||||||
|
<Setter Property="Content"> |
||||||
|
<Template> |
||||||
|
<TextBlock |
||||||
|
FontWeight="Medium" |
||||||
|
HorizontalAlignment="Center" |
||||||
|
Text="macOS" |
||||||
|
TextAlignment="Center" |
||||||
|
Foreground="Black" |
||||||
|
VerticalAlignment="Center" /> |
||||||
|
</Template> |
||||||
|
</Setter> |
||||||
|
</Style> |
||||||
|
<Style Selector="controls|Card[Tag=Cpu]"> |
||||||
|
<Setter Property="Background" |
||||||
|
Value="{DynamicResource ThemeBlueGreyColor}" /> |
||||||
|
<Setter Property="BorderBrush" |
||||||
|
Value="{DynamicResource ThemeBlueGreyColor}" /> |
||||||
|
<Style |
||||||
|
Selector="^ /template/ ContentPresenter#PART_ContentPresenter"> |
||||||
|
<Setter Property="Foreground" |
||||||
|
Value="{DynamicResource ButtonForeground}" /> |
||||||
|
</Style> |
||||||
|
<Setter Property="Content"> |
||||||
|
<Template> |
||||||
|
<TextBlock |
||||||
|
FontWeight="Medium" |
||||||
|
HorizontalAlignment="Center" |
||||||
|
Text="CPU" |
||||||
|
TextAlignment="Center" |
||||||
|
VerticalAlignment="Center" /> |
||||||
|
</Template> |
||||||
|
</Setter> |
||||||
|
</Style> |
||||||
|
</controls:Card.Styles> |
||||||
|
</controls:Card> |
||||||
|
</DataTemplate> |
||||||
|
</ItemsRepeater.ItemTemplate> |
||||||
|
</ItemsRepeater> |
||||||
|
|
||||||
|
<controls1:SymbolIcon Grid.Row="0" |
||||||
|
Grid.RowSpan="4" |
||||||
|
Grid.Column="2" |
||||||
|
HorizontalAlignment="Right" |
||||||
|
VerticalAlignment="Center" |
||||||
|
FontSize="24" |
||||||
|
Symbol="ChevronRight" /> |
||||||
|
</Grid> |
||||||
|
</controls:Card> |
||||||
|
</Button> |
||||||
|
</DataTemplate> |
||||||
|
</controls:UserControlBase.Resources> |
||||||
|
|
||||||
|
<Grid> |
||||||
|
<TabControl TabStripPlacement="Top"> |
||||||
|
<TabControl.Items> |
||||||
|
<TabItem Header="Inference" Margin="8,0,0,0"> |
||||||
|
<ScrollViewer Padding="8"> |
||||||
|
<ItemsRepeater ItemsSource="{Binding InferencePackages}"> |
||||||
|
<ItemsRepeater.Layout> |
||||||
|
<StackLayout Orientation="Vertical" /> |
||||||
|
</ItemsRepeater.Layout> |
||||||
|
<ItemsRepeater.ItemTemplate> |
||||||
|
<StaticResource ResourceKey="PackageTemplate" /> |
||||||
|
</ItemsRepeater.ItemTemplate> |
||||||
|
</ItemsRepeater> |
||||||
|
</ScrollViewer> |
||||||
|
</TabItem> |
||||||
|
<TabItem Header="Training"> |
||||||
|
<ScrollViewer Padding="8"> |
||||||
|
<ItemsRepeater ItemsSource="{Binding TrainingPackages}"> |
||||||
|
<ItemsRepeater.Layout> |
||||||
|
<StackLayout Orientation="Vertical" /> |
||||||
|
</ItemsRepeater.Layout> |
||||||
|
<ItemsRepeater.ItemTemplate> |
||||||
|
<StaticResource ResourceKey="PackageTemplate" /> |
||||||
|
</ItemsRepeater.ItemTemplate> |
||||||
|
</ItemsRepeater> |
||||||
|
</ScrollViewer> |
||||||
|
</TabItem> |
||||||
|
</TabControl.Items> |
||||||
|
</TabControl> |
||||||
|
</Grid> |
||||||
|
</controls:UserControlBase> |
||||||
@ -0,0 +1,13 @@ |
|||||||
|
using StabilityMatrix.Avalonia.Controls; |
||||||
|
using StabilityMatrix.Core.Attributes; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Avalonia.Views.Dialogs; |
||||||
|
|
||||||
|
[Singleton] |
||||||
|
public partial class NewInstallerDialog : UserControlBase |
||||||
|
{ |
||||||
|
public NewInstallerDialog() |
||||||
|
{ |
||||||
|
InitializeComponent(); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,49 @@ |
|||||||
|
<controls:UserControlBase xmlns="https://github.com/avaloniaui" |
||||||
|
xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml" |
||||||
|
xmlns:d="http://schemas.microsoft.com/expression/blend/2008" |
||||||
|
xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" |
||||||
|
xmlns:viewModels="clr-namespace:StabilityMatrix.Avalonia.ViewModels" |
||||||
|
xmlns:controls="clr-namespace:StabilityMatrix.Avalonia.Controls" |
||||||
|
xmlns:designData="clr-namespace:StabilityMatrix.Avalonia.DesignData" |
||||||
|
xmlns:ui="clr-namespace:FluentAvalonia.UI.Controls;assembly=FluentAvalonia.BreadcrumbBar" |
||||||
|
xmlns:controls1="clr-namespace:FluentAvalonia.UI.Controls;assembly=FluentAvalonia" |
||||||
|
xmlns:avalonia="clr-namespace:StabilityMatrix.Avalonia" |
||||||
|
xmlns:base="clr-namespace:StabilityMatrix.Avalonia.ViewModels.Base" |
||||||
|
mc:Ignorable="d" d:DesignWidth="800" d:DesignHeight="450" |
||||||
|
x:DataType="viewModels:NewPackageManagerViewModel" |
||||||
|
x:CompileBindings="True" |
||||||
|
d:DataContext="{x:Static designData:DesignData.NewPackageManagerViewModel}" |
||||||
|
x:Class="StabilityMatrix.Avalonia.Views.NewPackageManagerPage"> |
||||||
|
|
||||||
|
<controls:UserControlBase.Resources> |
||||||
|
<!-- Override styles for BreadcrumbBar --> |
||||||
|
<!-- ReSharper disable Xaml.RedundantResource --> |
||||||
|
<x:Double x:Key="BreadcrumbBarItemThemeFontSize">24</x:Double> |
||||||
|
<x:Double x:Key="BreadcrumbBarChevronFontSize">17</x:Double> |
||||||
|
<Thickness x:Key="BreadcrumbBarChevronPadding">6,3</Thickness> |
||||||
|
<FontWeight x:Key="BreadcrumbBarItemFontWeight">Medium</FontWeight> |
||||||
|
<!-- ReSharper restore Xaml.RedundantResource --> |
||||||
|
</controls:UserControlBase.Resources> |
||||||
|
|
||||||
|
<Grid RowDefinitions="Auto,*"> |
||||||
|
<ui:BreadcrumbBar |
||||||
|
Grid.Row="0" |
||||||
|
Margin="16,8" |
||||||
|
x:Name="BreadcrumbBar" |
||||||
|
ItemsSource="{Binding CurrentPagePath}"> |
||||||
|
<ui:BreadcrumbBar.ItemTemplate> |
||||||
|
<DataTemplate x:DataType="base:PageViewModelBase"> |
||||||
|
<ui:BreadcrumbBarItem Content="{Binding Title}" /> |
||||||
|
</DataTemplate> |
||||||
|
</ui:BreadcrumbBar.ItemTemplate> |
||||||
|
</ui:BreadcrumbBar> |
||||||
|
|
||||||
|
<controls1:Frame |
||||||
|
Grid.Row="1" |
||||||
|
Name="FrameView"> |
||||||
|
<controls1:Frame.NavigationPageFactory> |
||||||
|
<avalonia:ViewLocator/> |
||||||
|
</controls1:Frame.NavigationPageFactory> |
||||||
|
</controls1:Frame> |
||||||
|
</Grid> |
||||||
|
</controls:UserControlBase> |
||||||
@ -0,0 +1,99 @@ |
|||||||
|
using System; |
||||||
|
using System.ComponentModel; |
||||||
|
using System.Linq; |
||||||
|
using Avalonia.Interactivity; |
||||||
|
using Avalonia.Threading; |
||||||
|
using FluentAvalonia.UI.Controls; |
||||||
|
using FluentAvalonia.UI.Media.Animation; |
||||||
|
using FluentAvalonia.UI.Navigation; |
||||||
|
using Microsoft.Extensions.DependencyInjection; |
||||||
|
using StabilityMatrix.Avalonia.Animations; |
||||||
|
using StabilityMatrix.Avalonia.Controls; |
||||||
|
using StabilityMatrix.Avalonia.Models; |
||||||
|
using StabilityMatrix.Avalonia.Services; |
||||||
|
using StabilityMatrix.Avalonia.ViewModels; |
||||||
|
using StabilityMatrix.Avalonia.ViewModels.Base; |
||||||
|
using StabilityMatrix.Core.Attributes; |
||||||
|
using StabilityMatrix.Core.Models; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Avalonia.Views; |
||||||
|
|
||||||
|
[Singleton] |
||||||
|
public partial class NewPackageManagerPage : UserControlBase, IHandleNavigation |
||||||
|
{ |
||||||
|
private readonly INavigationService<NewPackageManagerViewModel> packageNavigationService; |
||||||
|
|
||||||
|
private bool hasLoaded; |
||||||
|
|
||||||
|
private NewPackageManagerViewModel ViewModel => (NewPackageManagerViewModel)DataContext!; |
||||||
|
|
||||||
|
[DesignOnly(true)] |
||||||
|
[Obsolete("For XAML use only", true)] |
||||||
|
public NewPackageManagerPage() |
||||||
|
: this(App.Services.GetRequiredService<INavigationService<NewPackageManagerViewModel>>()) { } |
||||||
|
|
||||||
|
public NewPackageManagerPage(INavigationService<NewPackageManagerViewModel> packageNavigationService) |
||||||
|
{ |
||||||
|
this.packageNavigationService = packageNavigationService; |
||||||
|
|
||||||
|
InitializeComponent(); |
||||||
|
|
||||||
|
packageNavigationService.SetFrame(FrameView); |
||||||
|
packageNavigationService.TypedNavigation += NavigationService_OnTypedNavigation; |
||||||
|
FrameView.Navigated += FrameView_Navigated; |
||||||
|
BreadcrumbBar.ItemClicked += BreadcrumbBar_ItemClicked; |
||||||
|
} |
||||||
|
|
||||||
|
/// <inheritdoc /> |
||||||
|
protected override void OnLoaded(RoutedEventArgs e) |
||||||
|
{ |
||||||
|
base.OnLoaded(e); |
||||||
|
|
||||||
|
if (!hasLoaded) |
||||||
|
{ |
||||||
|
// Initial load, navigate to first page |
||||||
|
Dispatcher.UIThread.Post( |
||||||
|
() => |
||||||
|
packageNavigationService.NavigateTo( |
||||||
|
ViewModel.SubPages[0], |
||||||
|
new SuppressNavigationTransitionInfo() |
||||||
|
) |
||||||
|
); |
||||||
|
|
||||||
|
hasLoaded = true; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
private void NavigationService_OnTypedNavigation(object? sender, TypedNavigationEventArgs e) |
||||||
|
{ |
||||||
|
ViewModel.CurrentPage = |
||||||
|
ViewModel.SubPages.FirstOrDefault(x => x.GetType() == e.ViewModelType) |
||||||
|
?? e.ViewModel as PageViewModelBase; |
||||||
|
} |
||||||
|
|
||||||
|
private void FrameView_Navigated(object? sender, NavigationEventArgs args) |
||||||
|
{ |
||||||
|
if (args.Content is not PageViewModelBase vm) |
||||||
|
{ |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
ViewModel.CurrentPage = vm; |
||||||
|
} |
||||||
|
|
||||||
|
private void BreadcrumbBar_ItemClicked(BreadcrumbBar sender, BreadcrumbBarItemClickedEventArgs args) |
||||||
|
{ |
||||||
|
// Skip if already on same page |
||||||
|
if (args.Item is not PageViewModelBase viewModel || viewModel == ViewModel.CurrentPage) |
||||||
|
{ |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
packageNavigationService.NavigateTo(viewModel, BetterSlideNavigationTransition.PageSlideFromLeft); |
||||||
|
} |
||||||
|
|
||||||
|
public bool GoBack() |
||||||
|
{ |
||||||
|
return packageNavigationService.GoBack(); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,195 @@ |
|||||||
|
<controls:UserControlBase xmlns="https://github.com/avaloniaui" |
||||||
|
xmlns:x="http://schemas.microsoft.com/winfx/2006/xaml" |
||||||
|
xmlns:d="http://schemas.microsoft.com/expression/blend/2008" |
||||||
|
xmlns:mc="http://schemas.openxmlformats.org/markup-compatibility/2006" |
||||||
|
xmlns:viewModels="clr-namespace:StabilityMatrix.Avalonia.ViewModels" |
||||||
|
xmlns:controls="clr-namespace:StabilityMatrix.Avalonia.Controls" |
||||||
|
xmlns:designData="clr-namespace:StabilityMatrix.Avalonia.DesignData" |
||||||
|
xmlns:lang="clr-namespace:StabilityMatrix.Avalonia.Languages" |
||||||
|
xmlns:avalonia="clr-namespace:FluentIcons.Avalonia;assembly=FluentIcons.Avalonia" |
||||||
|
xmlns:controls1="clr-namespace:FluentAvalonia.UI.Controls;assembly=FluentAvalonia" |
||||||
|
xmlns:models="clr-namespace:StabilityMatrix.Core.Models;assembly=StabilityMatrix.Core" |
||||||
|
xmlns:database="clr-namespace:StabilityMatrix.Core.Models.Database;assembly=StabilityMatrix.Core" |
||||||
|
xmlns:extensions="clr-namespace:StabilityMatrix.Core.Models.Packages.Extensions;assembly=StabilityMatrix.Core" |
||||||
|
mc:Ignorable="d" d:DesignWidth="800" d:DesignHeight="450" |
||||||
|
x:DataType="viewModels:PackageInstallDetailViewModel" |
||||||
|
x:CompileBindings="True" |
||||||
|
d:DataContext="{x:Static designData:DesignData.PackageInstallDetailViewModel}" |
||||||
|
x:Class="StabilityMatrix.Avalonia.Views.PackageInstallDetailView"> |
||||||
|
|
||||||
|
<Grid ColumnDefinitions="Auto, *" |
||||||
|
RowDefinitions="Auto, *"> |
||||||
|
<controls:BetterAdvancedImage |
||||||
|
MaxHeight="300" |
||||||
|
MaxWidth="300" |
||||||
|
Margin="16,16,8,8" |
||||||
|
CornerRadius="8" |
||||||
|
VerticalAlignment="Top" |
||||||
|
Stretch="UniformToFill" |
||||||
|
Source="{Binding SelectedPackage.PreviewImageUri}" /> |
||||||
|
<StackPanel Grid.Row="0" Grid.Column="1" |
||||||
|
Margin="8" |
||||||
|
Orientation="Vertical"> |
||||||
|
|
||||||
|
<controls:HyperlinkIconButton NavigateUri="{Binding SelectedPackage.LicenseUrl}" |
||||||
|
Content="{Binding SelectedPackage.LicenseType}"/> |
||||||
|
|
||||||
|
<ToggleSwitch OnContent="{x:Static lang:Resources.Label_Releases}" |
||||||
|
OffContent="{x:Static lang:Resources.Label_Branches}" |
||||||
|
IsVisible="{Binding ShowReleaseMode}" |
||||||
|
IsChecked="{Binding IsReleaseMode}" |
||||||
|
Margin="8,0,8,8" /> |
||||||
|
|
||||||
|
<!-- Version Selector --> |
||||||
|
<Label Margin="8,0" |
||||||
|
FontSize="14" |
||||||
|
FontWeight="Light" |
||||||
|
Content="{Binding ReleaseLabelText}" /> |
||||||
|
|
||||||
|
<controls1:FAComboBox |
||||||
|
ItemsSource="{Binding AvailableVersions}" |
||||||
|
MinWidth="250" |
||||||
|
HorizontalAlignment="Stretch" |
||||||
|
VerticalAlignment="Stretch" |
||||||
|
FontSize="16" |
||||||
|
Margin="8,0,8,8" |
||||||
|
SelectedItem="{Binding SelectedVersion}"> |
||||||
|
<controls1:FAComboBox.ItemTemplate> |
||||||
|
<DataTemplate DataType="{x:Type models:PackageVersion}"> |
||||||
|
<TextBlock |
||||||
|
Name="NameTextBlock" |
||||||
|
VerticalAlignment="Center" |
||||||
|
Text="{Binding TagName}" /> |
||||||
|
</DataTemplate> |
||||||
|
</controls1:FAComboBox.ItemTemplate> |
||||||
|
</controls1:FAComboBox> |
||||||
|
|
||||||
|
<Label Margin="8,0" |
||||||
|
FontSize="14" |
||||||
|
FontWeight="Light" |
||||||
|
ToolTip.Tip="This will also be the folder name" |
||||||
|
Content="{x:Static lang:Resources.Label_DisplayName}" /> |
||||||
|
<TextBox MinWidth="250" |
||||||
|
Margin="8,0" |
||||||
|
FontSize="16" |
||||||
|
FontWeight="SemiLight" |
||||||
|
VerticalAlignment="Stretch" |
||||||
|
VerticalContentAlignment="Center" |
||||||
|
ToolTip.Tip="This will also be the folder name" |
||||||
|
Text="{Binding InstallName, Mode=TwoWay}" /> |
||||||
|
<Label Margin="8,0" |
||||||
|
Content="{Binding FullInstallPath}" Foreground="LightGray" /> |
||||||
|
|
||||||
|
<StackPanel Orientation="Horizontal" |
||||||
|
Margin="8,8,8,8" |
||||||
|
IsVisible="{Binding ShowDuplicateWarning}"> |
||||||
|
<avalonia:SymbolIcon |
||||||
|
Foreground="{DynamicResource ThemeRedColor}" |
||||||
|
Margin="0,0,8,0" |
||||||
|
Symbol="Alert" /> |
||||||
|
<TextBlock |
||||||
|
Foreground="{DynamicResource ThemeRedColor}" |
||||||
|
TextAlignment="Left" |
||||||
|
TextWrapping="Wrap"> |
||||||
|
<Run Text="{x:Static lang:Resources.Label_InstallationWithThisNameExists}" /> |
||||||
|
<LineBreak /> |
||||||
|
<Run Text="{x:Static lang:Resources.Label_PleaseChooseDifferentName}" /> |
||||||
|
</TextBlock> |
||||||
|
</StackPanel> |
||||||
|
|
||||||
|
<Expander Header="Extensions" |
||||||
|
FontSize="16" |
||||||
|
HorizontalAlignment="Stretch" |
||||||
|
IsVisible="{Binding ShowExtensions}" |
||||||
|
Margin="8"> |
||||||
|
<ItemsRepeater ItemsSource="{Binding AvailableExtensions}"> |
||||||
|
<ItemsRepeater.Layout> |
||||||
|
<UniformGridLayout MinColumnSpacing="4" MinRowSpacing="4"/> |
||||||
|
</ItemsRepeater.Layout> |
||||||
|
<ItemsRepeater.ItemTemplate> |
||||||
|
<DataTemplate x:DataType="{x:Type viewModels:ExtensionViewModel}"> |
||||||
|
<CheckBox IsChecked="{Binding IsSelected, Mode=TwoWay}" |
||||||
|
Content="{Binding Extension.DisplayName}"/> |
||||||
|
</DataTemplate> |
||||||
|
</ItemsRepeater.ItemTemplate> |
||||||
|
</ItemsRepeater> |
||||||
|
</Expander> |
||||||
|
|
||||||
|
<Expander Header="{x:Static lang:Resources.Label_AdvancedOptions}" |
||||||
|
FontSize="16" |
||||||
|
HorizontalAlignment="Stretch" |
||||||
|
Margin="8"> |
||||||
|
<Grid ColumnDefinitions="*, *, *" |
||||||
|
RowDefinitions="Auto, Auto" |
||||||
|
Margin="8,0"> |
||||||
|
|
||||||
|
<Label Grid.Row="0" Grid.Column="0" |
||||||
|
Content="{x:Static lang:Resources.Label_SharedModelFolderStrategy}" |
||||||
|
Margin="0,0,0,4" /> |
||||||
|
<controls1:FAComboBox Grid.Row="1" Grid.Column="0" |
||||||
|
ItemsSource="{Binding SelectedPackage.AvailableSharedFolderMethods}" |
||||||
|
MinWidth="150" |
||||||
|
MinHeight="38" |
||||||
|
SelectedItem="{Binding SelectedSharedFolderMethod}"> |
||||||
|
<controls1:FAComboBox.ItemTemplate> |
||||||
|
<DataTemplate DataType="{x:Type models:SharedFolderMethod}"> |
||||||
|
<TextBlock |
||||||
|
Margin="8,4,0,4" |
||||||
|
Text="{Binding }" /> |
||||||
|
</DataTemplate> |
||||||
|
</controls1:FAComboBox.ItemTemplate> |
||||||
|
</controls1:FAComboBox> |
||||||
|
|
||||||
|
<Label Grid.Row="0" Grid.Column="1" |
||||||
|
Content="{x:Static lang:Resources.Label_PyTorchVersion}" |
||||||
|
IsVisible="{Binding ShowTorchVersionOptions}" |
||||||
|
Margin="0,0,0,4" /> |
||||||
|
<controls1:FAComboBox Grid.Row="1" Grid.Column="1" |
||||||
|
ItemsSource="{Binding SelectedPackage.AvailableTorchVersions}" |
||||||
|
MinWidth="150" |
||||||
|
MinHeight="38" |
||||||
|
IsVisible="{Binding ShowTorchVersionOptions}" |
||||||
|
SelectedItem="{Binding SelectedTorchVersion}"> |
||||||
|
<controls1:FAComboBox.ItemTemplate> |
||||||
|
<DataTemplate DataType="{x:Type models:TorchVersion}"> |
||||||
|
<TextBlock |
||||||
|
Margin="8,4,0,4" |
||||||
|
Text="{Binding }" /> |
||||||
|
</DataTemplate> |
||||||
|
</controls1:FAComboBox.ItemTemplate> |
||||||
|
</controls1:FAComboBox> |
||||||
|
|
||||||
|
<Label Grid.Row="0" Grid.Column="2" |
||||||
|
Content="{x:Static lang:Resources.Label_Commit}" |
||||||
|
Margin="0,0,0,4" |
||||||
|
IsVisible="{Binding !IsReleaseMode}" /> |
||||||
|
<controls1:FAComboBox Grid.Row="1" Grid.Column="2" |
||||||
|
IsVisible="{Binding !IsReleaseMode}" |
||||||
|
ItemsSource="{Binding AvailableCommits}" |
||||||
|
MinWidth="150" |
||||||
|
MinHeight="38" |
||||||
|
SelectedItem="{Binding SelectedCommit}"> |
||||||
|
<controls1:FAComboBox.ItemTemplate> |
||||||
|
<DataTemplate DataType="{x:Type database:GitCommit}"> |
||||||
|
<TextBlock |
||||||
|
Margin="8,4,0,4" |
||||||
|
Name="NameTextBlock" |
||||||
|
Text="{Binding ShortSha}" /> |
||||||
|
</DataTemplate> |
||||||
|
</controls1:FAComboBox.ItemTemplate> |
||||||
|
</controls1:FAComboBox> |
||||||
|
</Grid> |
||||||
|
</Expander> |
||||||
|
|
||||||
|
<Button Classes="success" |
||||||
|
Margin="8,8,8,8" |
||||||
|
HorizontalAlignment="Stretch" |
||||||
|
VerticalAlignment="Stretch" |
||||||
|
FontSize="16" |
||||||
|
Command="{Binding InstallCommand}" |
||||||
|
Content="{x:Static lang:Resources.Action_Install}" /> |
||||||
|
|
||||||
|
</StackPanel> |
||||||
|
</Grid> |
||||||
|
|
||||||
|
</controls:UserControlBase> |
||||||
@ -0,0 +1,13 @@ |
|||||||
|
using StabilityMatrix.Avalonia.Controls; |
||||||
|
using StabilityMatrix.Core.Attributes; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Avalonia.Views; |
||||||
|
|
||||||
|
[Transient] |
||||||
|
public partial class PackageInstallDetailView : UserControlBase |
||||||
|
{ |
||||||
|
public PackageInstallDetailView() |
||||||
|
{ |
||||||
|
InitializeComponent(); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,15 @@ |
|||||||
|
using StabilityMatrix.Core.Models.FileInterfaces; |
||||||
|
using StabilityMatrix.Core.Models.Packages.Extensions; |
||||||
|
using StabilityMatrix.Core.Models.Progress; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Core.Models.PackageModification; |
||||||
|
|
||||||
|
public class InstallExtensionStep(ExtensionBase extension, DirectoryPath extensionsDir) : IPackageStep |
||||||
|
{ |
||||||
|
public Task ExecuteAsync(IProgress<ProgressReport>? progress = null) |
||||||
|
{ |
||||||
|
return extension.InstallExtensionAsync(extensionsDir, extension.MainBranch); |
||||||
|
} |
||||||
|
|
||||||
|
public string ProgressTitle => $"Installing {extension.DisplayName}"; |
||||||
|
} |
||||||
@ -0,0 +1,7 @@ |
|||||||
|
namespace StabilityMatrix.Core.Models; |
||||||
|
|
||||||
|
public enum PackageType |
||||||
|
{ |
||||||
|
SdInference, |
||||||
|
SdTraining |
||||||
|
} |
||||||
@ -0,0 +1,28 @@ |
|||||||
|
using StabilityMatrix.Core.Helper; |
||||||
|
using StabilityMatrix.Core.Models.FileInterfaces; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Core.Models.Packages.Extensions; |
||||||
|
|
||||||
|
public class ComfyManager(IPrerequisiteHelper prerequisiteHelper) : ExtensionBase |
||||||
|
{ |
||||||
|
public override string RepoName => "ComfyUI-Manager"; |
||||||
|
public override string DisplayName { get; set; } = "ComfyUI Manager"; |
||||||
|
public override string Author => "ltdrdata"; |
||||||
|
|
||||||
|
public override string Blurb => |
||||||
|
"ComfyUI-Manager is an extension designed to enhance the usability of ComfyUI. It offers management functions to install, remove, disable, and enable various custom nodes of ComfyUI."; |
||||||
|
|
||||||
|
public override IEnumerable<string> CompatibleWith => [nameof(ComfyUI)]; |
||||||
|
public override string MainBranch => "main"; |
||||||
|
|
||||||
|
public override Task InstallExtensionAsync( |
||||||
|
DirectoryPath installDirectory, |
||||||
|
string branch, |
||||||
|
CancellationToken cancellationToken = default |
||||||
|
) |
||||||
|
{ |
||||||
|
return Directory.Exists(Path.Combine(installDirectory, RepoName)) |
||||||
|
? Task.CompletedTask |
||||||
|
: prerequisiteHelper.RunGit(new[] { "clone", GithubUrl }, installDirectory); |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,24 @@ |
|||||||
|
using StabilityMatrix.Core.Models.FileInterfaces; |
||||||
|
|
||||||
|
namespace StabilityMatrix.Core.Models.Packages.Extensions; |
||||||
|
|
||||||
|
public abstract class ExtensionBase |
||||||
|
{ |
||||||
|
public string ByAuthor => $"By {Author}"; |
||||||
|
|
||||||
|
public abstract string RepoName { get; } |
||||||
|
public abstract string DisplayName { get; set; } |
||||||
|
public abstract string Author { get; } |
||||||
|
|
||||||
|
public abstract string Blurb { get; } |
||||||
|
public abstract IEnumerable<string> CompatibleWith { get; } |
||||||
|
public abstract string MainBranch { get; } |
||||||
|
|
||||||
|
public abstract Task InstallExtensionAsync( |
||||||
|
DirectoryPath installDirectory, |
||||||
|
string branch, |
||||||
|
CancellationToken cancellationToken = default |
||||||
|
); |
||||||
|
|
||||||
|
public string GithubUrl => $"https://github.com/{Author}/{RepoName}"; |
||||||
|
} |
||||||
Loading…
Reference in new issue