mrleo1nid/ai-images
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Refactor AIImages mod to streamline API loading functionality, consolidating model, sampler, and scheduler loading into a single method. Update localized strings in English and Russian for improved clarity and consistency. Enhance UI with scrollable prompt display for better user experience. Update AIImages.dll to reflect these changes.

Browse Source
This commit is contained in:
Leonid Pershin
2025-10-26 18:38:18 +03:00
parent 2af1ef9292
commit 3434927342
5 changed files with 59 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
Source/AIImages/UI/AIImagesSettingsUI.cs
View File

@@ -70,18 +70,13 @@ namespace AIImages
_ = TestApiConnection(settings.apiEndpoint);
}
if (listingStandard.ButtonText("AIImages.Settings.LoadModels".Translate()))
if (listingStandard.ButtonText("AIImages.Settings.LoadFromApi".Translate()))
{
_ = LoadModelsFromApi(settings);
_ = LoadAllFromApi(settings);
}
DrawModelDropdown(listingStandard, settings);
if (listingStandard.ButtonText("AIImages.Settings.LoadSamplersSchedulers".Translate()))
{
_ = LoadSamplersAndSchedulers(settings);
}
listingStandard.Gap(12f);
}
@@ -376,23 +371,43 @@ namespace AIImages
}
}
private static async System.Threading.Tasks.Task LoadModelsFromApi(
private static async System.Threading.Tasks.Task LoadAllFromApi(
AIImagesModSettings settings
)
{
try
{
Log.Message("[AI Images] Loading models from API...");
Log.Message("[AI Images] Loading models, samplers and schedulers from API...");
// Загружаем модели
var models = await AIImagesMod.ApiService.GetAvailableModels(settings.apiEndpoint);
settings.availableModels = models;
if (models.Count > 0)
// Загружаем семплеры
var samplers = await AIImagesMod.ApiService.GetAvailableSamplers(
settings.apiEndpoint
);
settings.availableSamplers = samplers;
// Загружаем schedulers
var schedulers = await AIImagesMod.ApiService.GetAvailableSchedulers(
settings.apiEndpoint
);
settings.availableSchedulers = schedulers;
int totalCount = models.Count + samplers.Count + schedulers.Count;
if (totalCount > 0)
{
Messages.Message(
"AIImages.Settings.ModelsLoaded".Translate(models.Count),
"AIImages.Settings.AllLoaded".Translate(
models.Count,
samplers.Count,
schedulers.Count
),
MessageTypeDefOf.PositiveEvent
);
// Автовыбор модели
if (
(
string.IsNullOrEmpty(settings.selectedModel)
@@ -403,53 +418,8 @@ namespace AIImages
{
settings.selectedModel = models[0];
}
}
else
{
Messages.Message(
"AIImages.Settings.NoModelsFound".Translate(),
MessageTypeDefOf.RejectInput
);
}
}
catch (Exception ex)
{
Messages.Message(
$"Error loading models: {ex.Message}",
MessageTypeDefOf.RejectInput
);
}
}
private static async System.Threading.Tasks.Task LoadSamplersAndSchedulers(
AIImagesModSettings settings
)
{
try
{
Log.Message("[AI Images] Loading samplers and schedulers from API...");
var samplers = await AIImagesMod.ApiService.GetAvailableSamplers(
settings.apiEndpoint
);
settings.availableSamplers = samplers;
var schedulers = await AIImagesMod.ApiService.GetAvailableSchedulers(
settings.apiEndpoint
);
settings.availableSchedulers = schedulers;
int totalCount = samplers.Count + schedulers.Count;
if (totalCount > 0)
{
Messages.Message(
"AIImages.Settings.SamplersSchedulersLoaded".Translate(
samplers.Count,
schedulers.Count
),
MessageTypeDefOf.PositiveEvent
);
// Автовыбор семплера
if (
(
string.IsNullOrEmpty(settings.selectedSampler)
@@ -461,6 +431,7 @@ namespace AIImages
settings.selectedSampler = samplers[0];
}
// Автовыбор scheduler
if (
(
string.IsNullOrEmpty(settings.selectedScheduler)
@@ -475,7 +446,7 @@ namespace AIImages
else
{
Messages.Message(
"AIImages.Settings.NoSamplersSchedulersFound".Translate(),
"AIImages.Settings.NothingLoaded".Translate(),
MessageTypeDefOf.RejectInput
);
}
@@ -483,7 +454,7 @@ namespace AIImages
catch (Exception ex)
{
Messages.Message(
$"Error loading samplers/schedulers: {ex.Message}",
$"Error loading from API: {ex.Message}",
MessageTypeDefOf.RejectInput
);
}